node viewer expose undo & redo actions.

Author: jchanvfx

NodeGraphQt/base/graph.py

@@ -2151,8 +2151,11 @@ def _clone_context_menu_from_parent(self):
         """
         Clone the context menus from the parent node graph.
         """
-        undo_regex = re.compile(r'^\&Undo(?:| \w+)')
-        redo_regex = re.compile(r'^\&Redo(?:| \w+)')
+        graph_menu = self.get_context_menu('graph')
+        parent_menu = self.parent_graph.get_context_menu('graph')
+        parent_viewer = self.parent_graph.viewer()
+        excl_actions = [parent_viewer.qaction_for_undo(),
+                        parent_viewer.qaction_for_redo()]
 
         def clone_menu(menu, menu_to_clone):
             """
@@ -2161,7 +2164,7 @@ def clone_menu(menu, menu_to_clone):
                 menu_to_clone (NodeGraphQt.NodeGraphMenu):
             """
             sub_items = []
-            for idx, item in enumerate(menu_to_clone.get_items()):
+            for item in menu_to_clone.get_items():
                 if item is None:
                     menu.add_separator()
                     continue
@@ -2171,9 +2174,9 @@ def clone_menu(menu, menu_to_clone):
                     sub_items.append([sub_menu, item])
                     continue
 
-                if idx <= 1:
-                    if undo_regex.match(name) or redo_regex.match(name):
-                        continue
+                if item in excl_actions:
+                    continue
+
                 menu.add_command(
                     name,
                     func=item.slot_function,
@@ -2183,8 +2186,7 @@ def clone_menu(menu, menu_to_clone):
             for sub_menu, to_clone in sub_items:
                 clone_menu(sub_menu, to_clone)
 
-        graph_menu = self.get_context_menu('graph')
-        parent_menu = self.parent_graph.get_context_menu('graph')
+        # duplicate the menu items.
         clone_menu(graph_menu, parent_menu)
 
     def _build_port_nodes(self):

NodeGraphQt/widgets/viewer.py

@@ -950,7 +950,31 @@ def rebuild_tab_search(self):
         if isinstance(self._search_widget, TabSearchMenuWidget):
             self._search_widget.rebuild = True
 
+    def qaction_for_undo(self):
+        """
+        Get the undo QAction from the parent undo stack.
+
+        Returns:
+            QtWidgets.QAction: undo action.
+        """
+        return self._undo_action
+
+    def qaction_for_redo(self):
+        """
+        Get the redo QAction from the parent undo stack.
+
+        Returns:
+            QtWidgets.QAction: redo action.
+        """
+        return self._redo_action
+
     def context_menus(self):
+        """
+        All the available context menus for the viewer.
+
+        Returns:
+            dict: viewer context menu.
+        """
         return {'graph': self._ctx_graph_menu, 'nodes': self._ctx_node_menu}
 
     def question_dialog(self, text, title='Node Graph'):