data-anaxi/anaxi/cli.py

import importlib.metadata

import click
import pyfiglet

from anaxi.logging import get_anaxi_logger
from anaxi.processes import run_cosmos_db_healthcheck_process

logger = get_anaxi_logger(__name__)


@click.group()
def cli():
    logger.info(pyfiglet.figlet_format("\nWelcome to anaxi", font="big"))
    logger.info(f"Running anaxi version: {importlib.metadata.version('anaxi')}")


@cli.command()
def smoke_test():
    print("Oink oink!")
    print(
        """
           __,---.__
        ,-'         `-.__
      &/           `._\ _\\
      /               ''._
      |   ,             (")
      |__,'`-..--|__|--''
        """
    )


@cli.command()
@click.option("--cosmos-db-id", type=click.STRING, required=True)
def cosmos_db_healthcheck(cosmos_db_id):
    logger.info("Starting a Cosmos DB healthcheck.")
    run_cosmos_db_healthcheck_process(cosmos_db_id)
    logger.info("Finished the Cosmos DB healthcheck.")