occheung
/
artiq
forked from M-Labs/artiq
1
0
Fork 0
Code Pull Requests Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
artiq/compiler/transforms/asttyped_rewriter.py
View File

@ -38,8 +38,7 @@ class LocalExtractor(algorithm.Visitor):
def visit_Assign(self, node):
self.visit(node.value)
for target in node.targets:
self.visit_in_assign(target, in_assign=True)
self.visit_in_assign(node.targets, in_assign=True)
def visit_For(self, node):
self.visit(node.iter)
@ -49,14 +48,12 @@ class LocalExtractor(algorithm.Visitor):
def visit_withitem(self, node):
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):
self.visit(node.iter)
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):
if self.in_root: