Module cevast.cli

"""Entry point of CEVAST Command Line Interface."""

import os
import click
from .utils.logging import setup_cevast_logger, setup_cli_logger
from .certdb import cli as certdb_cli
from .dataset import cli as dataset_cli
from .analysis import cli as analysis_cli


@click.group()
@click.option('--debug/--no-debug', default=False, help='DEBUG logging level will be turned on.')
@click.option('--cpu', default=os.cpu_count() - 1, type=int, help='Max Number of CPU cores to use.')
@click.version_option()
@click.pass_context
def cli(ctx, debug, cpu):
    """Cevast is a set of tools for collection and analysis of X.509 certificate datasets."""

    # based on parameters setup logger
    setup_cevast_logger(debug=debug, process_id=cpu > 1)
    setup_cli_logger()

    if debug:
        click.echo('Debug mode is ON')

    ctx.ensure_object(dict)
    ctx.obj['cpu'] = cpu


cli.add_command(certdb_cli.certdb_group)
cli.add_command(dataset_cli.dataset_repository_group)
cli.add_command(dataset_cli.manager_group)
cli.add_command(analysis_cli.analysis_group)


if __name__ == "__main__":
    cli()  # pylint: disable=E1120