From 5eaf66c55cba67638571079f65ce2acd555cb8b1 Mon Sep 17 00:00:00 2001
From: aardschok <wikoreman@gmail.com>
Date: Wed, 26 Jul 2017 10:51:28 +0200
Subject: [PATCH] added post-integrate plugin - temp

---
 colorbleed/plugins/publish/integrate_asset.py | 24 +----------
 .../plugins/publish/post_look_integrate.py    | 40 +++++++++++++++++++
 2 files changed, 42 insertions(+), 22 deletions(-)

diff --git a/colorbleed/plugins/publish/integrate_asset.py b/colorbleed/plugins/publish/integrate_asset.py
index 7db26fdaab..83678f7242 100644
--- a/colorbleed/plugins/publish/integrate_asset.py
+++ b/colorbleed/plugins/publish/integrate_asset.py
@@ -47,35 +47,15 @@ class IntegrateAsset(pyblish.api.InstancePlugin):
         # get needed data
         traffic = instance.data["traffic"]
         representations = instance.data["representations"]
-        version_folder = instance.data["versionFolder"]
-        family = instance.data["family"]
-        resources = instance.data("resources", [])
 
         self.log.info("Registering {} items".format(len(representations)))
         io.insert_many(representations)
 
         # moving files
-        for src_dest in traffic:
-            src, dest = src_dest
+        for src, dest in traffic:
             self.log.info("Copying file .. {} -> {}".format(src, dest))
             self.copy_file(src, dest)
 
-        if family == "colorbleed.texture":
-            try:
-                lib.remap_resource_nodes(resources, folder=version_folder)
-            except Exception as e:
-                self.log.error(e)
-
-        if family == "colorbleed.lookdev":
-            try:
-                tmp_dir = lib.maya_temp_folder()
-                resource_file = os.path.join(tmp_dir, "resources.json")
-                with open(resource_file, "r") as f:
-                    resources = json.load(f)
-                lib.remap_resource_nodes(resources)
-            except Exception as e:
-                self.log.error(e)
-
         self.log.info("Removing temporary files and folders ...")
         stagingdir = instance.data["stagingDir"]
         shutil.rmtree(stagingdir)
@@ -100,4 +80,4 @@ class IntegrateAsset(pyblish.api.InstancePlugin):
                 self.log.critical("An unexpected error occurred.")
                 raise
 
-        shutil.copy(src, dst)
\ No newline at end of file
+        shutil.copy(src, dst)
diff --git a/colorbleed/plugins/publish/post_look_integrate.py b/colorbleed/plugins/publish/post_look_integrate.py
index e69de29bb2..e66828b3c8 100644
--- a/colorbleed/plugins/publish/post_look_integrate.py
+++ b/colorbleed/plugins/publish/post_look_integrate.py
@@ -0,0 +1,40 @@
+import json
+import os
+
+
+import colorbleed.maya.lib as lib
+
+import pyblish.api
+
+
+class IntegrateAsset(pyblish.api.InstancePlugin):
+    """Remap source paths for lookdev and textures
+
+    """
+
+    label = "Remap source paths"
+    order = pyblish.api.IntegratorOrder + 0.15
+    families = ["colorbleed.lookdev",
+                "colorbleed.texture"]
+
+    def process(self, instance):
+
+        family = instance.data['family']
+        resources = instance.data['resources']
+        version_folder = instance.data['versionFolder']
+
+        if family == "colorbleed.texture":
+            try:
+                lib.remap_resource_nodes(resources, folder=version_folder)
+            except Exception as e:
+                self.log.error(e)
+
+        if family == "colorbleed.lookdev":
+            try:
+                tmp_dir = lib.maya_temp_folder()
+                resource_file = os.path.join(tmp_dir, "resources.json")
+                with open(resource_file, "r") as f:
+                    resources = json.load(f)
+                lib.remap_resource_nodes(resources)
+            except Exception as e:
+                self.log.error(e)