Update leenkx/blender/lnx/nodes_logic.py

This commit is contained in:
2025-09-27 03:03:08 +00:00
parent 6c3efa6c83
commit a926fa8dbb

View File

@ -477,7 +477,6 @@ __REG_CLASSES = (
LnxOpenNodeWikiEntry,
LNX_OT_ReplaceNodesOperator,
LNX_OT_RecalculateRotations,
LNX_MT_NodeAddOverride,
LNX_OT_AddNodeOverride,
LNX_UL_InterfaceSockets,
LNX_PT_LogicNodePanel,
@ -492,8 +491,9 @@ def register():
lnx.logicnode.lnx_node_group.register()
lnx.logicnode.tree_variables.register()
LNX_MT_NodeAddOverride.overridden_menu = bpy.types.NODE_MT_add
# Store original draw method and restore during unregister
LNX_MT_NodeAddOverride.overridden_draw = bpy.types.NODE_MT_add.draw
bpy.types.NODE_MT_add.draw = LNX_MT_NodeAddOverride.draw
__reg_classes()
@ -508,8 +508,11 @@ 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()