Merge pull request #4 from CoderChen01/cjj_dev

Complete the config command and json data export

Author: CoderChen01

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/display_cmd.py

@@ -13,7 +13,7 @@
     from opendigger_pycli.console.print_indicator import SURPPORTED_DISPLAY_FORMAT_TYPE
     from opendigger_pycli.results.query import QueryResults
 
-    from .base import Environment
+    from ..base import Environment
 
 
 @click.command("display", help="Display metrics")

opendigger_pycli/cli/commands/export_cmd.py

@@ -1,29 +1,53 @@
 import typing as t
+from pathlib import Path
 
 import click
 
+from opendigger_pycli.exporters import (
+    SURPPORTED_EXPORT_FORMAT_TYPE,
+    SURPPORTED_EXPORT_FORMATS,
+)
+from opendigger_pycli.results.export import ExportResult
 from opendigger_pycli.utils.decorators import processor
 
 from ..base import pass_environment
 
 if t.TYPE_CHECKING:
+    from opendigger_pycli.results.query import QueryResults
+
     from ..base import Environment
 
 
 @click.command("export", help="Export metrics")
 @click.option(
     "--format",
     "-f",
-    "format_name",
-    type=click.Choice(["csv", "json", "mhtml"]),
+    type=click.Choice(SURPPORTED_EXPORT_FORMATS),
     required=True,
+    help="Format to export",
+)
+@click.option(
+    "--save-dir",
+    "-s",
+    "save_dir",
+    type=click.Path(file_okay=False, resolve_path=True, path_type=Path),
+    required=True,
+    help="Directory to save indicators",
+)
+@click.option(
+    "--split/--no-split",
+    "is_split",
+    default=True,
+    help="Save indicators in separate files",
 )
-@click.option("--filename", "-o", "filename", type=str, required=True)
 @processor
 @pass_environment
 def export(
     env: "Environment",
-    format_name: t.Literal["csv", "json", "mhtml"],
-    filename: click.Path,
+    results: "QueryResults",
+    format: SURPPORTED_EXPORT_FORMAT_TYPE,
+    save_dir: Path,
+    is_split: bool,
 ):
-    pass
+    ExportResult(results, format, save_dir, is_split).export()
+    yield from results

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

@@ -3,7 +3,7 @@
 import click
 from click.shell_completion import CompletionItem
 
-from opendigger_pycli.dataloader import (
+from opendigger_pycli.dataloaders import (
     DeveloperNetworkRepoDataloader,
     ProjectOpenRankNetworkRepoDataloader,
     RepoNetworkRepoDataloader,

opendigger_pycli/cli/config.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/custom_types.py

@@ -3,7 +3,7 @@
 
 import click
 
-from opendigger_pycli.dataloader import filter_dataloader
+from opendigger_pycli.dataloaders import filter_dataloader
 
 if t.TYPE_CHECKING:
     from click import Context