Move get_node() to tree_node and replace all the custom classdefs/funcdefs.

This commit is contained in:
Dave Halter
2017-01-05 23:43:12 +01:00
parent b44f0aae5d
commit ae8e43d3c7
24 changed files with 90 additions and 99 deletions
+2 -2
View File
@@ -71,7 +71,7 @@ def _fix_forward_reference(context, node):
else:
module = node.get_parent_until()
new_node.move(module.end_pos[0])
new_node.parent = context.get_node()
new_node.parent = context.tree_node
return new_node
else:
return node
@@ -154,7 +154,7 @@ def py__getitem__(context, typ, node):
assert len(factories) == 1
factory = list(factories)[0]
assert factory
function_body_nodes = factory.funcdef.children[4].children
function_body_nodes = factory.tree_node.children[4].children
valid_classnames = set(child.name.value
for child in function_body_nodes
if isinstance(child, tree.Class))