#!/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)