From d0c3a3df758a0d8383dd3ee1c0e55f905877641e Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Fri, 14 May 2021 11:03:24 +0200 Subject: [PATCH] added buttons to add asset and task --- .../project_manager/project_manager/window.py | 42 +++++++++++++++++-- 1 file changed, 38 insertions(+), 4 deletions(-) diff --git a/openpype/tools/project_manager/project_manager/window.py b/openpype/tools/project_manager/project_manager/window.py index ae1ab70c70..df762bacce 100644 --- a/openpype/tools/project_manager/project_manager/window.py +++ b/openpype/tools/project_manager/project_manager/window.py @@ -18,11 +18,13 @@ class Window(QtWidgets.QWidget): dbcon = AvalonMongoDB() - # TOP Project selection - project_widget = QtWidgets.QWidget(self) + # Top part of window + top_part_widget = QtWidgets.QWidget(self) + + # Project selection + project_widget = QtWidgets.QWidget(top_part_widget) project_model = ProjectModel(dbcon) - project_combobox = QtWidgets.QComboBox(project_widget) project_combobox.setModel(project_model) project_combobox.setRootModelIndex(QtCore.QModelIndex()) @@ -38,6 +40,30 @@ class Window(QtWidgets.QWidget): project_layout.addWidget(refresh_projects_btn, 0) project_layout.addStretch(1) + # Helper buttons + helper_btns_widget = QtWidgets.QWidget(top_part_widget) + + helper_label = QtWidgets.QLabel("Add:", helper_btns_widget) + add_asset_btn = QtWidgets.QPushButton(helper_btns_widget) + add_asset_btn.setIcon(ResourceCache.get_icon("asset", "existing")) + add_asset_btn.setText("Asset") + add_task_btn = QtWidgets.QPushButton("Task", helper_btns_widget) + add_task_btn.setIcon(ResourceCache.get_icon("task", "existing")) + add_task_btn.setText("Task") + + helper_btns_layout = QtWidgets.QHBoxLayout(helper_btns_widget) + helper_btns_layout.setContentsMargins(0, 0, 0, 0) + helper_btns_layout.addWidget(helper_label) + helper_btns_layout.addWidget(add_asset_btn) + helper_btns_layout.addWidget(add_task_btn) + helper_btns_layout.addStretch(1) + + # Add widgets to top widget layout + top_part_layout = QtWidgets.QVBoxLayout(top_part_widget) + top_part_layout.setContentsMargins(0, 0, 0, 0) + top_part_layout.addWidget(project_widget) + top_part_layout.addWidget(helper_btns_widget) + hierarchy_model = HierarchyModel(dbcon) hierarchy_view = HierarchyView(dbcon, hierarchy_model, self) @@ -74,13 +100,15 @@ class Window(QtWidgets.QWidget): buttons_layout.addWidget(save_btn) main_layout = QtWidgets.QVBoxLayout(self) - main_layout.addWidget(project_widget) + main_layout.addWidget(top_part_widget) main_layout.addWidget(hierarchy_view) main_layout.addWidget(buttons_widget) refresh_projects_btn.clicked.connect(self._on_project_refresh) project_combobox.currentIndexChanged.connect(self._on_project_change) save_btn.clicked.connect(self._on_save_click) + add_asset_btn.clicked.connect(self._on_add_asset) + add_task_btn.clicked.connect(self._on_add_task) self.project_model = project_model self.project_combobox = project_combobox @@ -124,6 +152,12 @@ class Window(QtWidgets.QWidget): def _on_save_click(self): self.hierarchy_model.save() + def _on_add_asset(self): + self.hierarchy_view.add_asset() + + def _on_add_task(self): + self.hierarchy_view.add_task() + def show_message(self, message): # TODO add nicer message pop self.message_label.setText(message)