Mods

cromulant/argspec.py

@@ -0,0 +1,83 @@
+from __future__ import annotations
+
+from typing import Any
+
+from .config import Config
+
+
+class DuplicateArgumentError(Exception):
+    def __init__(self, key: str) -> None:
+        self.message = f"Duplicate argument: {key}"
+
+    def __str__(self) -> str:
+        return self.message
+
+
+class MissingInfoError(Exception):
+    def __init__(self, key: str) -> None:
+        self.message = f"Missing info for argument: {key}"
+
+    def __str__(self) -> str:
+        return self.message
+
+
+class DuplicateInfoError(Exception):
+    def __init__(self, key: str) -> None:
+        self.message = f"Duplicate info for argument: {key}"
+
+    def __str__(self) -> str:
+        return self.message
+
+
+class ArgSpec:
+    vinfo: str
+    defaults: dict[str, Any]
+    arguments: dict[str, Any]
+    infos: list[str]
+
+    @staticmethod
+    def prepare() -> None:
+        ArgSpec.vinfo = f"{Config.title} {Config.version}"
+        ArgSpec.defaults = {}
+        ArgSpec.arguments = {}
+        ArgSpec.infos = []
+        ArgSpec.add_arguments()
+
+    @staticmethod
+    def add_argument(key: str, info: str, **kwargs: Any) -> None:
+        if key in ArgSpec.arguments:
+            raise DuplicateArgumentError(key)
+
+        if not info:
+            raise MissingInfoError(key)
+
+        if info in ArgSpec.infos:
+            raise DuplicateInfoError(key)
+
+        ArgSpec.arguments[key] = {
+            "help": info,
+            **kwargs,
+        }
+
+        ArgSpec.infos.append(info)
+
+    @staticmethod
+    def add_arguments() -> None:
+        ArgSpec.add_argument(
+            "version",
+            action="version",
+            info="Check the version of the program",
+            version=ArgSpec.vinfo,
+        )
+
+        ArgSpec.add_argument(
+            "names",
+            type=str,
+            info="Path to a JSON files with a list of names. The game will use these names instead of the default ones",
+        )
+
+        ArgSpec.add_argument(
+            "no_images",
+            action="store_false",
+            info="Don't show the images on the left",
+        )