From cde8c69cbabca6b9471cc30996b7a4a8000cfb06 Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Thu, 16 Dec 2021 21:55:12 +0100 Subject: [PATCH] modify tasks widget to use different models --- openpype/tools/launcher/widgets.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/openpype/tools/launcher/widgets.py b/openpype/tools/launcher/widgets.py index 33118e03be..4408219628 100644 --- a/openpype/tools/launcher/widgets.py +++ b/openpype/tools/launcher/widgets.py @@ -5,12 +5,15 @@ from Qt import QtWidgets, QtCore, QtGui from avalon.vendor import qtawesome from openpype.tools.flickcharm import FlickCharm +from openpype.tools.utils.tasks_widget import TasksWidget from .delegates import ActionDelegate from . import lib from .models import ( ActionModel, ProjectModel, + LauncherTaskModel, + LauncherTasksProxyModel ) from .constants import ( ACTION_ROLE, @@ -84,6 +87,21 @@ class ProjectBar(QtWidgets.QWidget): self._launcher_model.set_project_name(project_name) +class LauncherTaskWidget(TasksWidget): + def __init__(self, launcher_model, *args, **kwargs): + self._launcher_model = launcher_model + + super(LauncherTaskWidget, self).__init__(*args, **kwargs) + + def _create_source_model(self): + return LauncherTaskModel(self._launcher_model, self._dbcon) + + def _create_proxy_model(self, source_model): + proxy = LauncherTasksProxyModel(self._launcher_model) + proxy.setSourceModel(source_model) + return proxy + + class ActionBar(QtWidgets.QWidget): """Launcher interface"""