首页 > 编程 > Python > 正文

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

2019-11-25 12:25:09
字体:
来源:转载
供稿:网友

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

#!/usr/bin/env python# encoding: utf-8import logging# now we patch Python code to add color support to logging.StreamHandlerdef 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 newdef 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 newimport platformif 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版本以上的

以上这篇在PyCharm中控制台输出日志分层级分颜色显示的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持武林网。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表