AddAssetItem are skipped on remove

This commit is contained in:
iLLiCiTiT 2021-05-13 16:19:42 +02:00
parent 2e70323166
commit 9e40246514

View file

@ -504,7 +504,7 @@ class HierarchyModel(QtCore.QAbstractItemModel):
processed_ids.add(item_id)
item = self._items_by_id[item_id]
if isinstance(item, (TaskItem, AssetItem)):
if isinstance(item, (TaskItem, AssetItem, AddAssetItem)):
items_by_id[item_id] = item
if not items_by_id:
@ -514,6 +514,9 @@ class HierarchyModel(QtCore.QAbstractItemModel):
self._remove_item(item)
def _remove_item(self, item):
if isinstance(item, AddAssetItem):
return
is_removed = item.data(REMOVED_ROLE)
if is_removed:
return
@ -544,6 +547,9 @@ class HierarchyModel(QtCore.QAbstractItemModel):
task_children.append(child_item)
continue
elif isinstance(child_item, AddAssetItem):
continue
if not _fill_children(_all_descendants, child_item, cur_item):
remove_item = False