Source code for autojob.cli.main

"""Define the CLI subcommands."""

import importlib
import logging
from pathlib import Path

import click

from autojob import SETTINGS
from autojob.__about__ import __version__
from autojob.cli.utils import configure_logging

COMMANDS = (
    ("autojob.coordinator.cli", "coordinator"),
    ("autojob.harvest.cli", "harvest"),
    # ('autojob.cli.diff', 'diff'),
    ("autojob.cli.advance", "advance"),
    ("autojob.cli.init", "init"),
)


# TODO: Add select SETTINGS as CLI args
@click.group(
    name="autojob",
    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
):
    """Main CLI logic."""
    if version:
        click.echo(f"autojob-{__version__}")

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

    configure_logging(console_log_level)




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



add_subcommands()