Merge pull request #877 from pypeclub/feature/653-store-workfiles-information-in-the-db

Store workfiles information in the db

pype/hosts/maya/menu.py

@@ -13,12 +13,14 @@ self._menu = os.environ.get('PYPE_STUDIO_NAME') or "Pype"
 log = logging.getLogger(__name__)
 
 
-def _get_menu():
+def _get_menu(menu_name=None):
     """Return the menu instance if it currently exists in Maya"""
 
+    if menu_name is None:
+        menu_name = self._menu
     widgets = dict((
         w.objectName(), w) for w in QtWidgets.QApplication.allWidgets())
-    menu = widgets.get(self._menu)
+    menu = widgets.get(menu_name)
     return menu
 
 
@@ -40,10 +42,51 @@ def deferred():
             command=lambda *args: mayalookassigner.show()
         )
 
+    def modify_workfiles():
+        from pype.tools import workfiles
+
+        def launch_workfiles_app(*_args, **_kwargs):
+            workfiles.show(
+                os.path.join(
+                    cmds.workspace(query=True, rootDirectory=True),
+                    cmds.workspace(fileRuleEntry="scene")
+                ),
+                parent=pipeline._parent
+            )
+
+        # Find the pipeline menu
+        top_menu = _get_menu(pipeline._menu)
+
+        # Try to find workfile tool action in the menu
+        workfile_action = None
+        for action in top_menu.actions():
+            if action.text() == "Work Files":
+                workfile_action = action
+                break
+
+        # Add at the top of menu if "Work Files" action was not found
+        after_action = ""
+        if workfile_action:
+            # Use action's object name for `insertAfter` argument
+            after_action = workfile_action.objectName()
+
+        # Insert action to menu
+        cmds.menuItem(
+            "Work Files",
+            parent=pipeline._menu,
+            command=launch_workfiles_app,
+            insertAfter=after_action
+        )
+
+        # Remove replaced action
+        if workfile_action:
+            top_menu.removeAction(workfile_action)
+
     log.info("Attempting to install scripts menu..")
 
     add_build_workfiles_item()
     add_look_assigner_item()
+    modify_workfiles()
 
     try:
         import scriptsmenu.launchformaya as launchformaya

pype/hosts/nuke/menu.py

@@ -1,9 +1,11 @@
+import os
 import nuke
 from avalon.api import Session
 
 from pype.hosts.nuke import lib
 from ...lib import BuildWorkfile
 from pype.api import Logger
+from pype.tools import workfiles
 
 log = Logger().get_logger(__name__)
 
@@ -12,6 +14,24 @@ def install():
     menubar = nuke.menu("Nuke")
     menu = menubar.findItem(Session["AVALON_LABEL"])
     workfile_settings = lib.WorkfileSettings
+
+    # replace reset resolution from avalon core to pype's
+    name = "Work Files..."
+    rm_item = [
+        (i, item) for i, item in enumerate(menu.items()) if name in item.name()
+    ][0]
+
+    log.debug("Changing Item: {}".format(rm_item))
+
+    menu.removeItem(rm_item[1].name())
+    menu.addCommand(
+        name,
+        lambda: workfiles.show(
+            os.environ["AVALON_WORKDIR"]
+        ),
+        index=(rm_item[0])
+    )
+
     # replace reset resolution from avalon core to pype's
     name = "Reset Resolution"
     new_name = "Set Resolution"