forked from M-Labs/artiq
1
0
Fork 0

transforms.asttyped_rewriter: simplify.

This commit is contained in:
whitequark 2015-10-07 22:03:24 +03:00
parent 2ca84f9fea
commit 42b0089a4c
1 changed files with 3 additions and 6 deletions

View File

@ -38,8 +38,7 @@ class LocalExtractor(algorithm.Visitor):
def visit_Assign(self, node): def visit_Assign(self, node):
self.visit(node.value) self.visit(node.value)
for target in node.targets: self.visit_in_assign(node.targets, in_assign=True)
self.visit_in_assign(target, in_assign=True)
def visit_For(self, node): def visit_For(self, node):
self.visit(node.iter) self.visit(node.iter)
@ -49,14 +48,12 @@ class LocalExtractor(algorithm.Visitor):
def visit_withitem(self, node): def visit_withitem(self, node):
self.visit(node.context_expr) self.visit(node.context_expr)
if node.optional_vars is not None: self.visit_in_assign(node.optional_vars, in_assign=True)
self.visit_in_assign(node.optional_vars, in_assign=True)
def visit_comprehension(self, node): def visit_comprehension(self, node):
self.visit(node.iter) self.visit(node.iter)
self.visit_in_assign(node.target, in_assign=True) self.visit_in_assign(node.target, in_assign=True)
for if_ in node.ifs: self.visit(node.ifs)
self.visit(node.ifs)
def visit_generator(self, node): def visit_generator(self, node):
if self.in_root: if self.in_root: