From e52310e63825be5bcc85bb943ad71b9420a1c3bc Mon Sep 17 00:00:00 2001
From: aardschok <wikoreman@gmail.com>
Date: Tue, 25 Jul 2017 17:50:48 +0200
Subject: [PATCH] added get_reference_node function

---
 colorbleed/maya/lib.py                        | 27 ++++++++++++-------
 .../plugins/publish/post_look_integrate.py    |  0
 2 files changed, 18 insertions(+), 9 deletions(-)
 create mode 100644 colorbleed/plugins/publish/post_look_integrate.py

diff --git a/colorbleed/maya/lib.py b/colorbleed/maya/lib.py
index f1530fd109..686a08c6fe 100644
--- a/colorbleed/maya/lib.py
+++ b/colorbleed/maya/lib.py
@@ -665,6 +665,21 @@ def get_representation_file(representation, template=TEMPLATE):
     return template.format(**context)
 
 
+def get_reference_node(path):
+    """
+    Get the reference node when the path is found being used in a reference
+    Args:
+        path (str):
+
+    Returns:
+
+    """
+    node = cmds.file(path, query=True, referenceNode=True)
+    reference_path = cmds.referenceQuery(path, filename=True)
+    if os.path.normpath(path) == os.path.normpath(reference_path):
+        return node
+
+
 def list_looks(asset_id):
     """Return all look subsets for the given asset
 
@@ -706,14 +721,8 @@ def assign_look_by_version(nodes, version_id):
     shader_filepath = get_representation_file(shader_file)
     shader_relation = get_representation_file(shader_relations)
 
-    try:
-        existing_reference = cmds.file(shader_filepath,
-                                       query=True,
-                                       referenceNode=True)
-    except RuntimeError as e:
-        if e.message.rstrip() != "Cannot find the scene file.":
-            raise
-
+    reference_node = get_reference_node(shader_filepath)
+    if reference_node is None:
         log.info("Loading lookdev for the first time..")
 
         # Define namespace
@@ -731,7 +740,7 @@ def assign_look_by_version(nodes, version_id):
                                      returnNewNodes=True)
     else:
         log.info("Reusing existing lookdev..")
-        shader_nodes = cmds.referenceQuery(existing_reference, nodes=True)
+        shader_nodes = cmds.referenceQuery(reference_node, nodes=True)
 
     # Assign relationships
     with open(shader_relation, "r") as f:
diff --git a/colorbleed/plugins/publish/post_look_integrate.py b/colorbleed/plugins/publish/post_look_integrate.py
new file mode 100644
index 0000000000..e69de29bb2