1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
import asyncio
from typing import Optional
import typer
from robusta_krr.core.config import Config
from robusta_krr.core.runner import Runner
from robusta_krr.utils.version import get_version
app = typer.Typer()
@app.command()
def version() -> None:
typer.echo(get_version())
@app.command()
def run(
prometheus_url: Optional[str] = typer.Option(
None,
"--prometheus-url",
"-p",
help="Prometheus URL. If not provided, will attempt to find it in kubernetes cluster",
),
format: str = typer.Option("text", "--formatter", "-f", help="Output formatter"),
strategy: str = typer.Option("simple", "--strategy", "-s", help="Strategy to use"),
verbose: bool = typer.Option(False, "--verbose", "-v", help="Enable verbose mode"),
quiet: bool = typer.Option(False, "--quiet", "-q", help="Enable quiet mode"),
) -> None:
config = Config(
prometheus_url=prometheus_url,
format=format,
verbose=verbose,
quiet=quiet,
strategy=strategy,
)
runner = Runner(config)
asyncio.run(runner.run())
if __name__ == "__main__":
app()
|
