pass blacklisted models to find available language because they are i…

…rellevant
opengisch · Nov 5, 2024 · bef85ea · bef85ea
1 parent 4b06337
commit bef85ea
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 12 deletions.
diff --git a/QgisModelBaker/gui/workflow_wizard/project_creation_page.py b/QgisModelBaker/gui/workflow_wizard/project_creation_page.py
@@ -43,7 +43,7 @@
 from QgisModelBaker.libs.modelbaker.utils.qt_utils import make_file_selector
 from QgisModelBaker.utils import gui_utils
 from QgisModelBaker.utils.globals import CATALOGUE_DATASETNAME, displayLanguages
-from QgisModelBaker.utils.gui_utils import TRANSFERFILE_MODELS_BLACKLIST, LogLevel
+from QgisModelBaker.utils.gui_utils import MODELS_BLACKLIST, LogLevel
 
 PAGE_UI = gui_utils.get_ui_class("workflow_wizard/project_creation.ui")
 
@@ -244,8 +244,17 @@ def _update_translation_combo(self):
         self.translation_combo.clear()
 
         if self.db_connector:
-            for lang in self.db_connector.get_available_languages():
-                self.translation_combo.addItem(displayLanguages.get(lang, lang), lang)
+            available_languages = self.db_connector.get_available_languages(
+                MODELS_BLACKLIST
+            )
+            if len(available_languages) > 1:
+                for lang in available_languages:
+                    self.translation_combo.addItem(
+                        displayLanguages.get(lang, lang), lang
+                    )
+                self.translation_combo.setEnabled(True)
+            else:
+                self.translation_combo.setEnabled(False)
 
         self.translation_combo.addItem(self.tr("Original model language"), "__")
 
@@ -725,11 +734,7 @@ def _modelnames(self):
         for db_model in db_models:
             for modelname in regex.split(db_model["modelname"]):
                 name = modelname.strip()
-                if (
-                    name
-                    and name not in TRANSFERFILE_MODELS_BLACKLIST
-                    and name not in modelnames
-                ):
+                if name and name not in MODELS_BLACKLIST and name not in modelnames:
                     modelnames.append(name)
         return modelnames
 

diff --git a/QgisModelBaker/utils/gui_utils.py b/QgisModelBaker/utils/gui_utils.py
@@ -65,7 +65,7 @@ class DropMode(Enum):
     "XLSX",
 ]
 
-TRANSFERFILE_MODELS_BLACKLIST = [
+MODELS_BLACKLIST = [
     "CHBaseEx_MapCatalogue_V1",
     "CHBaseEx_WaterNet_V1",
     "CHBaseEx_Sewage_V1",
@@ -112,6 +112,7 @@ class DropMode(Enum):
     "StandardSymbology",
     "Time",
     "Units",
+    "",
 ]
 
 # style
@@ -714,8 +715,7 @@ def _transfer_file_models(self, data_file_path):
                     for sub_element in element:
                         if (
                             "NAME" in sub_element.attrib
-                            and sub_element.attrib["NAME"]
-                            not in TRANSFERFILE_MODELS_BLACKLIST
+                            and sub_element.attrib["NAME"] not in MODELS_BLACKLIST
                         ):
                             model = {}
                             model["name"] = sub_element.attrib["NAME"]
@@ -1049,7 +1049,7 @@ def refresh_model(self, db_connectors=[]):
                         name = modelname.strip()
                         if (
                             name
-                            and name not in TRANSFERFILE_MODELS_BLACKLIST
+                            and name not in MODELS_BLACKLIST
                             and name not in modelnames
                         ):
                             modelnames.append(name)