From 2ff21e39ec51db296a30c85a01006cf72fe70da8 Mon Sep 17 00:00:00 2001 From: Jakub Trllo <43494761+iLLiCiTiT@users.noreply.github.com> Date: Mon, 10 Jun 2024 16:03:11 +0200 Subject: [PATCH] add icons to simple folders widget --- client/ayon_core/tools/utils/folders_widget.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/client/ayon_core/tools/utils/folders_widget.py b/client/ayon_core/tools/utils/folders_widget.py index 2bfef1090b..3b53266c9f 100644 --- a/client/ayon_core/tools/utils/folders_widget.py +++ b/client/ayon_core/tools/utils/folders_widget.py @@ -5,6 +5,7 @@ from qtpy import QtWidgets, QtGui, QtCore from ayon_core.lib.events import QueuedEventSystem from ayon_core.style import get_default_entity_icon_color from ayon_core.tools.common_models import ( + ProjectsModel, HierarchyModel, HierarchyExpectedSelection, ) @@ -687,6 +688,7 @@ class SimpleSelectionModel(object): class SimpleFoldersController(object): def __init__(self): self._event_system = self._create_event_system() + self._projects_model = ProjectsModel(self) self._hierarchy_model = HierarchyModel(self) self._selection_model = SimpleSelectionModel(self) self._expected_selection = HierarchyExpectedSelection( @@ -707,6 +709,11 @@ class SimpleFoldersController(object): def get_folder_items(self, project_name, sender=None): return self._hierarchy_model.get_folder_items(project_name, sender) + def get_folder_type_items(self, project_name, sender=None): + return self._projects_model.get_folder_type_items( + project_name, sender + ) + def set_selected_project(self, project_name): self._selection_model.set_selected_project(project_name)