From eea6c1568e2ad9675032c3950902ead88bbeb5fc Mon Sep 17 00:00:00 2001 From: meow Date: Sun, 22 Dec 2019 13:47:16 +0500 Subject: [PATCH] colored error output --- scripts/ucloud | 11 +++++++++++ setup.py | 1 + 2 files changed, 12 insertions(+) diff --git a/scripts/ucloud b/scripts/ucloud index f741663..0a6c5ec 100755 --- a/scripts/ucloud +++ b/scripts/ucloud @@ -6,6 +6,8 @@ import importlib import multiprocessing as mp import sys +import colorama + from logging.handlers import SysLogHandler from ucloud.configure.main import configure_parser @@ -24,11 +26,20 @@ 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 == 'WARNING': + color = colorama.Fore.YELLOW + elif record.levelname == 'ERROR': + color = colorama.Fore.LIGHTRED_EX + else: + color = colorama.Fore.RED try: + print(color) super().handle(record) finally: record.exc_info = info record.exc_text = cache + print(colorama.Style.RESET_ALL) if __name__ == '__main__': sys.excepthook = exception_hook diff --git a/setup.py b/setup.py index 5a624db..b4341d3 100644 --- a/setup.py +++ b/setup.py @@ -37,6 +37,7 @@ setup(name='ucloud', 'sshtunnel', 'sphinx', 'pynetbox', + 'colorama', 'sphinx-rtd-theme', 'etcd3 @ https://github.com/kragniz/python-etcd3/tarball/master#egg=etcd3', ],