update mapper generator combobox when an image is loaded and a rule for it is also loaded and found

Author: mwinkens

src/backend/rule_manager.py

@@ -14,30 +14,51 @@ class RuleManager:
     def __init__(self):
         self._config: Dict[str, List[str]] = {}
         self._header: List[str] = []
+        self._loadedRules = False
 
     def _loadRuleFile(self, filename_base):
-        filename = f"{filename_base}.rules"
-        data_path = Path(ConfigManager.config()["data_path"])
-        if not data_path:
-            raise ValueError("No editor directory path known")
+        if not self._loadedRules:
+            filename = f"{filename_base}.rules"
+            data_path = Path(ConfigManager.config()["data_path"])
+            if not data_path:
+                raise ValueError("No editor directory path known")
 
-        automapper_path = data_path.joinpath(Path("editor/automap"))
-        if not automapper_path.exists():
-            automapper_path.mkdir()
+            automapper_path = data_path.joinpath(Path("editor/automap"))
+            if not automapper_path.exists():
+                automapper_path.mkdir()
 
-        full_file_path = automapper_path.joinpath(Path(filename))
+            full_file_path = automapper_path.joinpath(Path(filename))
 
-        # load file if it exists
-        if full_file_path.is_file():
-            self._readRuleFile(full_file_path)
+            # load file if it exists
+            if full_file_path.is_file():
+                self._readRuleFile(full_file_path)
 
-        # file doesn't exist, just to be explicit
-        else:
-            self._config = {}
-            self._header = []
+            # file doesn't exist, just to be explicit
+            else:
+                self._config = {}
+                self._header = []
+            self._loadedRules = True
+
+    def loadRules(self, filename):
+        if not self._loadedRules:
+            filename = RuleManager._getFileBase(filename)
+            self._loadRuleFile(filename)
 
     def saveRule(self, filename, rule_name):
+        filename = RuleManager._getFileBase(filename)
+        if not self._loadedRules:
+            self._loadRuleFile(filename)
+        self._config[rule_name] = []  # overwrite rules
+        self._config[rule_name] = RuleManager._createRulesFromTileHandler()
+        self._writeRuleFile(filename)
 
+    def getRules(self) -> List[str]:
+        if not self._loadedRules:
+            raise ValueError("Rules are not loaded yet")
+        return list(self._config.keys())
+
+    @staticmethod
+    def _getFileBase(filename):
         # remove mime type and check it if exists
         splits = filename.split(".")
         if len(splits) > 2:
@@ -46,11 +67,7 @@ def saveRule(self, filename, rule_name):
             if splits[1] != "rules":
                 raise ValueError(f"Unknown rule mime type '{splits[1]}'")
             filename = splits[0]
-
-        self._loadRuleFile(filename)
-        self._config[rule_name] = []  # overwrite rules
-        self._config[rule_name] = RuleManager._createRulesFromTileHandler()
-        self._writeRuleFile(filename)
+        return filename
 
     @staticmethod
     def _createRulesFromTileHandler():

src/config/app_state.py

@@ -8,6 +8,7 @@
 
 from src.backend.rule_manager import RuleManager
 from src.signals.signal_emitter import ApplicationStatusEmitter, ApplicationStatusEnum
+from pathlib import Path
 
 
 class AppState:
@@ -39,6 +40,7 @@ def ruleManager(cls):
     @classmethod
     def setImagePath(cls, image_path):
         cls.instance().main_image_path = image_path
+        cls.instance().rule_manager.loadRules(Path(image_path).stem)
         cls.instance().signal_emitter.application_status_signal.emit(ApplicationStatusEnum.IMAGE_LOADED, "")
 
     @classmethod

src/main_window.py

@@ -65,5 +65,6 @@ def showAbout(self):
     def statusUpdateReceived(self, status_type: ApplicationStatusEnum, message: str):
         if status_type == ApplicationStatusEnum.IMAGE_LOADED:
             self.mapper_generator.setEnabled(True)
+            self.mapper_generator.widget().rulesLoaded()
             # TODO set status dockwidget "Successfully loaded image"
         pass

src/widgets/widget_mapper_generator.py

@@ -11,7 +11,6 @@
 
 
 class MapperGeneratorWidget(QWidget):
-    # noinspection PyUnresolvedReferences
     def __init__(self, parent=None):
         super().__init__(parent)
 
@@ -130,3 +129,7 @@ def startDDNetCheck(self):
             return
         cmd = [client_path, map_name]
         subprocess.Popen(cmd, start_new_session=True)
+
+    def rulesLoaded(self):
+        rules = AppState.ruleManager().getRules()
+        self.existing_mapper_combobox.addItems(rules)