Blender 2.8 - 4.5 Support

leenkx/blender/lnx/nodes_logic.py

@@ -477,6 +477,7 @@ __REG_CLASSES = (
     LnxOpenNodeWikiEntry,
     LNX_OT_ReplaceNodesOperator,
     LNX_OT_RecalculateRotations,
+    LNX_MT_NodeAddOverride,
     LNX_OT_AddNodeOverride,
     LNX_UL_InterfaceSockets,
     LNX_PT_LogicNodePanel,
@@ -491,9 +492,8 @@ def register():
     lnx.logicnode.lnx_node_group.register()
     lnx.logicnode.tree_variables.register()
 
-    # Store original draw method and restore during unregister
+    LNX_MT_NodeAddOverride.overridden_menu = bpy.types.NODE_MT_add
     LNX_MT_NodeAddOverride.overridden_draw = bpy.types.NODE_MT_add.draw
-    bpy.types.NODE_MT_add.draw = LNX_MT_NodeAddOverride.draw
 
     __reg_classes()
 
@@ -508,11 +508,8 @@ def unregister():
     # Ensure that globals are reset if the addon is enabled again in the same Blender session
     lnx_nodes.reset_globals()
 
-    # Restore original draw method
-    if hasattr(LNX_MT_NodeAddOverride, 'overridden_draw'):
-        bpy.types.NODE_MT_add.draw = LNX_MT_NodeAddOverride.overridden_draw
-
     __unreg_classes()
+    bpy.utils.register_class(LNX_MT_NodeAddOverride.overridden_menu)
 
     lnx.logicnode.tree_variables.unregister()
     lnx.logicnode.lnx_node_group.unregister()