78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
import importlib
|
|
from types import SimpleNamespace
|
|
import typer
|
|
import sys
|
|
from pprint import pprint
|
|
from typing import List, Optional
|
|
from typing_extensions import Annotated
|
|
|
|
from beakers.beakers import SqliteBeaker
|
|
|
|
app = typer.Typer()
|
|
|
|
|
|
def _load_recipe(dotted_path: str):
|
|
sys.path.append(".")
|
|
path, name = dotted_path.rsplit(".", 1)
|
|
mod = importlib.import_module(path)
|
|
return getattr(mod, name)
|
|
|
|
|
|
@app.callback()
|
|
def main(
|
|
ctx: typer.Context,
|
|
recipe: str = typer.Option(None, envvar="BEAKER_RECIPE"),
|
|
):
|
|
if not recipe:
|
|
typer.secho(
|
|
"Missing recipe; pass --recipe or set env[BEAKER_RECIPE]",
|
|
fg=typer.colors.RED,
|
|
)
|
|
raise typer.Exit(1)
|
|
ctx.obj = _load_recipe(recipe)
|
|
|
|
|
|
@app.command()
|
|
def reset(ctx: typer.Context):
|
|
for beaker in ctx.obj.beakers.values():
|
|
if isinstance(beaker, SqliteBeaker):
|
|
if bl := len(beaker):
|
|
beaker.reset()
|
|
typer.secho(f"{beaker.name} reset ({bl})", fg=typer.colors.RED)
|
|
else:
|
|
typer.secho(f"{beaker.name} empty", fg=typer.colors.GREEN)
|
|
|
|
|
|
@app.command()
|
|
def show(ctx: typer.Context):
|
|
ctx.obj.show()
|
|
|
|
|
|
@app.command()
|
|
def graph(ctx: typer.Context):
|
|
pprint(ctx.obj.graph_data())
|
|
|
|
|
|
@app.command()
|
|
def run(
|
|
ctx: typer.Context,
|
|
input: Annotated[Optional[List[str]], typer.Option(...)] = None,
|
|
start: Optional[str] = typer.Option(None),
|
|
end: Optional[str] = typer.Option(None),
|
|
):
|
|
if ctx.obj.seeds:
|
|
typer.secho("Seeding beakers", fg=typer.colors.GREEN)
|
|
ctx.obj.process_seeds()
|
|
has_data = any(ctx.obj.beakers.values())
|
|
if not has_data and not input:
|
|
typer.secho("No data; pass --input to seed beaker(s)", fg=typer.colors.RED)
|
|
raise typer.Exit(1)
|
|
for input_str in input:
|
|
beaker, filename = input_str.split("=")
|
|
ctx.obj.csv_to_beaker(filename, beaker)
|
|
ctx.obj.run_once(start, end)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|