dialogs/download.py: actually select all files when requested

Author: mathiascode

pynicotine/gtkgui/dialogs/download.py

@@ -229,7 +229,7 @@ def update_files(self, data, partial_files=True, select_all=False):
                     [
                         folder_name,
                         "",
-                        selected,
+                        select_all or selected,
                         username,
                         folder_path,
                         0,
@@ -245,7 +245,7 @@ def update_files(self, data, partial_files=True, select_all=False):
                 [
                     file_name,
                     human_size(size),
-                    selected,
+                    select_all or selected,
                     username,
                     folder_path,
                     size,
@@ -259,11 +259,12 @@ def update_files(self, data, partial_files=True, select_all=False):
 
             if selected:
                 self.initial_selected_iterators.add(iterator)
+            else:
+                self.initial_selected_iterators.discard(parent_iterator)
 
+            if select_all or selected:
                 self.total_selected_size += size
                 self.num_selected_files[username][folder_path] += 1
-            else:
-                self.initial_selected_iterators.discard(parent_iterator)
 
         self.update_title()
         self.tree_view.expand_all_rows()