Merge pull request #1316 from marc-vdm/treeview_update_restore_state

Treeview restores open/close state on update

Author: mrvisscher

activity_browser/layouts/tabs/project_manager.py

@@ -307,8 +307,10 @@ def database_changed(self, database: bd.Database) -> None:
             self.mode_radio_list.show()
             self.mode_radio_tree.show()
         elif "ISIC rev.4 ecoinvent" in self.table.model._dataframe.columns:
-            # a treeview exists
+            # a treeview exists, update it
+            self.tree.get_expand_state()
             self.tree.model.setup_and_sync()
+            self.tree.set_expand_state()
 
             # make sure that the radio buttons are available
             self.mode_radio_list.show()

activity_browser/ui/tables/inventory.py

@@ -281,6 +281,7 @@ def __init__(self, parent=None, database_name=None):
         super().__init__(parent)
         self.database_name = database_name
         self.db_read_only = project_settings.db_is_readonly(self.database_name)
+        self.expand_state = []
         self.HEADERS = AB_metadata.get_existing_fields(self.HEADERS)
 
         # set drag ability
@@ -511,14 +512,25 @@ def find_levels(self, level=None) -> list:
             parent = parent.parent()
         return levels[::-1]
 
-    def expanded_list(self):
+    def get_expand_state(self) -> None:
+        """Store the expanded state of the tree.
+
+        Does not return anything but stores the expanded items in a self.expand_state."""
         it = self.model.iterator(None)
         expanded_items = []
         while it != None:
             if self.isExpanded(self.model.createIndex(it.row(), 0, it)):
                 expanded_items.append(self.build_path(it))
             it = self.model.iterator(it)
-        return expanded_items
+        self.expand_state = expanded_items
+
+    def set_expand_state(self) -> None:
+        """Sets any items in self.expand_state in the tree to expanded."""
+        it = self.model.iterator(None)
+        while it != None:
+            if self.build_path(it) in self.expand_state:
+                self.setExpanded(self.model.createIndex(it.row(), 0, it), True)
+            it = self.model.iterator(it)
 
     def build_path(self, iter):
         """Given an iterator of the TreeItem type build the path back to the