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

fix create write [subset, families]

Browse source
This commit is contained in:
Jakub Jezek 2018-11-30 16:43:54 +01:00
parent 40a5a4a065
commit c080823a44
4 changed files with 124 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

29
pype/nuke/lib.py
View file

@ -1,5 +1,4 @@
import sys
import os
from collections import OrderedDict
from pprint import pprint
from avalon.vendor.Qt import QtGui
@ -27,7 +26,7 @@ def format_anatomy(data):
"VERSION": pype.get_version_from_workfile(file)
})
log.info("format_anatomy:anatomy: {}".format(anatomy))
# log.info("format_anatomy:anatomy: {}".format(anatomy))
return anatomy.format(data)
@ -35,19 +34,19 @@ def script_name():
return nuke.root().knob('name').value()
def create_write_node(name, avalon_data, data_templates):
def create_write_node(name, data):
from .templates import (
get_dataflow,
get_colorspace
)
nuke_dataflow_writes = get_dataflow(**data_templates)
nuke_colorspace_writes = get_colorspace(**data_templates)
nuke_dataflow_writes = get_dataflow(**data)
nuke_colorspace_writes = get_colorspace(**data)
try:
anatomy_filled = format_anatomy({
"subset": avalon_data["subset"],
"asset": avalon_data["asset"],
"subset": data["avalon"]["subset"],
"asset": data["avalon"]["asset"],
"task": pype.get_task(),
"family": avalon_data["family"],
"family": data["avalon"]["family"],
"project": {"name": pype.get_project_name(),
"code": pype.get_project_code()},
"representation": nuke_dataflow_writes.file_type,
@ -57,28 +56,28 @@ def create_write_node(name, avalon_data, data_templates):
log.debug("anatomy_filled.render: {}".format(anatomy_filled.render))
data = OrderedDict({
_data = OrderedDict({
"file": str(anatomy_filled.render.path).replace("\\", "/")
})
# adding dataflow template
{data.update({k: v})
{_data.update({k: v})
for k, v in nuke_dataflow_writes.items()
if k not in ["id", "previous"]}
# adding dataflow template
{data.update({k: v})
{_data.update({k: v})
for k, v in nuke_colorspace_writes.items()}
data = avalon.nuke.lib.fix_data_for_node_create(data)
_data = avalon.nuke.lib.fix_data_for_node_create(_data)
log.debug(data)
log.debug(_data)
instance = avalon.nuke.lib.add_write_node(
name,
**data
**_data
)
instance = avalon.nuke.lib.imprint(instance, avalon_data)
instance = avalon.nuke.lib.imprint(instance, data["avalon"])
return instance

18
pype/nuke/templates.py
View file

@ -8,32 +8,34 @@ def get_anatomy(**kwarg):
def get_dataflow(**kwarg):
log.info(kwarg)
host = kwarg.get("host", "nuke")
cls = kwarg.get("cls", None)
family = kwarg.get("family", None)
cls = kwarg.get("class", None)
preset = kwarg.get("preset", None)
assert any([host, cls]), log.error("nuke.templates.get_dataflow():"
"Missing mandatory kwargs `host`, `cls`")
nuke_dataflow = getattr(pype.Dataflow, str(host), None)
nuke_dataflow_node = getattr(nuke_dataflow.nodes, str(cls), None)
if family:
nuke_dataflow_node = getattr(nuke_dataflow_node, str(family), None)
if preset:
nuke_dataflow_node = getattr(nuke_dataflow_node, str(preset), None)
log.info("Dataflow: {}".format(nuke_dataflow_node))
return nuke_dataflow_node
def get_colorspace(**kwarg):
log.info(kwarg)
host = kwarg.get("host", "nuke")
cls = kwarg.get("cls", None)
family = kwarg.get("family", None)
cls = kwarg.get("class", None)
preset = kwarg.get("preset", None)
assert any([host, cls]), log.error("nuke.templates.get_colorspace():"
"Missing mandatory kwargs `host`, `cls`")
nuke_colorspace = getattr(pype.Colorspace, str(host), None)
nuke_colorspace_node = getattr(nuke_colorspace, str(cls), None)
if family:
nuke_colorspace_node = getattr(nuke_colorspace_node, str(family), None)
if preset:
nuke_colorspace_node = getattr(nuke_colorspace_node, str(preset), None)
log.info("Colorspace: {}".format(nuke_colorspace_node))
return nuke_colorspace_node