node menu tweak.

Author: jchanvfx

NodeGraphQt/widgets/actions.py

@@ -9,9 +9,11 @@ def __init__(self, *args, **kwargs):
         super(BaseMenu, self).__init__(*args, **kwargs)
         self.setStyleSheet(STYLE_QMENU)
 
-    # def hideEvent(self, event):
-    #     super(BaseMenu, self).hideEvent(event)
-    #     print('foo')
+    def hideEvent(self, event):
+        super(BaseMenu, self).hideEvent(event)
+        for a in self.actions():
+            if hasattr(a, 'node_id'):
+                a.node_id = None
 
     def get_menu(self, name):
         for action in self.actions():

NodeGraphQt/widgets/viewer.py

@@ -163,12 +163,12 @@ def contextMenuEvent(self, event):
             pos = self.mapToScene(self._previous_pos)
             items = self._items_near(pos)
             nodes = [i for i in items if isinstance(i, AbstractNodeItem)]
-            # if nodes:
-            #     node = nodes[0]
-            #     ctx_menu = self._ctx_node_menu.get_menu(node.type_)
-            #     for action in ctx_menu.actions():
-            #         if not action.menu():
-            #             action.node_id = node.id
+            if nodes:
+                node = nodes[0]
+                ctx_menu = self._ctx_node_menu.get_menu(node.type_)
+                for action in ctx_menu.actions():
+                    if not action.menu():
+                        action.node_id = node.id
         if ctx_menu and ctx_menu.isEnabled():
             ctx_menu.exec_(event.globalPos())
         else: