From 803936950af79776aa318c550279bda86c462482 Mon Sep 17 00:00:00 2001 From: Jakub Jezek Date: Thu, 7 Nov 2019 15:56:25 +0100 Subject: [PATCH] fix(nuke): colorspace presets didn't work with families --- pype/nuke/presets.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/pype/nuke/presets.py b/pype/nuke/presets.py index a3f62764c8..e0c12e2671 100644 --- a/pype/nuke/presets.py +++ b/pype/nuke/presets.py @@ -39,8 +39,9 @@ def get_node_dataflow_preset(**kwarg): nuke_dataflow_node = nuke_dataflow_node.get(str(preset), None) # omit < 2.0.0v - for family in families: - nuke_dataflow_node = nuke_dataflow_node.get(str(family), None) + if families: + for family in families: + nuke_dataflow_node = nuke_dataflow_node.get(str(family), None) log.info("Dataflow: {}".format(nuke_dataflow_node)) return nuke_dataflow_node @@ -52,14 +53,22 @@ def get_node_colorspace_preset(**kwarg): log.info(kwarg) host = kwarg.get("host", "nuke") cls = kwarg.get("class", None) - preset = kwarg.get("preset", None) + families = kwarg.get("families", []) + preset = kwarg.get("preset", None) # omit < 2.0.0v + assert any([host, cls]), log.error( "`{}`: Missing mandatory kwargs `host`, `cls`".format(__file__)) nuke_colorspace = get_colorspace_preset().get(str(host), None) nuke_colorspace_node = nuke_colorspace.get(str(cls), None) - if preset: + + if preset: # omit < 2.0.0v nuke_colorspace_node = nuke_colorspace_node.get(str(preset), None) + # omit < 2.0.0v + + if families: + for family in families: + nuke_colorspace_node = nuke_colorspace_node.get(str(family), None) log.info("Colorspace: {}".format(nuke_colorspace_node)) return nuke_colorspace_node