From eb8380aeb25ee6db25a3198c0a232193bf113e23 Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Fri, 30 Apr 2021 11:13:21 +0200
Subject: [PATCH] model expect project name in set_project method

---
 .../project_manager/project_manager/model.py   | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/openpype/tools/project_manager/project_manager/model.py b/openpype/tools/project_manager/project_manager/model.py
index 5f1f1d5d0a..9c307063c0 100644
--- a/openpype/tools/project_manager/project_manager/model.py
+++ b/openpype/tools/project_manager/project_manager/model.py
@@ -73,6 +73,7 @@ class HierarchyModel(QtCore.QAbstractItemModel):
 
     def __init__(self, dbcon, parent=None):
         super(HierarchyModel, self).__init__(parent)
+        self._current_project = None
         self._root_item = None
         self._items_by_id = {}
         self._asset_items_by_name = collections.defaultdict(list)
@@ -91,11 +92,22 @@ class HierarchyModel(QtCore.QAbstractItemModel):
     def _reset_root_item(self):
         self._root_item = RootItem(self)
 
-    def set_project(self, project_doc):
+    def set_project(self, project_name):
+        if self._current_project == project_name:
+            return
+
         self.clear()
 
-        item = ProjectItem(project_doc)
-        self.add_item(item)
+        self._current_project = project_name
+        if not project_name:
+            return
+
+        project_doc = self.dbcon.database[project_name].find_one({
+            "type": "project"
+        })
+        if project_doc:
+            item = ProjectItem(project_doc)
+            self.add_item(item)
 
     def rowCount(self, parent=None):
         if parent is None or not parent.isValid():