@@ -130,7 +130,7 @@ def __godot_extension_unregister_class():
130130 raise RuntimeError("Not registered !")
131131
132132 { spec .class_name + "." + spec .unregister_class_hook + "()" if spec .unregister_class_hook is not None else "" }
133- unregister_extension_class (b"{ spec .class_name } ")
133+ __godot_extension_unregister_extension_class (b"{ spec .class_name } ")
134134 gd_string_name_del(&__godot_extension_{ spec .class_name } _class_name)
135135 gd_string_name_del(&__godot_extension_{ spec .class_name } _parent_class_name)
136136
@@ -150,7 +150,7 @@ def __godot_extension_register_class():
150150 __godot_extension_{ spec .class_name } _class_name = gd_string_name_from_unchecked_pystr("{ spec .class_name } ")
151151 __godot_extension_{ spec .class_name } _parent_class_name = gd_string_name_from_unchecked_pystr("{ spec .parent_class_name } ")
152152
153- register_extension_class_creation (
153+ __godot_extension_register_extension_class_creation (
154154 b"{ spec .class_name } ",
155155 b"{ spec .parent_class_name } ",
156156 &{ spec .class_name } .__godot_extension_create_instance,
@@ -173,7 +173,7 @@ def __godot_extension_register_class():
173173
174174 code += textwrap .indent (
175175 f"""
176- register_extension_class_method (
176+ __godot_extension_register_extension_class_method (
177177 b"{ spec .class_name } ",
178178 b"{ method .method_name } ",
179179 &{ spec .class_name } .__godot_extension_class_meth_{ method .method_name } ,
@@ -228,7 +228,13 @@ def generate_injected_class_code(spec: ClassDef) -> str:
228228
229229
230230def generate_injected_module_code (spec : ModuleDef ) -> str :
231- code = ""
231+ code = """
232+ from godot.hazmat.extension_class cimport (
233+ register_extension_class_creation as __godot_extension_register_extension_class_creation,
234+ register_extension_class_method as __godot_extension_register_extension_class_method,
235+ unregister_extension_class as __godot_extension_unregister_extension_class,
236+ )
237+ """
232238
233239 for klass in spec .classes :
234240 code += f"""
0 commit comments