✨ feat: archieve the config command

Author: chenjunjie

opendigger_pycli/cli/__init__.py

@@ -3,12 +3,8 @@
 from .commands.config_cmd import config
 from .commands.display_cmd import display
 from .commands.export_cmd import export
-from .commands.monitor_cmd import monitor
-from .commands.report_cmd import report
 
 opendigger.add_command(config)
 
 query.add_command(display)
 query.add_command(export)
-query.add_command(report)
-query.add_command(monitor)

opendigger_pycli/cli/commands/config_cmd.py

@@ -1,8 +1,107 @@
-import typing as t  # noqa: F401
+import typing as t
+from dataclasses import fields
 
 import click
+from click.shell_completion import CompletionItem
 
+from opendigger_pycli.console import CONSOLE
 
-@click.command("config")
-def config():
-    pass
+from ..base import pass_environment
+from ..config import ALL_CONFIGS
+
+if t.TYPE_CHECKING:
+    from click.core import Context, Parameter
+
+    from opendigger_pycli.datatypes.config import BaseConfig
+
+    from ..base import Environment
+
+
+def config_shell_completion(
+    ctx: t.Optional["Context"], param: t.Optional["Parameter"], incomplete: str
+) -> t.List[CompletionItem]:
+    incomplete_splited = incomplete.rsplit(".", 1)[0]
+    if incomplete_splited == "":
+        return [
+            CompletionItem(config_dataclass_key) for config_dataclass_key in ALL_CONFIGS
+        ]
+    is_key = False
+    last_config_dataclass: t.Optional[BaseConfig] = None
+    for config_dataclass_key in ALL_CONFIGS:
+        if incomplete_splited.startswith(config_dataclass_key):
+            if incomplete_splited != config_dataclass_key:
+                return [CompletionItem(config_dataclass_key + ".")]
+            else:
+                is_key = True
+                last_config_dataclass = ALL_CONFIGS[config_dataclass_key]  # type: ignore
+                break
+        else:
+            continue
+
+    if not is_key and last_config_dataclass is None:
+        return []
+
+    if incomplete[:-1] == incomplete_splited:
+        return [CompletionItem(f"{incomplete_splited}.{field.name}") for field in fields(last_config_dataclass)]  # type: ignore
+
+    for field in fields(last_config_dataclass):  # type: ignore
+        if incomplete in f"{incomplete_splited}.{field.name}":
+            return [CompletionItem(f"{incomplete_splited}.{field.name}")]
+
+    return []
+
+
+def parse_config_key(key: str) -> t.Tuple[str, str]:
+    section_name, config_key = key.rsplit(".", 1)
+    return section_name, config_key
+
+
+def check_config_setting(
+    ctx: "Context", param: "Parameter", values: t.List[t.Tuple[str, str]]
+) -> t.List[t.Tuple[str, str]]:
+    for value in values:
+        try:
+            section_name, config_key = parse_config_key(value[0])
+        except Exception:
+            raise click.BadParameter(f"{value[0]} is not a valid config key")
+        else:
+            if section_name not in ALL_CONFIGS or config_key not in [
+                field.name for field in fields(ALL_CONFIGS[section_name])
+            ]:
+                raise click.BadParameter(f"{value[0]} is not a valid config key")
+    return values
+
+
+@click.command("config")  # type: ignore
+@click.option(
+    "--set",
+    "-s",
+    "config_settings",
+    type=(str, str),
+    multiple=True,
+    callback=check_config_setting,
+    shell_complete=config_shell_completion,
+    help="Set config value",
+    required=True,
+)
+@click.option(
+    "--persist/--no-persist",
+    "-p/-n",
+    is_flag=True,
+    default=True,
+    help="Set config value persistently",
+)
+@pass_environment
+def config(
+    env: "Environment", config_settings: t.List[t.Tuple[str, str]], persist: bool
+) -> None:
+    """Set config value"""
+    for config_setting in config_settings:
+        key, value = config_setting
+        section_name, key = parse_config_key(key)
+        env.dlog(f"set config: {section_name}.{key}={value}")
+        env.set_config(section_name, key, value, is_persist=persist)
+        env.dlog(f"finished to set config: {section_name}.{key}={value}")
+
+    CONSOLE.print("[green]Config set successfully[/]")
+    CONSOLE.print(env.cli_config)

opendigger_pycli/cli/commands/monitor_cmd.py

@@ -24,7 +24,10 @@ def __init__(self):
     def user_config_file_path(self) -> str:
         config_dir_str = click.get_app_dir("opendigger-pycli")
         config_dir = Path(config_dir_str)
+        config_dir.mkdir(parents=True, exist_ok=True)
         user_config = config_dir / "config.ini"
+        if not user_config.exists():
+            user_config.touch()
         return str(user_config)
 
     @property
@@ -40,7 +43,9 @@ def __load_config(self):
         parser = configparser.RawConfigParser()
         parser.read(self.config_file_paths)
 
-        for config_dataclass in ALL_CONFIGS:
+        for config_dataclass_key in ALL_CONFIGS:
+            config_dataclass = ALL_CONFIGS[config_dataclass_key]
+            config_dataclass = ALL_CONFIGS[config_dataclass_key]
             if not is_dataclass(config_dataclass):
                 raise TypeError(f"{config_dataclass} is not a dataclass")
 
@@ -54,8 +59,10 @@ def __load_config(self):
 
     def update_config(self):
         parser = configparser.RawConfigParser()
+        parser.read(self.config_file_paths)
 
-        for config_dataclass in ALL_CONFIGS:
+        for config_dataclass_key in ALL_CONFIGS:
+            config_dataclass = ALL_CONFIGS[config_dataclass_key]
             if not is_dataclass(config_dataclass):
                 raise TypeError(f"{config_dataclass} is not a dataclass")
 
@@ -75,7 +82,8 @@ def __rich_console__(
         self, console: "Console", options: "ConsoleOptions"
     ) -> "RenderResult":
         yield "[b]OpenDigger Python CLI Configs:[/b]"
-        for config_dataclass in ALL_CONFIGS:
+        for config_dataclass_key in ALL_CONFIGS:
+            config_dataclass = ALL_CONFIGS[config_dataclass_key]
             if not is_dataclass(config_dataclass):
                 raise TypeError(f"{config_dataclass} is not a dataclass")
 

opendigger_pycli/cli/commands/report_cmd.py

@@ -107,3 +107,14 @@ def set_log_level(
             self.log("[bold green]verbose mode enabled")
 
         self.load_configs()
+
+    def set_config(
+        self, section_name: str, key: str, value: str, *, is_persist: bool = True
+    ) -> None:
+        config = getattr(self.cli_config, section_name.replace(".", "_"))
+        setattr(config, key, value)
+        if is_persist:
+            self.cli_config.update_config()
+            self.dlog(
+                f"[bold green]set config {section_name}.{key} to {value} successfully"
+            )

opendigger_pycli/cli/config.py

@@ -21,4 +21,7 @@ class UserInfoConfig(BaseConfig):
     email: str = ""
 
 
-ALL_CONFIGS: t.List[t.Type[BaseConfig]] = [AppKeyConfig, UserInfoConfig]
+ALL_CONFIGS: t.Dict[str, t.Type[BaseConfig]] = {
+    "app_keys": AppKeyConfig,
+    "user_info": UserInfoConfig,
+}