diff --git a/main.py b/main.py index 1d0bfa7..de28b1b 100755 --- a/main.py +++ b/main.py @@ -83,9 +83,7 @@ if code[0:2] == '#!': next_line = code.find('\n') + 1 code = code[next_line:] - tokens = parse.tokenize(code) - parser = parse.Parser(tokens) - ast = parser.parse_file() + ast = parse.parse_file(code) if not ast: return 1 env = { diff --git a/parse.py b/parse.py index a9573e2..7092b30 100644 --- a/parse.py +++ b/parse.py @@ -203,3 +203,8 @@ ast.append(directive) print("Parsed file") return ast + +def parse_file(code): + tokens = tokenize(code) + parser = Parser(tokens) + return parser.parse_file()