From 1dcdac7ae051cb359fb481a812027a553e4c79e1 Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Tue, 4 Feb 2020 15:18:11 +0100
Subject: [PATCH] asset_entity check moved back since nukestudio instances may
 have not set value

---
 pype/plugins/global/publish/integrate_new.py | 26 +++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/pype/plugins/global/publish/integrate_new.py b/pype/plugins/global/publish/integrate_new.py
index b71b5fb298..774a54ea7c 100644
--- a/pype/plugins/global/publish/integrate_new.py
+++ b/pype/plugins/global/publish/integrate_new.py
@@ -102,13 +102,37 @@ class IntegrateAssetNew(pyblish.api.InstancePlugin):
     def register(self, instance):
         # Required environment variables
         anatomy_data = instance.data["anatomyData"]
-        asset_entity = instance.data["assetEntity"]
         avalon_location = api.Session["AVALON_LOCATION"]
 
         io.install()
 
         context = instance.context
 
+        project_entity = instance.data["projectEntity"]
+
+        asset_name = instance.data["asset"]
+        asset_entity = instance.data.get("assetEntity")
+        if not asset_entity:
+            asset_entity = io.find_one({
+                "type": "asset",
+                "name": asset_name,
+                "parent": project_entity["_id"]
+            })
+
+            assert asset_entity, (
+                "No asset found by the name \"{0}\" in project \"{1}\""
+            ).format(asset_name, project_entity["name"])
+
+            instance.data["assetEntity"] = asset_entity
+
+            # update anatomy data with asset specific keys
+            # - name should already been set
+            hierarchy = ""
+            parents = asset_entity["data"]["parents"]
+            if parents:
+                hierarchy = "/".join(parents)
+            anatomy_data["hierarchy"] = hierarchy
+
         stagingdir = instance.data.get("stagingDir")
         if not stagingdir:
             self.log.info((