diff --git a/pype/plugins/nuke/publish/collect_asset_info.py b/pype/plugins/nuke/publish/collect_asset_info.py new file mode 100644 index 0000000000..ae49c6e86f --- /dev/null +++ b/pype/plugins/nuke/publish/collect_asset_info.py @@ -0,0 +1,21 @@ +import nuke +from avalon import api, io +import pyblish.api + + +class CollectAssetInfo(pyblish.api.ContextPlugin): + """Collect framerate.""" + + order = pyblish.api.CollectorOrder + label = "Collect Asset Info" + hosts = [ + "nuke", + "nukeassist" + ] + + def process(self, context): + asset_data = io.find_one({"type": "asset", + "name": api.Session["AVALON_ASSET"]}) + self.log.info("asset_data: {}".format(asset_data)) + + context.data['handles'] = int(asset_data["data"].get("handles", 0)) diff --git a/pype/plugins/nuke/publish/collect_instances.py b/pype/plugins/nuke/publish/collect_instances.py index 603dc89c4b..c8fad73c86 100644 --- a/pype/plugins/nuke/publish/collect_instances.py +++ b/pype/plugins/nuke/publish/collect_instances.py @@ -19,7 +19,7 @@ class CollectNukeInstances(pyblish.api.ContextPlugin): "name": api.Session["AVALON_ASSET"]}) # add handles into context - context.data['handles'] = int(asset_data["data"].get("handles", 0)) + context.data['handles'] = context.data['handles'] self.log.debug("asset_data: {}".format(asset_data["data"])) instances = [] diff --git a/pype/plugins/nuke/publish/collect_script.py b/pype/plugins/nuke/publish/collect_script.py index f0c917b449..d2585e4421 100644 --- a/pype/plugins/nuke/publish/collect_script.py +++ b/pype/plugins/nuke/publish/collect_script.py @@ -16,10 +16,6 @@ class CollectScript(pyblish.api.ContextPlugin): hosts = ['nuke'] def process(self, context): - asset_data = io.find_one({"type": "asset", - "name": api.Session["AVALON_ASSET"]}) - self.log.info("asset_data: {}".format(asset_data["data"])) - root = nuke.root() add_avalon_tab_knob(root) add_publish_knob(root) @@ -57,7 +53,7 @@ class CollectScript(pyblish.api.ContextPlugin): "publish": root.knob('publish').value(), "family": family, "representation": "nk", - "handles": int(asset_data["data"].get("handles", 0)), + "handles": context.data['handles'], "step": 1, "fps": int(root['fps'].value()), })