From 46e22241bd8d7d61b427f9d0f4fff930cc7549a2 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Tue, 14 Jun 2022 16:32:55 +0200
Subject: [PATCH] use query functions in store thumbnails to avalon

---
 .../action_store_thumbnails_to_avalon.py      | 56 ++++++++++---------
 1 file changed, 30 insertions(+), 26 deletions(-)

diff --git a/openpype/modules/ftrack/event_handlers_user/action_store_thumbnails_to_avalon.py b/openpype/modules/ftrack/event_handlers_user/action_store_thumbnails_to_avalon.py
index 62fdfa2bdd..d655dddcaf 100644
--- a/openpype/modules/ftrack/event_handlers_user/action_store_thumbnails_to_avalon.py
+++ b/openpype/modules/ftrack/event_handlers_user/action_store_thumbnails_to_avalon.py
@@ -5,6 +5,14 @@ import requests
 
 from bson.objectid import ObjectId
 
+from openpype.client import (
+    get_project,
+    get_asset_by_id,
+    get_assets,
+    get_subset_by_name,
+    get_version_by_name,
+    get_representations
+)
 from openpype_modules.ftrack.lib import BaseAction, statics_icon
 from openpype.api import Anatomy
 from openpype.pipeline import AvalonMongoDB
@@ -385,7 +393,7 @@ class StoreThumbnailsToAvalon(BaseAction):
 
         db_con.Session["AVALON_PROJECT"] = project_name
 
-        avalon_project = db_con.find_one({"type": "project"})
+        avalon_project = get_project(project_name)
         output["project"] = avalon_project
 
         if not avalon_project:
@@ -399,19 +407,17 @@ class StoreThumbnailsToAvalon(BaseAction):
         asset_mongo_id = parent["custom_attributes"].get(CUST_ATTR_ID_KEY)
         if asset_mongo_id:
             try:
-                asset_mongo_id = ObjectId(asset_mongo_id)
-                asset_ent = db_con.find_one({
-                    "type": "asset",
-                    "_id": asset_mongo_id
-                })
+                asset_ent = get_asset_by_id(project_name, asset_mongo_id)
             except Exception:
                 pass
 
         if not asset_ent:
-            asset_ent = db_con.find_one({
-                "type": "asset",
-                "data.ftrackId": parent["id"]
-            })
+            asset_docs = get_assets(project_name, asset_names=[parent["name"]])
+            for asset_doc in asset_docs:
+                ftrack_id = asset_doc.get("data", {}).get("ftrackId")
+                if ftrack_id == parent["id"]:
+                    asset_ent = asset_doc
+                    break
 
         output["asset"] = asset_ent
 
@@ -422,13 +428,11 @@ class StoreThumbnailsToAvalon(BaseAction):
             )
             return output
 
-        asset_mongo_id = asset_ent["_id"]
-
-        subset_ent = db_con.find_one({
-            "type": "subset",
-            "parent": asset_mongo_id,
-            "name": subset_name
-        })
+        subset_ent = get_subset_by_name(
+            project_name,
+            subset_name=subset_name,
+            asset_id=asset_ent["_id"]
+        )
 
         output["subset"] = subset_ent
 
@@ -439,11 +443,11 @@ class StoreThumbnailsToAvalon(BaseAction):
             ).format(subset_name, ent_path)
             return output
 
-        version_ent = db_con.find_one({
-            "type": "version",
-            "name": version,
-            "parent": subset_ent["_id"]
-        })
+        version_ent = get_version_by_name(
+            project_name,
+            version,
+            subset_ent["_id"]
+        )
 
         output["version"] = version_ent
 
@@ -454,10 +458,10 @@ class StoreThumbnailsToAvalon(BaseAction):
             ).format(version, subset_name, ent_path)
             return output
 
-        repre_ents = list(db_con.find({
-            "type": "representation",
-            "parent": version_ent["_id"]
-        }))
+        repre_ents = list(get_representations(
+            project_name,
+            version_ids=[version_ent["_id"]]
+        ))
 
         output["representations"] = repre_ents
         return output