From 4f8b1dfef7a807a3d7ff7323dcdfe9964ec29857 Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakub@orbi.tools>
Date: Fri, 2 Aug 2019 18:48:01 +0200
Subject: [PATCH] fix(nuke): didnt coult with handle_start, handle_end

---
 .../nuke/publish/collect_asset_info.py        |  2 ++
 pype/plugins/nuke/publish/collect_writes.py   | 32 ++++++++++++++++---
 2 files changed, 29 insertions(+), 5 deletions(-)

diff --git a/pype/plugins/nuke/publish/collect_asset_info.py b/pype/plugins/nuke/publish/collect_asset_info.py
index ae49c6e86f..500eaad8fb 100644
--- a/pype/plugins/nuke/publish/collect_asset_info.py
+++ b/pype/plugins/nuke/publish/collect_asset_info.py
@@ -19,3 +19,5 @@ class CollectAssetInfo(pyblish.api.ContextPlugin):
         self.log.info("asset_data: {}".format(asset_data))
 
         context.data['handles'] = int(asset_data["data"].get("handles", 0))
+        context.data['handle_start'] = int(asset_data["data"].get("handle_start", 0))
+        context.data['handle_end'] = int(asset_data["data"].get("handle_end", 0))
diff --git a/pype/plugins/nuke/publish/collect_writes.py b/pype/plugins/nuke/publish/collect_writes.py
index f98a3a0f7d..fb1fb37fd5 100644
--- a/pype/plugins/nuke/publish/collect_writes.py
+++ b/pype/plugins/nuke/publish/collect_writes.py
@@ -14,17 +14,24 @@ class CollectNukeWrites(pyblish.api.InstancePlugin):
     families = ["render", "render.local", "render.farm"]
 
     def process(self, instance):
-
         # if not instance.data["publish"]:
         #     continue
 
-        node = instance[0]
+        group = instance[0]
 
-        if node.Class() != "Write":
+        if group.Class() != "Group":
             return
 
         self.log.debug("checking instance: {}".format(instance))
 
+        group.begin()
+
+        for n in nuke.allNodes():
+            if n.Class() != "Write":
+                continue
+            node = n
+        group.end()
+
         # Determine defined file type
         ext = node["file_type"].value()
 
@@ -34,7 +41,9 @@ class CollectNukeWrites(pyblish.api.InstancePlugin):
             output_type = "mov"
 
         # Get frame range
-        handles = instance.context.data.get('handles', 0)
+        handles = instance.context.data['handles']
+        handle_start = instance.context.data['handle_start']
+        handle_end = instance.context.data['handle_end']
         first_frame = int(nuke.root()["first_frame"].getValue())
         last_frame = int(nuke.root()["last_frame"].getValue())
 
@@ -85,7 +94,20 @@ class CollectNukeWrites(pyblish.api.InstancePlugin):
         if 'render.local' in instance.data['families']:
             instance.data['families'].append('ftrack')
 
+        # Add version data to instance
+        version_data = {
+            "handles": handle_start,
+            "handleStart": handle_start,
+            "handleEnd": handle_end,
+            "version": int(version),
+            "colorspace":  node["colorspace"].value(),
+            "families": [instance.data["family"]],
+            "subset": instance.data["subset"],
+            "fps": instance.context.data["fps"]
+        }
+
         instance.data.update({
+            "versionData": version_data,
             "path": path,
             "outputDir": output_dir,
             "ext": ext,
@@ -96,5 +118,5 @@ class CollectNukeWrites(pyblish.api.InstancePlugin):
             "outputType": output_type,
             "colorspace": node["colorspace"].value(),
         })
-
+        instance[0] = node
         self.log.debug("instance.data: {}".format(instance.data))