OP-2895 - added functionality to create flatten image if no instances present

openpype/hosts/photoshop/plugins/publish/collect_instances.py

@@ -1,6 +1,9 @@
+from avalon import api
 import pyblish.api
 
+from openpype.settings import get_project_settings
 from openpype.hosts.photoshop import api as photoshop
+from openpype.lib import prepare_template_data
 
 
 class CollectInstances(pyblish.api.ContextPlugin):
@@ -19,13 +22,16 @@ class CollectInstances(pyblish.api.ContextPlugin):
     families_mapping = {
         "image": []
     }
+    flatten_subset_template = ""
 
     def process(self, context):
         stub = photoshop.stub()
         layers = stub.get_layers()
         layers_meta = stub.get_layers_metadata()
         instance_names = []
+        all_layer_ids = []
         for layer in layers:
+            all_layer_ids.append(layer.id)
             layer_data = stub.read(layer, layers_meta)
 
             # Skip layers without metadata.
@@ -59,3 +65,33 @@ class CollectInstances(pyblish.api.ContextPlugin):
         if len(instance_names) != len(set(instance_names)):
             self.log.warning("Duplicate instances found. " +
                              "Remove unwanted via SubsetManager")
+
+        if len(instance_names) == 0 and self.flatten_subset_template:
+            project_name = context.data["projectEntity"]["name"]
+            variants = get_project_settings(project_name).get(
+                "photoshop", {}).get(
+                "create", {}).get(
+                "CreateImage", {}).get(
+                "defaults", [''])
+            family = "image"
+            task_name = api.Session["AVALON_TASK"]
+            asset_name = context.data["assetEntity"]["name"]
+
+            fill_pairs = {
+                "variant": variants[0],
+                "family": family,
+                "task": task_name
+            }
+
+            subset = self.flatten_subset_template.format(
+                **prepare_template_data(fill_pairs))
+
+            instance = context.create_instance(subset)
+            instance.data["family"] = family
+            instance.data["asset"] = asset_name
+            instance.data["subset"] = subset
+            instance.data["ids"] = all_layer_ids
+            instance.data["families"] = self.families_mapping[family]
+            instance.data["publish"] = True
+
+            self.log.info("flatten instance: {} ".format(instance.data))