Start implementing GlobalStmt.

This commit is contained in:
Dave Halter
2014-10-17 15:13:45 +02:00
parent 6abafc40fa
commit f08811fba7
3 changed files with 14 additions and 2 deletions
+8 -1
View File
@@ -49,6 +49,7 @@ class Parser(object):
if not source.endswith('\n'):
source += '\n'
self.global_names = []
#if self.options["print_function"]:
# python_grammar = pygram.python_grammar_no_print_statement
#else:
@@ -56,10 +57,16 @@ class Parser(object):
# files processed even if they were not changed during refactoring. If
# and only if the refactor method's write parameter was True.
logger = logging.getLogger("RefactoringTool")
d = Driver(pytree.python_grammar, convert=pytree.convert, logger=logger)
d = Driver(pytree.python_grammar, convert=self.convert, logger=logger)
print(repr(source))
self.module = d.parse_string(source).get_parent_until()
def convert(self, grammar, raw_node):
new_node = pytree.convert(grammar, raw_node)
if isinstance(new_node, pr.GlobalStmt):
self.global_names += new_node.names()
return new_node
def __init__old__(self, source, module_path=None, no_docstr=False,
tokenizer=None, top_module=None):
self.no_docstr = no_docstr