moved maya implementation to openpype

openpype/hosts/maya/api/workio.py

@@ -0,0 +1,67 @@
+"""Host API required Work Files tool"""
+import os
+from maya import cmds
+from avalon import api
+
+
+def file_extensions():
+    return api.HOST_WORKFILE_EXTENSIONS["maya"]
+
+
+def has_unsaved_changes():
+    return cmds.file(query=True, modified=True)
+
+
+def save_file(filepath):
+    cmds.file(rename=filepath)
+    ext = os.path.splitext(filepath)[1]
+    if ext == ".mb":
+        file_type = "mayaBinary"
+    else:
+        file_type = "mayaAscii"
+    cmds.file(save=True, type=file_type)
+
+
+def open_file(filepath):
+    return cmds.file(filepath, open=True, force=True)
+
+
+def current_file():
+
+    current_filepath = cmds.file(query=True, sceneName=True)
+    if not current_filepath:
+        return None
+
+    return current_filepath
+
+
+def work_root(session):
+    work_dir = session["AVALON_WORKDIR"]
+    scene_dir = None
+
+    # Query scene file rule from workspace.mel if it exists in WORKDIR
+    # We are parsing the workspace.mel manually as opposed to temporarily
+    # setting the Workspace in Maya in a context manager since Maya had a
+    # tendency to crash on frequently changing the workspace when this
+    # function was called many times as one scrolled through Work Files assets.
+    workspace_mel = os.path.join(work_dir, "workspace.mel")
+    if os.path.exists(workspace_mel):
+        scene_rule = 'workspace -fr "scene" '
+        # We need to use builtins as `open` is overridden by the workio API
+        open_file = __builtins__["open"]
+        with open_file(workspace_mel, "r") as f:
+            for line in f:
+                if line.strip().startswith(scene_rule):
+                    # remainder == "rule";
+                    remainder = line[len(scene_rule):]
+                    # scene_dir == rule
+                    scene_dir = remainder.split('"')[1]
+    else:
+        # We can't query a workspace that does not exist
+        # so we return similar to what we do in other hosts.
+        scene_dir = session.get("AVALON_SCENEDIR")
+
+    if scene_dir:
+        return os.path.join(work_dir, scene_dir)
+    else:
+        return work_dir