Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2026-01-02 00:44:52 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

integrate thumbnails now works

Browse source
This commit is contained in:
iLLiCiTiT 2019-12-05 19:31:58 +01:00
parent 9228536f52
commit 7b2e5f1b9e
1 changed files with 24 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

48
pype/plugins/global/publish/integrate_thumbnail.py
View file

@ -8,7 +8,7 @@ import six
import pyblish.api
from bson.objectid import ObjectId
from avalon import api, dbio
from avalon import api, io
class IntegrateThumbnails(pyblish.api.InstancePlugin):
@ -19,7 +19,7 @@ class IntegrateThumbnails(pyblish.api.InstancePlugin):
families = ["review"]
def process(self, instance):
repre_ids = instance.get("published_representation_ids")
repre_ids = instance.data.get("published_representation_ids")
if not repre_ids:
self.log.debug(
"There are not published representation ids on the instance."
@ -30,27 +30,24 @@ class IntegrateThumbnails(pyblish.api.InstancePlugin):
anatomy = instance.context.data["anatomy"]
if "publish" not in anatomy.templates:
self.log.error("Anatomy does not have set publish key!")
return
raise AssertionError("Anatomy does not have set publish key!")
if "thumbnail" not in anatomy.templates["publish"]:
self.log.warning((
"There is not set \"thumbnail\" template for project {}"
raise AssertionError((
"There is not set \"thumbnail\" template for project \"{}\""
).format(project_name))
return
thumbnail_template = anatomy.templates["publish"]["thumbnail"]
dbio.install()
repres = dbio.find({
io.install()
repres = io.find({
"_id": {"$in": repre_ids},
"type": "representation"
})
if not repres:
self.log.debug((
raise AssertionError((
"There are not representations in database with ids {}"
).format(str(repre_ids)))
return
thumb_repre = None
for repre in repres:
@ -64,12 +61,13 @@ class IntegrateThumbnails(pyblish.api.InstancePlugin):
)
return
version = dbio.find_one({"_id": thumb_repre["parent"]})
version = io.find_one({"_id": thumb_repre["parent"]})
if not version:
self.log.warning("There does not exist version with id {}".format(
str(thumb_repre["parent"])
))
return
raise AssertionError(
"There does not exist version with id {}".format(
str(thumb_repre["parent"])
)
)
# Get full path to thumbnail file from representation
src_full_path = os.path.normpath(thumb_repre["data"]["path"])
@ -79,25 +77,27 @@ class IntegrateThumbnails(pyblish.api.InstancePlugin):
))
return
filename, file_extension = os.path.splitext(src_full_path)
# Create id for mongo entity now to fill anatomy template
thumbnail_id = ObjectId()
# Prepare anatomy template fill data
template_data = copy.deepcopy(thumb_repre["context"])
template_data["_id"] = str(thumbnail_id)
template_data["thumbnail_root"] = os.environ.get(
"AVALON_THUMBNAIL_ROOT"
)
template_data.update({
"_id": str(thumbnail_id),
"thumbnail_root": os.environ.get("AVALON_THUMBNAIL_ROOT"),
"ext": file_extension,
"thumbnail_type": "thumbnail"
})
anatomy_filled = anatomy.format(template_data)
final_path = anatomy_filled.get("publish", {}).get("thumbnail")
if not final_path:
self.log.warning((
raise AssertionError((
"Anatomy template was not filled with entered data"
"\nTemplate: {} "
"\nData: {}"
).format(thumbnail_template, str(template_data)))
return
dst_full_path = os.path.normpath(final_path)
self.log.debug(
@ -127,12 +127,12 @@ class IntegrateThumbnails(pyblish.api.InstancePlugin):
}
}
# Create thumbnail entity
dbio.insert_one(thumbnail_entity)
io.insert_one(thumbnail_entity)
self.log.debug(
"Creating entity in database {}".format(str(thumbnail_entity))
)
# Set thumbnail id for version
dbio.update_one(
io.update_many(
{"_id": version["_id"]},
{"$set": {"data.thumbnail_id": thumbnail_id}}
)