Skip to content

Commit 1400289

Browse files
committed
Add __name__ attribute to a class if different from the name in dir()
1 parent 9b3c9a4 commit 1400289

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

pypredefgen.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,12 @@ def get_ast_node_for_class(class_element, module_root=None):
187187

188188
class_element.set_node(class_node)
189189

190+
if class_element.name_from_dir != class_element.object.__name__:
191+
class_element.node.body.insert(
192+
0, ast.Assign(
193+
targets=[ast.Name(id="__name__")],
194+
value=ast.Str(s=bytes(get_full_type_name(class_element.object, module_root)))))
195+
190196
insert_ast_nodes(class_element)
191197

192198
return class_node

0 commit comments

Comments
 (0)