From ce95ee27de1262d527e67c645af9f954932112c4 Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Thu, 6 May 2021 20:49:01 +0200 Subject: [PATCH] implemented few necessary properties --- .../project_manager/project_manager/model.py | 29 +++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/openpype/tools/project_manager/project_manager/model.py b/openpype/tools/project_manager/project_manager/model.py index db016b38c9..d7abca47f0 100644 --- a/openpype/tools/project_manager/project_manager/model.py +++ b/openpype/tools/project_manager/project_manager/model.py @@ -924,9 +924,22 @@ class ProjectItem(BaseItem): } def __init__(self, project_doc): + self._mongo_id = project_doc["_id"] + data = self.data_from_doc(project_doc) super(ProjectItem, self).__init__(data) + @property + def project_id(self): + return self._mongo_id + + @property + def asset_id(self): + return None + + @property + def name(self): + return self._data["name"] @classmethod def data_from_doc(cls, project_doc): data = { @@ -997,9 +1010,25 @@ class AssetItem(BaseItem): } def __init__(self, asset_doc): + self.mongo_id = asset_doc.get("_id") + self._project_id = None + data = self.data_from_doc(asset_doc) super(AssetItem, self).__init__(data) + @property + def project_id(self): + if self._project_id is None: + self._project_id = self.parent().project_id + return self._project_id + + @property + def asset_id(self): + return self.mongo_id + + @property + def name(self): + return self._data["name"] @classmethod def data_from_doc(cls, asset_doc): data = {