Create workfile using auto creator

openpype/hosts/maya/plugins/create/create_workfile.py

@@ -0,0 +1,90 @@
+# -*- coding: utf-8 -*-
+"""Creator plugin for creating workfiles."""
+from openpype.pipeline import CreatedInstance, AutoCreator
+from openpype.pipeline import legacy_io
+from openpype.client import get_asset_by_name
+from openpype.hosts.maya.api import plugin
+from maya import cmds
+
+
+class CreateWorkfile(plugin.MayaCreatorBase, AutoCreator):
+    """Workfile auto-creator."""
+    identifier = "io.openpype.creators.maya.workfile"
+    label = "Workfile"
+    family = "workfile"
+    icon = "document"
+
+    default_variant = "Main"
+
+    def create(self):
+
+        print("Create...")
+        variant = self.default_variant
+        current_instance = next(
+            (
+                instance for instance in self.create_context.instances
+                if instance.creator_identifier == self.identifier
+            ), None)
+
+        project_name = self.project_name
+        asset_name = legacy_io.Session["AVALON_ASSET"]
+        task_name = legacy_io.Session["AVALON_TASK"]
+        host_name = legacy_io.Session["AVALON_APP"]
+
+        if current_instance is None:
+            asset_doc = get_asset_by_name(project_name, asset_name)
+            subset_name = self.get_subset_name(
+                variant, task_name, asset_doc, project_name, host_name
+            )
+            data = {
+                "asset": asset_name,
+                "task": task_name,
+                "variant": variant
+            }
+            data.update(
+                self.get_dynamic_data(
+                    variant, task_name, asset_doc,
+                    project_name, host_name, current_instance)
+            )
+            self.log.info("Auto-creating workfile instance...")
+            current_instance = CreatedInstance(
+                self.family, subset_name, data, self
+            )
+            self._add_instance_to_context(current_instance)
+        elif (
+                current_instance["asset"] != asset_name
+                or current_instance["task"] != task_name
+        ):
+            # Update instance context if is not the same
+            asset_doc = get_asset_by_name(project_name, asset_name)
+            subset_name = self.get_subset_name(
+                variant, task_name, asset_doc, project_name, host_name
+            )
+            current_instance["asset"] = asset_name
+            current_instance["task"] = task_name
+            current_instance["subset"] = subset_name
+
+    def collect_instances(self):
+        self.cache_subsets(self.collection_shared_data)
+        cached_subsets = self.collection_shared_data["maya_cached_subsets"]
+        for node in cached_subsets.get(self.identifier, []):
+            node_data = self.read_instance_node(node)
+
+            created_instance = CreatedInstance.from_existing(node_data, self)
+            self._add_instance_to_context(created_instance)
+
+    def update_instances(self, update_list):
+        for created_inst, _changes in update_list:
+            data = created_inst.data_to_store()
+            node = data.get("instance_node")
+            if not node:
+                node = self.create_node()
+                created_inst["instance_node"] = node
+                data = created_inst.data_to_store()
+
+            self.imprint_instance_node(node, data)
+
+    def create_node(self):
+        node = cmds.sets(empty=True, name="workfileMain")
+        cmds.setAttr(node + ".hiddenInOutliner", True)
+        return node

openpype/hosts/maya/plugins/publish/collect_current_file.py

@@ -0,0 +1,17 @@
+
+import pyblish.api
+
+from maya import cmds
+
+
+class CollectCurrentFile(pyblish.api.ContextPlugin):
+    """Inject the current working file."""
+
+    order = pyblish.api.CollectorOrder - 0.4
+    label = "Maya Current File"
+    hosts = ['maya']
+    families = ["workfile"]
+
+    def process(self, context):
+        """Inject the current working file"""
+        context.data['currentFile'] = cmds.file(query=True, sceneName=True)

openpype/hosts/maya/plugins/publish/collect_workfile.py

@@ -1,46 +1,30 @@
 import os
 import pyblish.api
 
-from maya import cmds
-from openpype.pipeline import legacy_io
 
-
-class CollectWorkfile(pyblish.api.ContextPlugin):
-    """Inject the current working file into context"""
+class CollectWorkfileData(pyblish.api.InstancePlugin):
+    """Inject data into Workfile instance"""
 
     order = pyblish.api.CollectorOrder - 0.01
     label = "Maya Workfile"
     hosts = ['maya']
+    families = ["workfile"]
 
-    def process(self, context):
+    def process(self, instance):
         """Inject the current working file"""
-        current_file = cmds.file(query=True, sceneName=True)
-        context.data['currentFile'] = current_file
 
+        context = instance.context
+        current_file = instance.context.data['currentFile']
         folder, file = os.path.split(current_file)
         filename, ext = os.path.splitext(file)
 
-        task = legacy_io.Session["AVALON_TASK"]
-
-        data = {}
-
-        # create instance
-        instance = context.create_instance(name=filename)
-        subset = 'workfile' + task.capitalize()
-
-        data.update({
-            "subset": subset,
-            "asset": os.getenv("AVALON_ASSET", None),
-            "label": subset,
-            "publish": True,
-            "family": 'workfile',
-            "families": ['workfile'],
+        data = {  # noqa
             "setMembers": [current_file],
             "frameStart": context.data['frameStart'],
             "frameEnd": context.data['frameEnd'],
             "handleStart": context.data['handleStart'],
             "handleEnd": context.data['handleEnd']
-        })
+        }
 
         data['representations'] = [{
             'name': ext.lstrip("."),
@@ -50,8 +34,3 @@ class CollectWorkfile(pyblish.api.ContextPlugin):
         }]
 
         instance.data.update(data)
-
-        self.log.info('Collected instance: {}'.format(file))
-        self.log.info('Scene path: {}'.format(current_file))
-        self.log.info('staging Dir: {}'.format(folder))
-        self.log.info('subset: {}'.format(subset))