mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-01 08:24:53 +01:00
101 lines
3.2 KiB
Python
101 lines
3.2 KiB
Python
"""Collect Anatomy and global anatomy data.
|
|
|
|
Requires:
|
|
session -> AVALON_ASSET
|
|
context -> projectName
|
|
context -> asset
|
|
context -> task
|
|
|
|
Provides:
|
|
context -> projectEntity - Project document from database.
|
|
context -> assetEntity - Asset document from database only if 'asset' is
|
|
set in context.
|
|
"""
|
|
|
|
import pyblish.api
|
|
|
|
from openpype.client import get_project, get_asset_by_name
|
|
from openpype.pipeline import KnownPublishError
|
|
|
|
|
|
class CollectContextEntities(pyblish.api.ContextPlugin):
|
|
"""Collect entities into Context."""
|
|
|
|
order = pyblish.api.CollectorOrder - 0.1
|
|
label = "Collect Context Entities"
|
|
|
|
def process(self, context):
|
|
project_name = context.data["projectName"]
|
|
asset_name = context.data["asset"]
|
|
task_name = context.data["task"]
|
|
|
|
project_entity = get_project(project_name)
|
|
if not project_entity:
|
|
raise KnownPublishError(
|
|
"Project '{0}' was not found.".format(project_name)
|
|
)
|
|
self.log.debug("Collected Project \"{}\"".format(project_entity))
|
|
|
|
context.data["projectEntity"] = project_entity
|
|
|
|
if not asset_name:
|
|
self.log.info("Context is not set. Can't collect global data.")
|
|
return
|
|
|
|
asset_entity = get_asset_by_name(project_name, asset_name)
|
|
assert asset_entity, (
|
|
"No asset found by the name '{0}' in project '{1}'"
|
|
).format(asset_name, project_name)
|
|
|
|
self.log.debug("Collected Asset \"{}\"".format(asset_entity))
|
|
|
|
context.data["assetEntity"] = asset_entity
|
|
|
|
data = asset_entity['data']
|
|
|
|
# Task type
|
|
asset_tasks = data.get("tasks") or {}
|
|
task_info = asset_tasks.get(task_name) or {}
|
|
task_type = task_info.get("type")
|
|
context.data["taskType"] = task_type
|
|
|
|
frame_start = data.get("frameStart")
|
|
if frame_start is None:
|
|
frame_start = 1
|
|
self.log.warning("Missing frame start. Defaulting to 1.")
|
|
|
|
frame_end = data.get("frameEnd")
|
|
if frame_end is None:
|
|
frame_end = 2
|
|
self.log.warning("Missing frame end. Defaulting to 2.")
|
|
|
|
context.data["frameStart"] = frame_start
|
|
context.data["frameEnd"] = frame_end
|
|
|
|
handles = data.get("handles") or 0
|
|
handle_start = data.get("handleStart")
|
|
if handle_start is None:
|
|
handle_start = handles
|
|
self.log.info((
|
|
"Key \"handleStart\" is not set."
|
|
" Using value from \"handles\" key {}."
|
|
).format(handle_start))
|
|
|
|
handle_end = data.get("handleEnd")
|
|
if handle_end is None:
|
|
handle_end = handles
|
|
self.log.info((
|
|
"Key \"handleEnd\" is not set."
|
|
" Using value from \"handles\" key {}."
|
|
).format(handle_end))
|
|
|
|
context.data["handles"] = int(handles)
|
|
context.data["handleStart"] = int(handle_start)
|
|
context.data["handleEnd"] = int(handle_end)
|
|
|
|
frame_start_h = frame_start - context.data["handleStart"]
|
|
frame_end_h = frame_end + context.data["handleEnd"]
|
|
context.data["frameStartHandle"] = frame_start_h
|
|
context.data["frameEndHandle"] = frame_end_h
|
|
|
|
context.data["fps"] = data["fps"]
|