feat: error checking on menu type

Author: BrendanParmer

compositor/operator.py

@@ -185,7 +185,8 @@ def _process_node_tree(self, node_tree: CompositorNodeTree):
         self._write(f"{nt_var} = {nt_var}_node_group()\n", self._outer)
 
     def execute(self, context):
-        self._setup_options(context.scene.ntp_options)
+        if not self._setup_options(context.scene.ntp_options):
+            return {'CANCELLED'}
 
         #find node group to replicate
         if self.is_scene:

geometry/operator.py

@@ -171,7 +171,8 @@ def _apply_modifier(self, nt: GeometryNodeTree, nt_var: str):
 
 
     def execute(self, context):
-        self._setup_options(context.scene.ntp_options)
+        if not self._setup_options(context.scene.ntp_options):
+            return {'CANCELLED'}
 
         #find node group to replicate
         nt = bpy.data.node_groups[self.geo_nodes_group_name]

ntp_operator.py

@@ -113,7 +113,7 @@ def _write(self, string: str, indent: str = None):
             indent = self._inner
         self._file.write(f"{indent}{string}\n")
 
-    def _setup_options(self, options: NTPOptions) -> None:
+    def _setup_options(self, options: NTPOptions) -> bool:
         # General
         self._mode = options.mode
         self._include_group_socket_values = options.include_group_socket_values
@@ -134,7 +134,12 @@ def _setup_options(self, options: NTPOptions) -> None:
             self._location = options.location
             self._category = options.category
             self._custom_category = options.custom_category
-            self._menu_id = options.menu_id
+            if options.menu_id in dir(bpy.types):
+                self._menu_id = options.menu_id
+            else:
+                self.report({'ERROR'}, f"{options.menu_id} is not a valid menu")
+                return False
+            return True
 
     def _setup_addon_directories(self, context: Context, nt_var: str) -> bool:
         """

shader/operator.py

@@ -128,7 +128,8 @@ def _process_node_tree(self, node_tree: ShaderNodeTree) -> None:
 
 
     def execute(self, context):
-        self._setup_options(context.scene.ntp_options)
+        if not self._setup_options(context.scene.ntp_options):
+            return {'CANCELLED'}
 
         #find node group to replicate
         self._base_node_tree = bpy.data.materials[self.material_name].node_tree