uncloud-mravi/scripts/ucloud

#!/usr/bin/env python3

import argparse
import logging
import importlib
import os
import multiprocessing as mp
import sys

from ucloud.configure.main import update_config, configure_parser


def exception_hook(exc_type, exc_value, exc_traceback):
        logger.error(
            "Uncaught exception",
            exc_info=(exc_type, exc_value, exc_traceback)
        )
        print(exc_type, exc_value)


if __name__ == '__main__':
    logging.basicConfig(level=logging.DEBUG,
                    format='%(pathname)s:%(lineno)d -- %(levelname)-8s %(message)s',
                    filename='/var/log/ucloud.log', filemode='a')
    logger = logging.getLogger("ucloud")

    sys.excepthook = exception_hook
    mp.set_start_method('spawn')

    arg_parser = argparse.ArgumentParser()
    subparsers = arg_parser.add_subparsers(dest="command")
    
    api_parser = subparsers.add_parser("api")
    
    host_parser = subparsers.add_parser("host")
    host_parser.add_argument("--hostname", required=True)

    scheduler_parser = subparsers.add_parser("scheduler")

    filescanner_parser = subparsers.add_parser("filescanner")

    imagescanner_parser = subparsers.add_parser("imagescanner")

    metadata_parser = subparsers.add_parser("metadata")

    config_parser = subparsers.add_parser("configure")
    configure_parser(config_parser)
    args = arg_parser.parse_args()

    if not args.command:
        arg_parser.print_help()
    else:
        arguments = vars(args)
        try:
            name = arguments.pop('command')
            mod = importlib.import_module("ucloud.{}.main".format(name))
            main = getattr(mod, "main")
            main(**arguments)

        except Exception as e:
            logger.exception(e)
            print(e)