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


						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版本以上的

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