forked from M-Labs/artiq
transforms.asttyped_rewriter: simplify.
This commit is contained in:
parent
2ca84f9fea
commit
42b0089a4c
|
@ -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,13 +48,11 @@ 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):
|
||||||
|
|
Loading…
Reference in New Issue