Merge pull request #141 from BrendanParmer/Pre41MenuSocketFix

Pre-4.1 menu socket fix

Author: BrendanParmer

NodeToPython/__init__.py

@@ -2,7 +2,7 @@
     "name": "Node to Python", 
     "description": "Convert Blender node groups to a Python add-on!",
     "author": "Brendan Parmer",
-    "version": (3, 3, 0),
+    "version": (3, 3, 1),
     "blender": (3, 0, 0),
     "location": "Node", 
     "category": "Node",

NodeToPython/blender_manifest.toml

@@ -1,7 +1,7 @@
 schema_version = "1.0.0"
 
 id = "node_to_python"
-version = "3.3.0"
+version = "3.3.1"
 name = "Node To Python"
 tagline = "Turn node groups into Python code"
 maintainer = "Brendan Parmer <[email protected]>"

NodeToPython/ntp_operator.py

@@ -569,21 +569,22 @@ def _set_tree_socket_defaults(self, socket_interface: NodeTreeInterfaceSocket,
 
             dv = socket_interface.default_value
 
-            if type(socket_interface) is bpy.types.NodeTreeInterfaceSocketMenu:
-                if dv == "":
-                    self.report({'WARNING'},
-                        "NodeToPython: No menu found for socket "
-                        f"{socket_interface.name}"
+            if bpy.app.version >= (4, 1, 0):
+                if type(socket_interface) is bpy.types.NodeTreeInterfaceSocketMenu:
+                    if dv == "":
+                        self.report({'WARNING'},
+                            "NodeToPython: No menu found for socket "
+                            f"{socket_interface.name}"
+                        )
+                        return
+
+                    self._write_after_links.append(
+                        lambda _socket_var=socket_var, _dv=enum_to_py_str(dv): (
+                            self._write(f"{_socket_var}.default_value = {_dv}")
+                        )
                     )
                     return
-
-                self._write_after_links.append(
-                    lambda _socket_var=socket_var, _dv=enum_to_py_str(dv): (
-                        self._write(f"{_socket_var}.default_value = {_dv}")
-                    )
-                )
-                return
-            elif type(socket_interface) == bpy.types.NodeTreeInterfaceSocketColor:
+            if type(socket_interface) == bpy.types.NodeTreeInterfaceSocketColor:
                 dv = vec4_to_py_str(dv)
             elif type(dv) in {mathutils.Vector, mathutils.Euler}:
                 dv = vec3_to_py_str(dv)