From 8c8a4e11ab9a17c2319569d64673f20cc15c9773 Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakub@pype.club>
Date: Wed, 18 Mar 2020 19:09:05 +0100
Subject: [PATCH] feat(nuke): prerender family clarifying

---
 pype/plugins/nuke/publish/collect_writes.py | 22 ++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/pype/plugins/nuke/publish/collect_writes.py b/pype/plugins/nuke/publish/collect_writes.py
index b1213199f5..aa5f825a98 100644
--- a/pype/plugins/nuke/publish/collect_writes.py
+++ b/pype/plugins/nuke/publish/collect_writes.py
@@ -66,7 +66,6 @@ class CollectNukeWrites(pyblish.api.InstancePlugin):
             int(last_frame)
         )
 
-        if 'render' in instance.data['families']:
         if [fm for fm in instance.data['families']
                 if fm in ["render", "prerender"]]:
             if "representations" not in instance.data:
@@ -96,7 +95,8 @@ class CollectNukeWrites(pyblish.api.InstancePlugin):
                     # this will only run if slate frame is not already
                     # rendered from previews publishes
                     if "slate" in instance.data["families"] \
-                            and (frame_length == collected_frames_len):
+                            and (frame_length == collected_frames_len) \
+                            and ("prerender" not in instance.data["families"]):
                         frame_slate_str = "%0{}d".format(
                             len(str(last_frame))) % (first_frame - 1)
                         slate_frame = collected_frames[0].replace(
@@ -105,6 +105,8 @@ class CollectNukeWrites(pyblish.api.InstancePlugin):
 
                 representation['files'] = collected_frames
                 instance.data["representations"].append(representation)
+                if "render" not in instance.data["families"]:
+                    instance.data["families"].append("render")
             except Exception:
                 instance.data["representations"].append(representation)
                 self.log.debug("couldn't collect frames: {}".format(label))
@@ -144,5 +146,19 @@ class CollectNukeWrites(pyblish.api.InstancePlugin):
             "deadlineChunkSize": deadlineChunkSize,
             "deadlinePriority": deadlinePriority
         })
-
+        self.log.debug("families: {}".format(families))
+        if "prerender" in families:
+            _families = list()
+            for fm in families:
+                if fm in _families:
+                    continue
+                if "render" in fm:
+                    if "prerender" in fm:
+                        continue
+                    _families.append(fm)
+            instance.data.update({
+                "family": "prerender",
+                "families": _families
+                })
+            self.log.debug("_families: {}".format(_families))
         self.log.debug("instance.data: {}".format(instance.data))