mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-01 16:34:53 +01:00
148 lines
4.4 KiB
Python
148 lines
4.4 KiB
Python
import os
|
|
import copy
|
|
import logging
|
|
|
|
from openpype.client import get_project
|
|
from . import legacy_io
|
|
from .anatomy import Anatomy
|
|
from .plugin_discover import (
|
|
discover,
|
|
register_plugin,
|
|
register_plugin_path,
|
|
)
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def get_thumbnail_binary(thumbnail_entity, thumbnail_type, dbcon=None):
|
|
if not thumbnail_entity:
|
|
return
|
|
|
|
resolvers = discover_thumbnail_resolvers()
|
|
resolvers = sorted(resolvers, key=lambda cls: cls.priority)
|
|
if dbcon is None:
|
|
dbcon = legacy_io
|
|
|
|
for Resolver in resolvers:
|
|
available_types = Resolver.thumbnail_types
|
|
if (
|
|
thumbnail_type not in available_types
|
|
and "*" not in available_types
|
|
and (
|
|
isinstance(available_types, (list, tuple))
|
|
and len(available_types) == 0
|
|
)
|
|
):
|
|
continue
|
|
try:
|
|
instance = Resolver(dbcon)
|
|
result = instance.process(thumbnail_entity, thumbnail_type)
|
|
if result:
|
|
return result
|
|
|
|
except Exception:
|
|
log.warning("Resolver {0} failed durring process.".format(
|
|
Resolver.__class__.__name__, exc_info=True
|
|
))
|
|
|
|
|
|
class ThumbnailResolver(object):
|
|
"""Determine how to get data from thumbnail entity.
|
|
|
|
"priority" - determines the order of processing in `get_thumbnail_binary`,
|
|
lower number is processed earlier.
|
|
"thumbnail_types" - it is expected that thumbnails will be used in more
|
|
more than one level, there is only ["thumbnail"] type at the moment
|
|
of creating this docstring but it is expected to add "ico" and "full"
|
|
in future.
|
|
"""
|
|
|
|
priority = 100
|
|
thumbnail_types = ["*"]
|
|
|
|
def __init__(self, dbcon):
|
|
self._log = None
|
|
self.dbcon = dbcon
|
|
|
|
@property
|
|
def log(self):
|
|
if self._log is None:
|
|
self._log = logging.getLogger(self.__class__.__name__)
|
|
return self._log
|
|
|
|
def process(self, thumbnail_entity, thumbnail_type):
|
|
pass
|
|
|
|
|
|
class TemplateResolver(ThumbnailResolver):
|
|
priority = 90
|
|
|
|
def process(self, thumbnail_entity, thumbnail_type):
|
|
template = thumbnail_entity["data"].get("template")
|
|
if not template:
|
|
self.log.debug("Thumbnail entity does not have set template")
|
|
return
|
|
|
|
thumbnail_root_format_key = "{thumbnail_root}"
|
|
thumbnail_root = os.environ.get("AVALON_THUMBNAIL_ROOT") or ""
|
|
# Check if template require thumbnail root and if is avaiable
|
|
if thumbnail_root_format_key in template and not thumbnail_root:
|
|
return
|
|
|
|
project_name = self.dbcon.active_project()
|
|
project = get_project(project_name, fields=["name", "data.code"])
|
|
|
|
template_data = copy.deepcopy(
|
|
thumbnail_entity["data"].get("template_data") or {}
|
|
)
|
|
template_data.update({
|
|
"_id": str(thumbnail_entity["_id"]),
|
|
"thumbnail_type": thumbnail_type,
|
|
"thumbnail_root": thumbnail_root,
|
|
"project": {
|
|
"name": project["name"],
|
|
"code": project["data"].get("code")
|
|
},
|
|
})
|
|
# Add anatomy roots if is in template
|
|
if "{root" in template:
|
|
anatomy = Anatomy(project_name)
|
|
template_data["root"] = anatomy.roots
|
|
|
|
try:
|
|
filepath = os.path.normpath(template.format(**template_data))
|
|
except KeyError:
|
|
self.log.warning((
|
|
"Missing template data keys for template <{0}> || Data: {1}"
|
|
).format(template, str(template_data)))
|
|
return
|
|
|
|
if not os.path.exists(filepath):
|
|
self.log.warning("File does not exist \"{0}\"".format(filepath))
|
|
return
|
|
|
|
with open(filepath, "rb") as _file:
|
|
content = _file.read()
|
|
|
|
return content
|
|
|
|
|
|
class BinaryThumbnail(ThumbnailResolver):
|
|
def process(self, thumbnail_entity, thumbnail_type):
|
|
return thumbnail_entity["data"].get("binary_data")
|
|
|
|
|
|
# Thumbnail resolvers
|
|
def discover_thumbnail_resolvers():
|
|
return discover(ThumbnailResolver)
|
|
|
|
|
|
def register_thumbnail_resolver(plugin):
|
|
register_plugin(ThumbnailResolver, plugin)
|
|
|
|
|
|
def register_thumbnail_resolver_path(path):
|
|
register_plugin_path(ThumbnailResolver, path)
|
|
|
|
|
|
register_thumbnail_resolver(TemplateResolver)
|
|
register_thumbnail_resolver(BinaryThumbnail)
|