Add select/deselect all layers button and fix progress bar value error

GispoCoding · Jan 19, 2024 · cedb871 · cedb871
1 parent cf88cf8
commit cedb871
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 10 deletions.
diff --git a/nlsgpkgloader/nls_geopackage_loader.py b/nlsgpkgloader/nls_geopackage_loader.py
@@ -36,7 +36,7 @@
     QgsVectorLayer,
 )
 from qgis.gui import QgsFileWidget
-from qgis.PyQt.QtCore import QCoreApplication, QTimer, QTranslator, qVersion
+from qgis.PyQt.QtCore import QCoreApplication, Qt, QTimer, QTranslator, qVersion
 from qgis.PyQt.QtGui import QIcon
 from qgis.PyQt.QtWidgets import QAction, QListWidgetItem, QMessageBox
 from qgis.utils import iface
@@ -286,6 +286,10 @@ def run(self):
 
         self.municipalities_dialog.show()
 
+        self.municipalities_dialog.selectDeselectAll.clicked.connect(
+            self.select_deselect_all_layers
+        )
+
         result = self.municipalities_dialog.exec_()
         if result:
             self.fileName = self.municipalities_dialog.fileNameEdit.text().strip()
@@ -660,6 +664,21 @@ def load_layers(self):
 
         return True
 
+    def select_deselect_all_layers(self):
+        if self.municipalities_dialog.selectDeselectAll.isChecked():
+            items_to_select = []
+            for value in self.product_types.values():
+                items = self.municipalities_dialog.productListWidget.findItems(
+                    value, Qt.MatchExactly
+                )
+                if items:
+                    items_to_select.append(items[0])
+
+            for item in items_to_select:
+                item.setSelected(True)
+        else:
+            self.municipalities_dialog.productListWidget.clearSelection()
+
     def get_layers(self, root):
         layers = []
         for node in root.children():
@@ -800,7 +819,7 @@ def download_one_file(self):
 
         self.download_count += 1
         percentage = self.download_count / float(self.total_download_count) * 100.0
-        self.progress_dialog.progressBar.setValue(percentage)
+        self.progress_dialog.progressBar.setValue(int(percentage))
 
         if self.download_count == self.total_download_count:
             QgsMessageLog.logMessage("done downloading data", "NLSgpkgloader", 0)
@@ -835,7 +854,7 @@ def create_geopackage(self):
     def run_task(self, task):
         self.progress_dialog.label.setText(task.description())
         task.progressChanged.connect(
-            lambda: self.progress_dialog.progressBar.setValue(task.progress())
+            lambda: self.progress_dialog.progressBar.setValue(int(task.progress()))
         )
         QgsApplication.taskManager().addTask(task)
 

diff --git a/nlsgpkgloader/resources/ui/nls_geopackage_loader_dialog_municipality_selection.ui b/nlsgpkgloader/resources/ui/nls_geopackage_loader_dialog_municipality_selection.ui
@@ -6,8 +6,8 @@
    <rect>
     <x>0</x>
     <y>0</y>
-    <width>555</width>
-    <height>329</height>
+    <width>841</width>
+    <height>494</height>
    </rect>
   </property>
   <property name="windowTitle">
@@ -45,11 +45,34 @@
        <item>
         <layout class="QVBoxLayout" name="verticalLayout_2">
          <item>
-          <widget class="QLabel" name="label_2">
-           <property name="text">
-            <string>Choose layer(s):</string>
-           </property>
-          </widget>
+          <layout class="QHBoxLayout" name="horizontalLayout_4">
+           <item>
+            <widget class="QLabel" name="label_2">
+             <property name="sizePolicy">
+              <sizepolicy hsizetype="Preferred" vsizetype="Preferred">
+               <horstretch>0</horstretch>
+               <verstretch>0</verstretch>
+              </sizepolicy>
+             </property>
+             <property name="minimumSize">
+              <size>
+               <width>100</width>
+               <height>17</height>
+              </size>
+             </property>
+             <property name="text">
+              <string>Choose layer(s):</string>
+             </property>
+            </widget>
+           </item>
+           <item>
+            <widget class="QRadioButton" name="selectDeselectAll">
+             <property name="text">
+              <string>Select/deselect all</string>
+             </property>
+            </widget>
+           </item>
+          </layout>
          </item>
          <item>
           <widget class="QListWidget" name="productListWidget">