got rid of 'clear' argument in refresh method of assets widget

This commit is contained in:
iLLiCiTiT 2022-01-04 10:57:59 +01:00
parent 3ca931097d
commit 03fa4e7443

View file

@ -306,6 +306,8 @@ class AssetModel(QtGui.QStandardItemModel):
self._items_with_color_by_id = {}
self._items_by_asset_id = {}
self._last_project_name = None
@property
def refreshing(self):
return self._refreshing
@ -347,7 +349,7 @@ class AssetModel(QtGui.QStandardItemModel):
return self.get_indexes_by_asset_ids(asset_ids)
def refresh(self, force=False, clear=False):
def refresh(self, force=False):
"""Refresh the data for the model.
Args:
@ -360,7 +362,13 @@ class AssetModel(QtGui.QStandardItemModel):
return
self.stop_refresh()
if clear:
project_name = self.dbcon.Session.get("AVALON_PROJECT")
clear_model = False
if project_name != self._last_project_name:
clear_model = True
self._last_project_name = project_name
if clear_model:
self._clear_items()
# Fetch documents from mongo
@ -401,11 +409,18 @@ class AssetModel(QtGui.QStandardItemModel):
self._clear_items()
return
self._fill_assets(self._doc_payload)
self.refreshed.emit(bool(self._items_by_asset_id))
self._stop_fetch_thread()
def _fill_assets(self, asset_docs):
# Collect asset documents as needed
asset_ids = set()
asset_docs_by_id = {}
asset_ids_by_parents = collections.defaultdict(set)
for asset_doc in self._doc_payload:
for asset_doc in asset_docs:
asset_id = asset_doc["_id"]
asset_data = asset_doc.get("data") or {}
parent_id = asset_data.get("visualParent")
@ -511,10 +526,6 @@ class AssetModel(QtGui.QStandardItemModel):
except Exception:
pass
self.refreshed.emit(bool(self._items_by_asset_id))
self._stop_fetch_thread()
def _threaded_fetch(self):
asset_docs = self._fetch_asset_docs()
if not self._refreshing:
@ -652,12 +663,7 @@ class AssetsWidget(QtWidgets.QWidget):
return self._model.refreshing
def refresh(self):
project_name = self.dbcon.Session.get("AVALON_PROJECT")
clear_model = False
if project_name != self._last_project_name:
clear_model = True
self._last_project_name = project_name
self._refresh_model(clear_model)
self._refresh_model()
def stop_refresh(self):
self._model.stop_refresh()
@ -709,14 +715,14 @@ class AssetsWidget(QtWidgets.QWidget):
self._set_loading_state(loading=False, empty=not has_item)
self.refreshed.emit()
def _refresh_model(self, clear=False):
def _refresh_model(self):
# Store selection
self._set_loading_state(loading=True, empty=True)
# Trigger signal before refresh is called
self.refresh_triggered.emit()
# Refresh model
self._model.refresh(clear=clear)
self._model.refresh()
def _set_loading_state(self, loading, empty):
self._view.set_loading_state(loading, empty)