Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2025-12-26 22:02:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

use query functions in store thumbnails to avalon

Browse source
This commit is contained in:
Jakub Trllo 2022-06-14 16:32:55 +02:00
parent 38f058fca0
commit 46e22241bd
1 changed files with 30 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

56
openpype/modules/ftrack/event_handlers_user/action_store_thumbnails_to_avalon.py
View file

@ -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