kiwi-scp/src/kiwi-config.py

#!/usr/bin/env python3
import logging

import kiwi
from kiwi.subcommands import *


def set_verbosity(logger, handler, verbosity):
    if verbosity >= 2:
        log_level = logging.DEBUG
        log_format = "[%(asctime)s] %(levelname)s @ %(filename)s:%(funcName)s:%(lineno)d: %(message)s"
    elif verbosity >= 1:
        log_level = logging.INFO
        log_format = "[%(asctime)s] %(levelname)s: %(message)s"
    else:
        log_level = logging.WARNING
        log_format = "%(levelname)s: %(message)s"

    logger.setLevel(log_level)
    handler.setFormatter(logging.Formatter(log_format))


def main():
    commands = [
        InitCommand,
        ShowCommand,
        LogsCommand
    ]

    for cmd in commands:
        cmd.setup()

    args = kiwi.Parser.get_args()

    log_handler = logging.StreamHandler()
    logging.getLogger().addHandler(log_handler)

    set_verbosity(logging.getLogger(), log_handler, args.verbosity)

    for cmd in commands:
        if cmd.command == args.command:
            cmd.run()
            return


if __name__ == "__main__":
    main()