Try to fix updating a row after updating

pardusflatpakgui/updateallwindow.py

@@ -223,15 +223,16 @@ class UpdateAllWindow(object):
                 download_size_mib_str = ""
                 name = updated_ref.get_appdata_name()
 
-                tree_iter = self.TreeModel.get_model().get_iter_first()
+                tree_model = self.TreeModel().get_model()
+                tree_iter = tree_model.get_iter_first()
                 while tree_iter:
-                    real_name = self.TreeModel.get_value(tree_iter, 0)
-                    arch = self.TreeModel.get_value(tree_iter, 1)
-                    branch = self.TreeModel.get_value(tree_iter, 2)
+                    real_name = tree_model.get_value(tree_iter, 0)
+                    arch = tree_model.get_value(tree_iter, 1)
+                    branch = tree_model.get_value(tree_iter, 2)
                     if real_name == updated_ref_real_name and \
                        arch == updated_ref_arch and \
                        branch == updated_ref_branch:
-                        GLib.idle_add(self.TreeModel.set_row,
+                        GLib.idle_add(tree_model.set_row,
                                       tree_iter, [updated_ref_real_name,
                                                   updated_ref_arch,
                                                   updated_ref_branch,
@@ -242,9 +243,9 @@ class UpdateAllWindow(object):
                                       priority=GLib.PRIORITY_DEFAULT)
                         time.sleep(0.2)
 
-                        self.TreeModel.refilter()
+                        tree_model.refilter()
                         time.sleep(0.3)
-                    tree_iter = self.TreeModel.iter_next(tree_iter)
+                    tree_iter = tree_model.iter_next(tree_iter)
 
     def update_all_progress_callback_error(self, transaction, operation, error, details):
         ref_to_update_all = Flatpak.Ref.parse(operation.get_ref())