Initial render publishing

pype/plugins/harmony/publish/extract_render.py

@@ -0,0 +1,82 @@
+import os
+import tempfile
+
+import pyblish.api
+from avalon import harmony
+import pype.lib
+
+import clique
+
+
+class ExtractRender(pyblish.api.InstancePlugin):
+    """Produce a flattened image file from instance.
+    This plug-in only takes into account the nodes connected to the composite.
+    """
+
+    label = "Extract Render"
+    order = pyblish.api.ExtractorOrder
+    hosts = ["harmony"]
+    families = ["render"]
+
+    def process(self, instance):
+        # Collect scene data.
+        func = """function func(write_node)
+        {
+            return [
+                about.getApplicationPath(),
+                scene.currentProjectPath(),
+                scene.currentScene()
+            ]
+        }
+        func
+        """
+        result = harmony.send(
+            {"function": func, "args": [instance[0]]}
+        )["result"]
+        application_path = result[0]
+        project_path = result[1]
+        scene_path = os.path.join(result[1], result[2] + ".xstage")
+
+        # Set output path to temp folder.
+        path = tempfile.mkdtemp()
+        func = """function func(args)
+        {
+            node.setTextAttr(args[0], "DRAWING_NAME", 1, args[1]);
+            scene.saveAll();
+        }
+        func
+        """
+        result = harmony.send(
+            {
+                "function": func,
+                "args": [instance[0], path + "/" + instance.data["name"]]
+            }
+        )
+
+        # Execute rendering.
+        output = pype.lib._subprocess([application_path, "-batch", scene_path])
+        self.log.info(output)
+
+        # Collect rendered files.
+        files = os.listdir(path)
+        collections, remainder = clique.assemble(files, minimum_items=1)
+        assert not remainder, (
+            "There shouldn't have been a remainder for '%s': "
+            "%s" % (instance[0], remainder)
+        )
+        assert len(collections) == 1, (
+            "There should only be one image sequence in {}. Found: {}".format(
+                path, len(collections)
+            )
+        )
+
+        extension = os.path.splitext(list(collections[0])[0])[-1][1:]
+        representation = {
+            "name": extension,
+            "ext": extension,
+            "files": list(collections[0]),
+            "stagingDir": path,
+        }
+        instance.data["representations"] = [representation]
+
+        self.log.info("Extracted {instance} to {path}".format(**locals()))