Early development of command line tool for generating a configuration file

Author: d-j-hatton

pyproject.toml

@@ -84,6 +84,7 @@ murfey = "murfey.client:run"
 "murfey.db_sql" = "murfey.cli.murfey_db_sql:run"
 "murfey.decrypt_password" = "murfey.cli.decrypt_db_password:run"
 "murfey.dlq_murfey" = "murfey.cli.dlq_resubmit:run"
+"murfey.generate_config" = "murfey.cli.generate_config:run"
 "murfey.generate_key" = "murfey.cli.generate_crypto_key:run"
 "murfey.generate_password" = "murfey.cli.generate_db_password:run"
 "murfey.instrument_server" = "murfey.instrument_server:run"

src/murfey/cli/generate_config.py

@@ -0,0 +1,27 @@
+from pydantic import ValidationError
+from pydantic.fields import UndefinedType
+from rich.pretty import pprint
+from rich.prompt import Prompt
+
+from murfey.util.config import MachineConfig
+
+
+def run():
+    new_config = {}
+    for k, field in MachineConfig.__fields__.items():
+        pprint(field.name)
+        pprint(field.field_info.description)
+        if isinstance(field.field_info.default, UndefinedType):
+            value = Prompt.ask("Please provide a value")
+        else:
+            value = Prompt.ask(
+                "Please provide a value", default=field.field_info.default
+            )
+        new_config[k] = value
+
+        try:
+            MachineConfig.validate(new_config)
+        except ValidationError as exc:
+            for ve in exc.errors():
+                if ve["type"] != "value_error.missing":
+                    print("Validation failed")

src/murfey/util/config.py

@@ -103,7 +103,7 @@ class MachineConfig(BaseModel):
     def __validate_rsync_basepath_if_transfer_enabled__(
         cls, v: Optional[str], values: Mapping[str, Any]
     ) -> Any:
-        if values["data_transfer_enabled"]:
+        if values.get("data_transfer_enabled"):
             if v is None:
                 raise NoneIsNotAllowedError
         return v
@@ -112,7 +112,9 @@ def __validate_rsync_basepath_if_transfer_enabled__(
     def __validate_default_model_if_processing_enabled_and_spa_possible__(
         cls, v: Optional[str], values: Mapping[str, Any]
     ) -> Any:
-        if values["processing_enabled"] and "epu" in values["acquisition_software"]:
+        if values.get("processing_enabled") and "epu" in values.get(
+            "acquisition_software", []
+        ):
             if v is None:
                 raise NoneIsNotAllowedError
         return v