camisatoshi-wordpress-reports/camisatoshi_wordpress_reports/cli.py

import datetime

import typer

from typing_extensions import Annotated

import camisatoshi_wordpress_reports.controllers as controllers
import camisatoshi_wordpress_reports.logging as our_logging


our_logging.set_config_level()
app = typer.Typer()


@app.command()
def check_health():
    controllers.check_health()


@app.command()
def generate_sku_report(
    start_date: Annotated[datetime.datetime, typer.Option(prompt=True)],
    end_date: Annotated[datetime.datetime, typer.Option(prompt=True)],
    sku: Annotated[str, typer.Option(prompt=True)],
):
    controllers.generate_sku_report(start_date, end_date, sku)


@app.command()
def generate_um_report(
    start_date: Annotated[datetime.datetime, typer.Option(prompt=True)],
    end_date: Annotated[datetime.datetime, typer.Option(prompt=True)],
):
    controllers.generate_um_report(start_date, end_date)