From 44cc3d40113d2110d0f274e9b2f1da61fe7db018 Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Wed, 12 May 2021 13:08:32 +0200
Subject: [PATCH] implemeted method that cares about removing single item

---
 .../project_manager/project_manager/model.py  | 104 ++++++++++++++++++
 1 file changed, 104 insertions(+)

diff --git a/openpype/tools/project_manager/project_manager/model.py b/openpype/tools/project_manager/project_manager/model.py
index c0a96058e3..cb51323975 100644
--- a/openpype/tools/project_manager/project_manager/model.py
+++ b/openpype/tools/project_manager/project_manager/model.py
@@ -436,6 +436,110 @@ class HierarchyModel(QtCore.QAbstractItemModel):
     def remove_indexes(self, indexes):
         items_by_id = {}
 
+    def _remove_item(self, item):
+        is_removed = item.data(None, REMOVED_ROLE)
+        if is_removed:
+            return
+
+        parent = item.parent()
+
+        all_descendants = collections.defaultdict(dict)
+        all_descendants[parent.id][item.id] = item
+
+        def _fill_children(_all_descendants, cur_item, parent_item=None):
+            if parent_item is not None:
+                _all_descendants[parent_item.id][cur_item.id] = cur_item
+
+            if isinstance(cur_item, TaskItem):
+                was_removed = cur_item.data(None, REMOVED_ROLE)
+                task_removed = True
+                if not was_removed and parent_item is not None:
+                    task_removed = parent_item.data(None, REMOVED_ROLE)
+                if not was_removed:
+                    cur_item.setData(None, task_removed, REMOVED_ROLE)
+                return task_removed
+
+            remove_item = cur_item.data(None, HIERARCHY_CHANGE_ABLE_ROLE)
+            for row in range(cur_item.rowCount()):
+                child_item = cur_item.child(row)
+                if not _fill_children(_all_descendants, child_item, cur_item):
+                    remove_item = False
+
+            if remove_item:
+                cur_item.setData(None, True, REMOVED_ROLE)
+            return remove_item
+
+        _fill_children(all_descendants, item)
+
+        modified_children = []
+        while all_descendants:
+            for parent_id in tuple(all_descendants.keys()):
+                children = all_descendants[parent_id]
+                if not children:
+                    all_descendants.pop(parent_id)
+                    continue
+
+                parent_children = {}
+                all_without_children = True
+                for child_id in tuple(children.keys()):
+                    if child_id in all_descendants:
+                        all_without_children = False
+                        break
+                    parent_children[child_id] = children[child_id]
+
+                if not all_without_children:
+                    continue
+
+                parent_item = self._items_by_id[parent_id]
+                row_ranges = []
+                start_row = end_row = None
+                chilren_by_row = {}
+                for row in range(parent_item.rowCount()):
+                    child_item = parent_item.child(row)
+                    child_id = child_item.id
+                    if child_id not in children:
+                        continue
+
+                    chilren_by_row[row] = child_item
+                    children.pop(child_item.id)
+
+                    remove_item = child_item.data(None, REMOVED_ROLE)
+                    if not remove_item or not child_item.is_new:
+                        modified_children.append(child_item)
+                        if end_row is not None:
+                            row_ranges.append((start_row, end_row))
+                        start_row = end_row = None
+                        continue
+
+                    end_row = row
+                    if start_row is None:
+                        start_row = row
+
+                if end_row is not None:
+                    row_ranges.append((start_row, end_row))
+
+                parent_index = None
+                for start, end in row_ranges:
+                    if parent_index is None:
+                        parent_index = self.index_for_item(parent_item)
+
+                    self.beginRemoveRows(parent_index, start, end)
+
+                    for idx in range(start, end + 1):
+                        child_item = chilren_by_row[idx]
+                        # Force name validation
+                        if isinstance(child_item, AssetItem):
+                            self._rename_asset(child_item, None)
+                        child_item.set_parent(None)
+                        self._items_by_id.pop(child_item.id)
+
+                    self.endRemoveRows()
+
+        for item in modified_children:
+            s_index = self.index_for_item(item)
+            e_index = self.index_for_item(item, column=self.columns_len - 1)
+            self.dataChanged.emit(s_index, e_index, [QtCore.Qt.BackgroundRole])
+
     def _rename_asset(self, asset_item, new_name):
         if not isinstance(asset_item, AssetItem):
             return