From 5432c2fa699ed0c1ebae49c64e8b4e08ca4dfadd Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakubjezek001@gmail.com>
Date: Tue, 21 Nov 2023 15:25:22 +0100
Subject: [PATCH] Global: adding `need_thumbnail` tag worklfow

extract review and extract review slate
---
 openpype/plugins/publish/extract_review.py    | 29 +++++++++++++++----
 .../plugins/publish/extract_review_slate.py   | 10 +++++--
 2 files changed, 30 insertions(+), 9 deletions(-)

diff --git a/openpype/plugins/publish/extract_review.py b/openpype/plugins/publish/extract_review.py
index cd0f78530a..8448c45e70 100644
--- a/openpype/plugins/publish/extract_review.py
+++ b/openpype/plugins/publish/extract_review.py
@@ -89,8 +89,18 @@ class ExtractReview(pyblish.api.InstancePlugin):
         # Make sure cleanup happens and pop representations with "delete" tag.
         for repre in tuple(instance.data["representations"]):
             tags = repre.get("tags") or []
-            if "delete" in tags and "thumbnail" not in tags:
-                instance.data["representations"].remove(repre)
+            # Representation is not marked to be deleted
+            if "delete" not in tags:
+                continue
+
+            # The representation can be used as thumbnail source
+            if "thumbnail" in tags or "need_thumbnail" in tags:
+                continue
+
+            self.log.debug(
+                "Removing representation: {}".format(repre)
+            )
+            instance.data["representations"].remove(repre)
 
     def _get_outputs_for_instance(self, instance):
         host_name = instance.context.data["hostName"]
@@ -321,19 +331,26 @@ class ExtractReview(pyblish.api.InstancePlugin):
 
             # Create copy of representation
             new_repre = copy.deepcopy(repre)
+            new_tags = new_repre.get("tags") or []
             # Make sure new representation has origin staging dir
             #   - this is because source representation may change
             #       it's staging dir because of ffmpeg conversion
             new_repre["stagingDir"] = src_repre_staging_dir
 
             # Remove "delete" tag from new repre if there is
-            if "delete" in new_repre["tags"]:
-                new_repre["tags"].remove("delete")
+            if "delete" in new_tags:
+                new_tags.remove("delete")
+
+            if "need_thumbnail" in new_tags:
+                new_tags.remove("need_thumbnail")
 
             # Add additional tags from output definition to representation
             for tag in output_def["tags"]:
-                if tag not in new_repre["tags"]:
-                    new_repre["tags"].append(tag)
+                if tag not in new_tags:
+                    new_tags.append(tag)
+
+            # Return tags to new representation
+            new_repre["tags"] = new_tags
 
             # Add burnin link from output definition to representation
             for burnin in output_def["burnins"]:
diff --git a/openpype/plugins/publish/extract_review_slate.py b/openpype/plugins/publish/extract_review_slate.py
index d89fbb90c4..4e3406d3f9 100644
--- a/openpype/plugins/publish/extract_review_slate.py
+++ b/openpype/plugins/publish/extract_review_slate.py
@@ -376,9 +376,13 @@ class ExtractReviewSlate(publish.Extractor):
 
         # Remove any representations tagged for deletion.
         for repre in inst_data.get("representations", []):
-            if "delete" in repre.get("tags", []):
-                self.log.debug("Removing representation: {}".format(repre))
-                inst_data["representations"].remove(repre)
+            tags = repre.get("tags", [])
+            if "delete" not in tags:
+                continue
+            if "need_thumbnail" in tags:
+                continue
+            self.log.debug("Removing representation: {}".format(repre))
+            inst_data["representations"].remove(repre)
 
         self.log.debug(inst_data["representations"])