fix: function name collision with exporting multiple objects of the same type

Author: BrendanParmer

NodeToPython/export/compositor/exporter.py

@@ -112,7 +112,9 @@ def _initialize_node_tree(
     ) -> None:
         nt_name = ntp_node_tree._node_tree.name
 
-        self._node_tree_info._func = f"{ntp_node_tree._var}_node_group"
+        self._node_tree_info._func = self._operator._create_var(
+            f"{ntp_node_tree._var}_node_group"
+        )
         #initialize node group
         self._write(f"def {self._node_tree_info._func}("
                     f"{NODE_TREE_NAMES}: dict[typing.Callable, str]):", 

NodeToPython/export/geometry/exporter.py

@@ -87,7 +87,9 @@ def _initialize_node_tree(
         ntp_node_tree: NTP_NodeTree
     ) -> None:
         nt_name = ntp_node_tree._node_tree.name
-        self._node_tree_info._func = f"{ntp_node_tree._var}_node_group"
+        self._node_tree_info._func = self._operator._create_var(
+            f"{ntp_node_tree._var}_node_group"
+        )
         #initialize node group
         self._write(f"def {self._node_tree_info._func}("
                     f"{NODE_TREE_NAMES}: dict[typing.Callable, str]):", 

NodeToPython/export/ntp_operator.py

@@ -78,8 +78,10 @@ def __init__(self, *args, **kwargs):
         for name in RESERVED_NAMES:
             self._used_vars[name] = 0
 
+        # Useful information about exported node trees
         self._node_trees: dict[bpy.types.NodeTree, NodeTreeInfo] = {}
 
+        # Number of objects we end up exporting
         self._num_objs: int = 0
 
         # Generate socket default, min, and max values

NodeToPython/export/shader/exporter.py

@@ -53,7 +53,9 @@ def _create_obj(self):
     # NodeTreeExporter interface
     def _initialize_node_tree(self, ntp_node_tree: NTP_NodeTree) -> None:
         nt_name = ntp_node_tree._node_tree.name
-        self._node_tree_info._func = f"{ntp_node_tree._var}_node_group"
+        self._node_tree_info._func = self._operator._create_var(
+            f"{ntp_node_tree._var}_node_group"
+        )
         #initialize node group
         self._write(f"def {self._node_tree_info._func}("
                     f"{NODE_TREE_NAMES}: dict[typing.Callable, str]):",