Skip to content

Commit 9b3c9a4

Browse files
committed
Refactoring: Rename Element.node setter to Element.set_node
1 parent 041654e commit 9b3c9a4

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

pypredefgen.py

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)