diff options
| -rw-r--r-- | inspector.py | 6 | ||||
| -rw-r--r-- | utils.py | 5 |
2 files changed, 4 insertions, 7 deletions
diff --git a/inspector.py b/inspector.py index ff98feee..4ca46668 100644 --- a/inspector.py +++ b/inspector.py @@ -142,7 +142,7 @@ class Linker(IdGeneratorMixIn, LocalsVisitor): node.instance_attrs_type = {} for assattrs in node.instance_attrs.values(): for assattr in assattrs: - self.visit_assattr(assattr, node) + self.handle_assattr_type(assattr, node) # resolve implemented interface try: node.implements = list(node.interfaces(self.inherited_interfaces)) @@ -183,8 +183,8 @@ class Linker(IdGeneratorMixIn, LocalsVisitor): frame.locals_type[node.name] = values except astng.InferenceError: pass - - def visit_assattr(self, node, parent): + + def handle_assattr_type(self, node, parent): """visit an astng.AssAttr node handle instance_attrs_type @@ -112,10 +112,7 @@ class LocalsVisitor(ASTWalker): if recurse: if hasattr(node, 'locals') and not isinstance(node, Instance): for name, local_node in node.items(): - if local_node.__class__.__name__ == 'AssAttr': - self.visit_assattr(local_node, node) - else: - self.visit(local_node) + self.visit(local_node) if methods[1] is not None: return methods[1](node) |