Source code for ccu.cli.main

"""Module that contains the command line app."""

import importlib
import logging
from pathlib import Path

import click

import ccu
from ccu.cli.utils import configure_logging

COMMANDS = (
    ("ccu.adsorption.cli", "adsorption"),
    ("ccu.bader.cli", "bader"),
    ("ccu.fancyplots.cli", "fed"),
    ("ccu.structure.cli", "structure"),
    ("ccu.thermo.cli", "thermo"),
)


@click.group(
    name="ccu",
    invoke_without_command=True,
    context_settings={"help_option_names": ["-h", "--help"]},
)
@click.option(
    "-V",
    "--version",
    is_flag=True,
    default=False,
    help="Show the version and exit.",
)
@click.option(
    "-v",
    "--verbose",
    "verbosity",
    default=0,
    count=True,
    help="Control how much information is printed to the terminal. This "
    "option can be repeated.",
)
@click.option(
    "-q",
    "--quiet",
    "quietness",
    default=0,
    count=True,
)
@click.option(
    "--log-file",
    default=None,
    type=click.Path(dir_okay=False, writable=True, path_type=Path),
    help="Specify a log file for all logging messages.",
)
@click.option(
    "--log-level",
    default="WARNING",
    type=click.Choice(
        ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"], case_sensitive=False
    ),
    help="Specify the log level for all logging messages.",
)
def main(
    version: bool, verbosity: int, quietness: int, log_file, log_level: str
) -> None:
    """The main CLI function."""
    if version:
        click.echo(f"ccu-{ccu.__version__}")

    console_log_level = 10 * (quietness - verbosity) + logging.WARNING
    file_log_level = getattr(logging, log_level.upper())

    configure_logging(
        console_log_level=console_log_level,
        log_file=log_file,
        file_log_level=file_log_level,
    )




def add_subcommands() -> None:
    """Add the subcommands."""
    for module, name in COMMANDS:
        command = importlib.import_module(module).main
        main.add_command(command, name=name)



add_subcommands()