From 7a9d28804428f5cfc5ac4099aa920ff8b2e4922c Mon Sep 17 00:00:00 2001 From: Toke Jepsen Date: Thu, 27 Jun 2019 18:54:14 +0100 Subject: [PATCH] Support collecting all items when nothing is selected. --- .../nukestudio/publish/collect_selection.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/pype/plugins/nukestudio/publish/collect_selection.py b/pype/plugins/nukestudio/publish/collect_selection.py index e22ea79a05..e87f9d03ec 100644 --- a/pype/plugins/nukestudio/publish/collect_selection.py +++ b/pype/plugins/nukestudio/publish/collect_selection.py @@ -2,6 +2,7 @@ import pyblish.api import hiero + class CollectSelection(pyblish.api.ContextPlugin): """Inject the selection in the context.""" @@ -9,7 +10,16 @@ class CollectSelection(pyblish.api.ContextPlugin): label = "Selection" def process(self, context): - selection = getattr(hiero, "selection") + selection = list(hiero.selection) self.log.debug("selection: {}".format(selection)) - context.data["selection"] = hiero.selection + + if not selection: + self.log.debug( + "Nothing is selected. Collecting all items from sequence " + "\"{}\"".format(hiero.ui.activeSequence()) + ) + for track in hiero.ui.activeSequence().items(): + selection.extend(track.items()) + + context.data["selection"] = selection