diff --git a/parse.py b/parse.py index 9e9666f..f5e8fe3 100644 --- a/parse.py +++ b/parse.py @@ -184,6 +184,9 @@ log.log(log.PARSER, log.TRACE, "Peeked %s" % (token)) return token + def eof(self): + return self.tokens[self.pos].type == "EOF" + def create_context(self, context, text): return ParseContext(context, text, self.tokens[self.pos]) @@ -317,7 +320,7 @@ log.log(log.PARSER, log.TRACE, "Parsing file...") ast = [] version = self.parse_version(None) - while self.peek().type != "EOF": + while not self.eof(): log.log(log.PARSER, log.TRACE, "Parsing next directive in file...") ast.append(self.parse_directive(None)) log.log(log.PARSER, log.DEBUG, "Parsed file, AST is %s" % (ast))