From f13d8c7cd9fe0032fdb0533fc526540de1f83853 Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Fri, 30 Apr 2021 11:09:15 +0200 Subject: [PATCH] dbconection is passed from window --- openpype/tools/project_manager/project_manager/model.py | 6 ++---- openpype/tools/project_manager/project_manager/window.py | 6 +++++- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/openpype/tools/project_manager/project_manager/model.py b/openpype/tools/project_manager/project_manager/model.py index 08621075c0..a0f75b5121 100644 --- a/openpype/tools/project_manager/project_manager/model.py +++ b/openpype/tools/project_manager/project_manager/model.py @@ -7,9 +7,7 @@ from .constants import ( DUPLICATED_ROLE ) -from avalon.api import AvalonMongoDB from avalon.vendor import qtawesome - from Qt import QtCore, QtGui @@ -73,12 +71,12 @@ class HierarchyModel(QtCore.QAbstractItemModel): ] index_moved = QtCore.Signal(QtCore.QModelIndex) - def __init__(self, parent=None): + def __init__(self, dbcon, parent=None): super(HierarchyModel, self).__init__(parent) self._root_item = None self._items_by_id = {} self._asset_items_by_name = collections.defaultdict(list) - self.dbcon = AvalonMongoDB() + self.dbcon = dbcon self._hierarchy_mode = True self._reset_root_item() diff --git a/openpype/tools/project_manager/project_manager/window.py b/openpype/tools/project_manager/project_manager/window.py index db5527b4ac..257850c014 100644 --- a/openpype/tools/project_manager/project_manager/window.py +++ b/openpype/tools/project_manager/project_manager/window.py @@ -6,12 +6,16 @@ from . import ( HierarchyView ) +from avalon.api import AvalonMongoDB + class Window(QtWidgets.QWidget): def __init__(self, parent=None): super(Window, self).__init__(parent) - model = HierarchyModel() + dbcon = AvalonMongoDB() + + model = HierarchyModel(dbcon) view = HierarchyView(model, self) view.setModel(model) _selection_model = HierarchySelectionModel()