PyCharm中控制台输出日志分层级分颜色显示

分手后的思念是犯贱 2022-05-08 10:30 287阅读 0赞

1.把下面代码复制到一个.py文件中

#!/usr/bin/env python
    # encoding: utf-8
    import logging
    
    
    # now we patch Python code to add color support to logging.StreamHandler
    def add_coloring_to_emit_windows(fn):
        # add methods we need to the class
        def _out_handle(self):
            import ctypes
            return ctypes.windll.kernel32.GetStdHandle(self.STD_OUTPUT_HANDLE)
    
        out_handle = property(_out_handle)
    
        def _set_color(self, code):
            import ctypes
            # Constants from the Windows API
            self.STD_OUTPUT_HANDLE = -11
            hdl = ctypes.windll.kernel32.GetStdHandle(self.STD_OUTPUT_HANDLE)
            ctypes.windll.kernel32.SetConsoleTextAttribute(hdl, code)
    
        setattr(logging.StreamHandler, '_set_color', _set_color)
    
        def new(*args):
            FOREGROUND_BLUE = 0x0001  # text color contains blue.
            FOREGROUND_GREEN = 0x0002  # text color contains green.
            FOREGROUND_RED = 0x0004  # text color contains red.
            FOREGROUND_INTENSITY = 0x0008  # text color is intensified.
            FOREGROUND_WHITE = FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED
            # winbase.h
            STD_INPUT_HANDLE = -10
            STD_OUTPUT_HANDLE = -11
            STD_ERROR_HANDLE = -12
    
            # wincon.h
            FOREGROUND_BLACK = 0x0000
            FOREGROUND_BLUE = 0x0001
            FOREGROUND_GREEN = 0x0002
            FOREGROUND_CYAN = 0x0003
            FOREGROUND_RED = 0x0004
            FOREGROUND_MAGENTA = 0x0005
            FOREGROUND_YELLOW = 0x0006
            FOREGROUND_GREY = 0x0007
            FOREGROUND_INTENSITY = 0x0008  # foreground color is intensified.
    
            BACKGROUND_BLACK = 0x0000
            BACKGROUND_BLUE = 0x0010
            BACKGROUND_GREEN = 0x0020
            BACKGROUND_CYAN = 0x0030
            BACKGROUND_RED = 0x0040
            BACKGROUND_MAGENTA = 0x0050
            BACKGROUND_YELLOW = 0x0060
            BACKGROUND_GREY = 0x0070
            BACKGROUND_INTENSITY = 0x0080  # background color is intensified.
    
            levelno = args[1].levelno
            if (levelno >= 50):
                color = BACKGROUND_YELLOW | FOREGROUND_RED | FOREGROUND_INTENSITY | BACKGROUND_INTENSITY
            elif (levelno >= 40):
                color = FOREGROUND_RED | FOREGROUND_INTENSITY
            elif (levelno >= 30):
                color = FOREGROUND_YELLOW | FOREGROUND_INTENSITY
            elif (levelno >= 20):
                color = FOREGROUND_GREEN
            elif (levelno >= 10):
                color = FOREGROUND_MAGENTA
            else:
                color = FOREGROUND_WHITE
            args[0]._set_color(color)
    
            ret = fn(*args)
            args[0]._set_color(FOREGROUND_WHITE)
            # print "after"
            return ret
    
        return new
    
    
    def add_coloring_to_emit_ansi(fn):
        # add methods we need to the class
        def new(*args):
            levelno = args[1].levelno
            if (levelno >= 50):
                color = '\x1b[31m'  # red
            elif (levelno >= 40):
                color = '\x1b[31m'  # red
            elif (levelno >= 30):
                color = '\x1b[33m'  # yellow
            elif (levelno >= 20):
                color = '\x1b[32m'  # green
            elif (levelno >= 10):
                color = '\x1b[35m'  # pink
            else:
                color = '\x1b[0m'  # normal
            try:
                args[1].msg = color + args[1].msg + '\x1b[0m'  # normal
            except Exception as e:
                pass
            # print "after"
            return fn(*args)
    
        return new
    
    
    import platform
    
    if platform.system() == 'Windows':
        # Windows does not support ANSI escapes and we are using API calls to set the console color
        logging.StreamHandler.emit = add_coloring_to_emit_windows(logging.StreamHandler.emit)
    else:
        # all non-Windows platforms are supporting ANSI escapes so we use them
        logging.StreamHandler.emit = add_coloring_to_emit_ansi(logging.StreamHandler.emit)
        # log = logging.getLogger()
        # log.addFilter(log_filter())
        # //hdlr = logging.StreamHandler()
        # //hdlr.setFormatter(formatter())

2.将文件放到项目的utils包中(或者项目其他路径)  
3.在配置文件中导入这个py文件  
比如在Django的setting.py文件中配置日志信息时,可以先导入这个文件  
之后运行代码,控制台会根据日志等级显示不同的颜色  
4.pycharm版本最好是17版本以上的