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

1.把下面代码复制到一个.py文件中,命名为colorer(可以自己定义)
2.把这个文件放到项目的utils包中(第三方包文件夹中或者项目的任何地方)
3.在配置文件中导入这个模块即可
4.pycharm版本最好是新版本的,16.3的版本显示不出来

#!/usr/bin/env python

encoding: utf-8

import logging
import ctypes

def add_coloring_to_emit_windows(fn):
​ def _out_handle(self):
​ 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):

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
    
    return fn(*args)

return new

import platform

if platform.system() == ‘Windows’:

logging.StreamHandler.emit = add_coloring_to_emit_windows(logging.StreamHandler.emit)

else:

logging.StreamHandler.emit = add_coloring_to_emit_ansi(logging.StreamHandler.emit)

你可能感兴趣的:(项目配置)