diff --git a/scripts/uncloud b/scripts/uncloud index 05e47a5..e5c4081 100755 --- a/scripts/uncloud +++ b/scripts/uncloud @@ -6,9 +6,7 @@ import multiprocessing as mp import sys from logging.handlers import SysLogHandler - -from ucloud.common.logging import NoTracebackStreamHandler -from ucloud.configure.main import configure_parser +from uncloud.configure.main import configure_parser def exception_hook(exc_type, exc_value, exc_traceback): @@ -31,19 +29,13 @@ if __name__ == '__main__': syslog_formatter = logging.Formatter('%(pathname)s:%(lineno)d -- %(levelname)-8s %(message)s') syslog_handler.setFormatter(syslog_formatter) - stream_handler = NoTracebackStreamHandler() - stream_handler.setLevel(logging.INFO) - stream_formatter = logging.Formatter('%(message)s') - stream_handler.setFormatter(stream_formatter) - logger.addHandler(syslog_handler) - logger.addHandler(stream_handler) 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) @@ -72,7 +64,7 @@ if __name__ == '__main__': arguments = vars(args) try: name = arguments.pop('command') - mod = importlib.import_module("ucloud.{}.main".format(name)) + mod = importlib.import_module("uncloud.{}.main".format(name)) main = getattr(mod, "main") main(**arguments) except Exception as err: diff --git a/uncloud/common/logging.py b/uncloud/common/logging.py deleted file mode 100644 index 9e0d2be..0000000 --- a/uncloud/common/logging.py +++ /dev/null @@ -1,27 +0,0 @@ -import logging -import colorama - - -class NoTracebackStreamHandler(logging.StreamHandler): - def handle(self, record): - info, cache = record.exc_info, record.exc_text - record.exc_info, record.exc_text = None, None - - if record.levelname in ["WARNING", "WARN"]: - color = colorama.Fore.LIGHTYELLOW_EX - elif record.levelname == "ERROR": - color = colorama.Fore.LIGHTRED_EX - elif record.levelname == "INFO": - color = colorama.Fore.LIGHTGREEN_EX - elif record.levelname == "CRITICAL": - color = colorama.Fore.LIGHTCYAN_EX - else: - color = colorama.Fore.WHITE - - try: - print(color, end="", flush=True) - super().handle(record) - finally: - record.exc_info = info - record.exc_text = cache - print(colorama.Style.RESET_ALL, end="", flush=True)