diff --git a/main.py b/main.py index 7dac268..be4fb71 100755 --- a/main.py +++ b/main.py @@ -43,8 +43,11 @@ print("Usage: main.py FILENAME") return 1 filename = args[1] - testcode = open(filename).read() - tokens = tokenizer(testcode) + code = open(filename).read() + if code[0:2] == '#!': + next_line = code.find('\n') + 1 + code = code[next_line:] + tokens = tokenizer(code) for t in tokens: print("TOKEN {0}: {1}".format(t[0], t[1])) return 0