From e6abb640d732f6e4e267c172a4ebc1f14aa464bf Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Fri, 17 Sep 2021 16:22:25 +0200 Subject: [PATCH] added ability to return families of current subsets --- openpype/tools/loader/app.py | 2 ++ openpype/tools/loader/model.py | 3 +++ openpype/tools/loader/widgets.py | 3 +++ 3 files changed, 8 insertions(+) diff --git a/openpype/tools/loader/app.py b/openpype/tools/loader/app.py index 5cb0bf41a9..f36248b0c0 100644 --- a/openpype/tools/loader/app.py +++ b/openpype/tools/loader/app.py @@ -221,6 +221,8 @@ class LoaderWidow(QtWidgets.QDialog): familis_widget = self.data["widgets"]["families"] subsets_widget.set_loading_state(loading=False, empty=not has_item) + families = subsets_widget.get_subsets_families() + familis_widget.set_enabled_families(families) def _on_load_end(self): # Delay hiding as click events happened during loading should be diff --git a/openpype/tools/loader/model.py b/openpype/tools/loader/model.py index 1668fc4a27..0ad8e88593 100644 --- a/openpype/tools/loader/model.py +++ b/openpype/tools/loader/model.py @@ -190,6 +190,9 @@ class SubsetsModel(TreeModel, BaseRepresentationModel): self._grouping = state self.on_doc_fetched() + def get_subsets_families(self): + return self._doc_payload.get("subset_families") or set() + def setData(self, index, value, role=QtCore.Qt.EditRole): # Trigger additional edit when `version` column changed # because it also updates the information in other columns diff --git a/openpype/tools/loader/widgets.py b/openpype/tools/loader/widgets.py index 5a04cbac8f..22d5f8ec3a 100644 --- a/openpype/tools/loader/widgets.py +++ b/openpype/tools/loader/widgets.py @@ -247,6 +247,9 @@ class SubsetWidget(QtWidgets.QWidget): self.model.refresh() + def get_subsets_families(self): + return self.model.get_subsets_families() + def set_family_filters(self, families): self.family_proxy.setFamiliesFilter(families)