@@ -39,7 +39,7 @@ def __init__(self, object_, name_from_dir, module, ast_node=None):
3939 self ._module = module
4040
4141 if ast_node is not None :
42- self .node = ast_node
42+ self .set_node ( ast_node )
4343 else :
4444 self ._node = None
4545
@@ -59,8 +59,7 @@ def module(self):
5959 def node (self ):
6060 return self ._node
6161
62- @node .setter
63- def node (self , node ):
62+ def set_node (self , node ):
6463 self ._node = node
6564
6665 self .node_element_map [node ] = self
@@ -116,12 +115,12 @@ def insert_ast_node(child_member_name, element, module=None):
116115 getattr (element .object , child_member_name , None ), child_member_name , module )
117116
118117 if inspect .ismodule (child_element .object ):
119- child_element .node = get_ast_node_for_import (child_element )
118+ child_element .set_node ( get_ast_node_for_import (child_element ) )
120119 element .node .body .insert (0 , child_element .node )
121120 elif (inspect .isclass (child_element .object )
122121 and _can_inspect_class_element (child_element )):
123- child_element .node = get_ast_node_for_class (
124- child_element , module_root = element .module )
122+ child_element .set_node (
123+ get_ast_node_for_class ( child_element , module_root = element .module ) )
125124
126125 element .node .body .append (child_element .node )
127126 insert_ast_docstring (child_element )
@@ -132,14 +131,14 @@ def insert_ast_node(child_member_name, element, module=None):
132131 element .node .body .insert (0 , get_ast_node_for_import_by_module_name (module_name ))
133132 elif inspect .isroutine (child_element .object ):
134133 if not inspect .isclass (element .object ):
135- child_element .node = get_ast_node_for_function (child_element )
134+ child_element .set_node ( get_ast_node_for_function (child_element ) )
136135 else :
137- child_element .node = get_ast_node_for_method (child_element )
136+ child_element .set_node ( get_ast_node_for_method (child_element ) )
138137
139138 element .node .body .append (child_element .node )
140139 insert_ast_docstring (child_element )
141140 else :
142- child_element .node = get_ast_node_for_assignment_of_type_to_name (child_element )
141+ child_element .set_node ( get_ast_node_for_assignment_of_type_to_name (child_element ) )
143142 element .node .body .append (child_element .node )
144143
145144
@@ -186,7 +185,7 @@ def get_ast_node_for_class(class_element, module_root=None):
186185 body = [],
187186 decorator_list = [])
188187
189- class_element .node = class_node
188+ class_element .set_node ( class_node )
190189
191190 insert_ast_nodes (class_element )
192191
0 commit comments