#2533: added tests + log level for output

src/primaite/__init__.py

@@ -123,9 +123,8 @@ PRIMAITE_PATHS: Final[_PrimaitePaths] = _PrimaitePaths()
 
 
 def _host_primaite_config() -> None:
-    if not PRIMAITE_PATHS.app_config_file_path.exists():
-        pkg_config_path = Path(pkg_resources.resource_filename("primaite", "setup/_package_data/primaite_config.yaml"))
-        shutil.copy2(pkg_config_path, PRIMAITE_PATHS.app_config_file_path)
+    pkg_config_path = Path(pkg_resources.resource_filename("primaite", "setup/_package_data/primaite_config.yaml"))
+    shutil.copy(pkg_config_path, PRIMAITE_PATHS.app_config_file_path)
 
 
 _host_primaite_config()

src/primaite/cli.py

@@ -9,9 +9,8 @@ import typer
 import yaml
 from typing_extensions import Annotated
 
-from primaite import _PRIMAITE_ROOT, PRIMAITE_PATHS
+from primaite import PRIMAITE_PATHS
 from primaite.utils.cli import dev_cli
-from primaite.utils.cli.primaite_config_utils import get_primaite_config_dict, update_primaite_config
 
 app = typer.Typer(no_args_is_help=True)
 app.add_typer(dev_cli.dev, name="dev-mode")
@@ -116,15 +115,4 @@ def setup(overwrite_existing: bool = True) -> None:
     _LOGGER.info("Rebuilding the example notebooks...")
     reset_example_configs.run(overwrite_existing=True)
 
-    _LOGGER.info("Setting default simulation output")
-    config_dict = get_primaite_config_dict()
-
-    if config_dict is None:
-        return
-
-    config_dict["developer_mode"]["output_dir"] = str(_PRIMAITE_ROOT.parent.parent / "simulation_output")
-    print(f"PrimAITE dev-mode config updated  output_dir {config_dict['developer_mode']['output_dir']}")
-    # update application config
-    update_primaite_config(config_dict)
-
     _LOGGER.info("PrimAITE setup complete!")

src/primaite/setup/_package_data/primaite_config.yaml

@@ -2,6 +2,7 @@
 
 developer_mode:
   enabled: False  # not enabled by default
+  sys_log_level: DEBUG # level of output for system logs, DEBUG by default
   output_sys_logs: False  # system logs not output by default
   output_pcap_logs: False  # pcap logs not output by default
   output_to_terminal: False  # do not output to terminal by default

src/primaite/simulator/__init__.py

@@ -48,7 +48,7 @@ class _SimOutput:
             return self._path
         if is_dev_mode():
             dev_config = get_primaite_config_dict().get("developer_mode")
-            return Path(dev_config["output_path"]) if dev_config["output_path"] else self._default_path
+            return Path(dev_config["output_dir"]) if dev_config["output_dir"] else self._default_path
 
     @path.setter
     def path(self, new_path: Path) -> None:

src/primaite/utils/cli/dev_cli.py

@@ -1,8 +1,12 @@
+from typing import List
+
+import click
 import typer
 from rich import print
 from typing_extensions import Annotated
 
 from primaite import _PRIMAITE_ROOT
+from primaite.simulator import LogLevel
 from primaite.utils.cli.primaite_config_utils import get_primaite_config_dict, is_dev_mode, update_primaite_config
 
 dev = typer.Typer()
@@ -69,19 +73,44 @@ def disable():
 
 
 def config_callback(
-    ctx: typer.Context,
-    output_sys_logs: Annotated[
-        bool, typer.Option("--output-sys-logs/--no-sys-logs", "-sys/-nsys", help="Output system logs to file.")
-    ] = None,
-    output_pcap_logs: Annotated[
-        bool,
-        typer.Option(
-            "--output-pcap-logs/--no-pcap-logs", "-pcap/-npcap", help="Output network packet capture logs to file."
-        ),
-    ] = None,
-    output_to_terminal: Annotated[
-        bool, typer.Option("--output-to_terminal/--no-terminal", "-t/-nt", help="Output system logs to terminal.")
-    ] = None,
+        ctx: typer.Context,
+        sys_log_level: Annotated[
+            LogLevel,
+            typer.Option(
+                "--sys-log-level",
+                "-level",
+                click_type=click.Choice(LogLevel._member_names_, case_sensitive=False),
+                help="The level of system logs to output.",
+                show_default=False
+            )
+        ] = None,
+        output_sys_logs: Annotated[
+            bool,
+            typer.Option(
+                "--output-sys-logs/--no-sys-logs",
+                "-sys/-nsys",
+                help="Output system logs to file.",
+                show_default=False
+            )
+        ] = None,
+        output_pcap_logs: Annotated[
+            bool,
+            typer.Option(
+                "--output-pcap-logs/--no-pcap-logs",
+                "-pcap/-npcap",
+                help="Output network packet capture logs to file.",
+                show_default=False
+            ),
+        ] = None,
+        output_to_terminal: Annotated[
+            bool,
+            typer.Option(
+                "--output-to-terminal/--no-terminal",
+                "-t/-nt",
+                help="Output system logs to terminal.",
+                show_default=False
+            )
+        ] = None,
 ):
     """Configure the development tools and environment."""
     config_dict = get_primaite_config_dict()
@@ -89,6 +118,11 @@ def config_callback(
     if config_dict is None:
         return
 
+    if ctx.params.get("sys_log_level") is not None:
+        config_dict["developer_mode"]["sys_log_level"] = ctx.params.get("sys_log_level")
+        print(f"PrimAITE dev-mode config updated sys_log_level={ctx.params.get('sys_log_level')}")
+
+
     if output_sys_logs is not None:
         config_dict["developer_mode"]["output_sys_logs"] = output_sys_logs
         print(f"PrimAITE dev-mode config updated {output_sys_logs=}")
@@ -105,28 +139,33 @@ def config_callback(
     update_primaite_config(config_dict)
 
 
-config_typer = typer.Typer(callback=config_callback, name="config", invoke_without_command=True)
+config_typer = typer.Typer(
+    callback=config_callback,
+    name="config",
+    no_args_is_help=True,
+    invoke_without_command=True,
+)
 dev.add_typer(config_typer)
 
 
 @config_typer.command()
 def path(
-    directory: Annotated[
-        str,
-        typer.Argument(
-            help="Directory where the system logs and PCAP logs will be output. By default, this will be where the"
-            "root of the PrimAITE repository is located.",
-            show_default=False,
-        ),
-    ] = None,
-    default: Annotated[
-        bool,
-        typer.Option(
-            "--default",
-            "-root",
-            help="Set PrimAITE to output system logs and pcap logs to the PrimAITE repository root.",
-        ),
-    ] = None,
+        directory: Annotated[
+            str,
+            typer.Argument(
+                help="Directory where the system logs and PCAP logs will be output. By default, this will be where the"
+                     "root of the PrimAITE repository is located.",
+                show_default=False,
+            ),
+        ] = None,
+        default: Annotated[
+            bool,
+            typer.Option(
+                "--default",
+                "-root",
+                help="Set PrimAITE to output system logs and pcap logs to the PrimAITE repository root.",
+            ),
+        ] = None,
 ):
     """Set the output directory for the PrimAITE system and PCAP logs."""
     config_dict = get_primaite_config_dict()
@@ -136,16 +175,15 @@ def path(
 
     if default:
         config_dict["developer_mode"]["output_dir"] = None
-        print(
-            f"PrimAITE dev-mode config updated output directory will be in "
-            f"{str(_PRIMAITE_ROOT.parent.parent / 'simulation_output')}"
-        )
         # update application config
         update_primaite_config(config_dict)
+        print(f"PrimAITE dev-mode output_dir [medium_turquoise]"
+              f"{str(_PRIMAITE_ROOT.parent.parent / 'simulation_output')}"
+              f"[/medium_turquoise]")
         return
 
     if directory:
         config_dict["developer_mode"]["output_dir"] = directory
-        print(f"PrimAITE dev-mode config updated output_dir={directory}")
         # update application config
         update_primaite_config(config_dict)
+        print(f"PrimAITE dev-mode output_dir [medium_turquoise]{directory}[/medium_turquoise]")

src/primaite/utils/cli/primaite_config_utils.py

@@ -1,17 +1,30 @@
-from typing import Dict
+from pathlib import Path
+from typing import Dict, Optional
 
 import yaml
 
 from primaite import PRIMAITE_PATHS
 
 
-def get_primaite_config_dict() -> Dict:
-    """Returns a dict containing the PrimAITE application config."""
-    if PRIMAITE_PATHS.app_config_file_path.exists():
-        with open(PRIMAITE_PATHS.app_config_file_path, "r") as file:
+def get_primaite_config_dict(
+        config_path: Optional[Path] = None
+) -> Dict:
+    """
+    Returns a dict containing the PrimAITE application config.
+
+    :param: config_path: takes in a path object - leave empty to use the default app config path
+    """
+    err_msg = "PrimAITE application config could not be loaded."
+
+    if config_path is None:
+        config_path = PRIMAITE_PATHS.app_config_file_path
+        err_msg = "PrimAITE application config was not found. Have you run `primaite setup`?"
+
+    if config_path.exists():
+        with open(config_path, "r") as file:
             return yaml.safe_load(file)
     else:
-        print("PrimAITE application config was not found. Have you run [bold red]primaite setup[/bold red]?")
+        print(err_msg)
 
 
 def is_dev_mode() -> bool: