From 98548aabd2f3064c3e7a0923a38ebbe081471fbc Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Tue, 14 Apr 2020 20:27:59 +0200
Subject: [PATCH] it is possible to have set specific root instead of
 thumbnail_root in thumbnail template

---
 .../action_store_thumbnails_to_avalon.py      | 73 ++++++++++---------
 1 file changed, 38 insertions(+), 35 deletions(-)

diff --git a/pype/ftrack/actions/action_store_thumbnails_to_avalon.py b/pype/ftrack/actions/action_store_thumbnails_to_avalon.py
index 7adc36f4b5..a8fc8cb06f 100644
--- a/pype/ftrack/actions/action_store_thumbnails_to_avalon.py
+++ b/pype/ftrack/actions/action_store_thumbnails_to_avalon.py
@@ -54,41 +54,6 @@ class StoreThumbnailsToAvalon(BaseAction):
         })
         session.commit()
 
-        thumbnail_roots = os.environ.get(self.thumbnail_key)
-        if not thumbnail_roots:
-            msg = "`{}` environment is not set".format(self.thumbnail_key)
-
-            action_job["status"] = "failed"
-            session.commit()
-
-            self.log.warning(msg)
-
-            return {
-                "success": False,
-                "message": msg
-            }
-
-        existing_thumbnail_root = None
-        for path in thumbnail_roots.split(os.pathsep):
-            if os.path.exists(path):
-                existing_thumbnail_root = path
-                break
-
-        if existing_thumbnail_root is None:
-            msg = (
-                "Can't access paths, set in `{}` ({})"
-            ).format(self.thumbnail_key, thumbnail_roots)
-
-            action_job["status"] = "failed"
-            session.commit()
-
-            self.log.warning(msg)
-
-            return {
-                "success": False,
-                "message": msg
-            }
-
         project = get_project_from_entity(entities[0])
         project_name = project["full_name"]
         anatomy = Anatomy(project_name)
@@ -122,6 +87,44 @@ class StoreThumbnailsToAvalon(BaseAction):
                 "message": msg
             }
 
+        thumbnail_roots = os.environ.get(self.thumbnail_key)
+        if (
+            "{thumbnail_root}" in anatomy.templates["publish"]["thumbnail"]
+            and not thumbnail_roots
+        ):
+            msg = "`{}` environment is not set".format(self.thumbnail_key)
+
+            action_job["status"] = "failed"
+            session.commit()
+
+            self.log.warning(msg)
+
+            return {
+                "success": False,
+                "message": msg
+            }
+
+        existing_thumbnail_root = None
+        for path in thumbnail_roots.split(os.pathsep):
+            if os.path.exists(path):
+                existing_thumbnail_root = path
+                break
+
+        if existing_thumbnail_root is None:
+            msg = (
+                "Can't access paths, set in `{}` ({})"
+            ).format(self.thumbnail_key, thumbnail_roots)
+
+            action_job["status"] = "failed"
+            session.commit()
+
+            self.log.warning(msg)
+
+            return {
+                "success": False,
+                "message": msg
+            }
+
         example_template_data = {
             "_id": "ID",
             "thumbnail_root": "THUBMNAIL_ROOT",