summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorEmile Anclin <emile.anclin@logilab.fr>2009-03-03 17:22:10 +0100
committerEmile Anclin <emile.anclin@logilab.fr>2009-03-03 17:22:10 +0100
commit77dc4b2f200bec5e5527a85e9cb0b30ba2fe6fad (patch)
tree3cf7608fcbfaf9e268081103a5241be24ab21ef7
parent8fa49d6c6f4bede06586cbeb81e3f893dd6770db (diff)
downloadastroid-git-77dc4b2f200bec5e5527a85e9cb0b30ba2fe6fad.tar.gz
fix AssName/AssAttr transformation
--HG-- branch : _ast_compat
-rw-r--r--_nodes_ast.py16
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