From 48405b4da8600bb60cb4111d786e2594bb0d89a8 Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Wed, 17 Nov 2021 21:58:05 +0100 Subject: [PATCH] implemented single selection asset widget SingleSelectAssetsWidget --- openpype/tools/utils/assets_widget.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/openpype/tools/utils/assets_widget.py b/openpype/tools/utils/assets_widget.py index e8b19db9cf..6176f5cbe0 100644 --- a/openpype/tools/utils/assets_widget.py +++ b/openpype/tools/utils/assets_widget.py @@ -605,3 +605,21 @@ class AssetsWidget(QtWidgets.QWidget): self._view.expand(self._proxy.parent(index)) selection_model.select(index, mode) self._view.setCurrentIndex(valid_indexes[0]) + + +class SingleSelectAssetsWidget(AssetsWidget): + def get_selected_asset_id(self): + """Return the asset item of the current selection.""" + selection_model = self._view.selectionModel() + indexes = selection_model.selectedRows() + for index in indexes: + return index.data(ASSET_ID_ROLE) + return None + + def get_selected_asset_name(self): + """Return the asset document of the current selection.""" + selection_model = self._view.selectionModel() + indexes = selection_model.selectedRows() + for index in indexes: + return index.data(ASSET_NAME_ROLE) + return None