Update test.py

simonlindholm · Sep 25, 2020 · 8a2c931 · 8a2c931
1 parent 2d3b306
commit 8a2c931
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/test.py b/test.py
@@ -1,14 +1,13 @@
+#!/usr/bin/env python3
 import sys
 
 from pycparser import parse_file, c_generator
 
-if len(sys.argv) < 2:
-    print(f"Usage: {sys.argv[0]} filename.c")
-    exit()
+fname = "test.c" if len(sys.argv) < 2 else sys.argv[1]
 
 
 # ast = c_parser.CParser().parse(src)
-ast = parse_file(sys.argv[1], use_cpp=True)
+ast = parse_file(fname, use_cpp=True)
 # ast.show()
 # print(c_generator.CGenerator().visit(ast))
 print(ast)