data-anaxi/anaxi/cli.py

import importlib.metadata
import logging

import click
import pyfiglet

from anaxi.processes import run_cosmos_db_healthcheck_process

logging.basicConfig(
    level=logging.DEBUG,
    format="%(asctime)s - [%(levelname)s] - %(filename)s - L%(lineno)d - %(message)s",
    handlers=[logging.StreamHandler()],
)

logger = logging.getLogger()


@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)
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.")