diff options
| -rw-r--r-- | _nodes_ast.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/_nodes_ast.py b/_nodes_ast.py index 698a7a98..a2c16cfa 100644 --- a/_nodes_ast.py +++ b/_nodes_ast.py @@ -221,11 +221,11 @@ class TreeRebuilder(ASTVisitor): node.attrname = node.attr node.expr = node.value del node.attr, node.value - if isinstance(self.visitor.asscontext, Assign): - node.__class__ = AssAttr - elif isinstance(self.visitor.asscontext, Delete): + if isinstance(self.visitor.asscontext, Delete): node.__class__ = DelAttr - + elif self.visitor.asscontext is not None: + node.__class__ = AssAttr + def visit_if(self, node): tests, orelse = _recurse_if(node, [], []) node.tests = tests @@ -248,11 +248,11 @@ class TreeRebuilder(ASTVisitor): def visit_name(self, node): node.name = node.id del node.id - if isinstance(self.visitor.asscontext, Assign): - node.__class__ = AssName - elif isinstance(self.visitor.asscontext, Delete): + if isinstance(self.visitor.asscontext, Delete): node.__class__ = DelName - + elif self.visitor.asscontext is not None: + node.__class__ = AssName + def visit_num(self, node): node.__class__ = Const node.value = node.n |