refactor: more zone cleanup

Author: BrendanParmer

NodeToPython/geometry/node_tree.py

@@ -15,10 +15,10 @@
 class NTP_GeoNodeTree(NTP_NodeTree):
     def __init__(self, node_tree: GeometryNodeTree, var: str):
         super().__init__(node_tree, var)
-        self.zone_inputs_: dict[list[GeometryNode]] = {}
+        self.zone_inputs: dict[list[GeometryNode]] = {}
         if bpy.app.version >= (3, 6, 0):
-            self.zone_inputs_["GeometryNodeSimulationInput"] = []
+            self.zone_inputs["GeometryNodeSimulationInput"] = []
         if bpy.app.version >= (4, 0, 0):
-            self.zone_inputs_["GeometryNodeRepeatInput"] = []
+            self.zone_inputs["GeometryNodeRepeatInput"] = []
         if bpy.app.version >= (4, 3, 0):
-            self.zone_inputs_["GeometryNodeForeachGeometryElementInput"] = []
+            self.zone_inputs["GeometryNodeForeachGeometryElementInput"] = []

NodeToPython/geometry/operator.py

@@ -49,14 +49,12 @@ def _process_node(self, node: Node, ntp_nt: NTP_GeoNodeTree) -> None:
                 self._group_io_settings(node, "output", ntp_nt)
                 ntp_nt.outputs_set = True
 
-        if node.bl_idname in ntp_nt.zone_inputs_:
-            ntp_nt.zone_inputs_[node.bl_idname].append(node)
+        if node.bl_idname in ntp_nt.zone_inputs:
+            ntp_nt.zone_inputs[node.bl_idname].append(node)
 
         self._hide_hidden_sockets(node)
 
-        if node.bl_idname not in {'GeometryNodeSimulationInput', 
-                                  'GeometryNodeRepeatInput',
-                                  'GeometryNodeForeachGeometryElementInput'}:
+        if node.bl_idname not in ntp_nt.zone_inputs:
             self._set_socket_defaults(node)
 
     if bpy.app.version >= (3, 6, 0):
@@ -137,7 +135,7 @@ def _process_node_tree(self, node_tree: GeometryNodeTree) -> None:
         for node in node_tree.nodes:
             self._process_node(node, ntp_nt)
 
-        for zone_list in ntp_nt.zone_inputs_.values():
+        for zone_list in ntp_nt.zone_inputs.values():
             self._process_zones(zone_list)
 
         #set look of nodes