Python日志输出加颜色

 

替换颜色

xxxxxxxxxx
import logging
 
 
 
BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = range(8)
 
#The background is set with 40 plus the number of the color, and the foreground with 30
 
#These are the sequences need to get colored ouput
RESET_SEQ = "\033[0m"
COLOR_SEQ = "\033[1;%dm"
BOLD_SEQ = "\033[1m"
 
def formatter_message(message, use_color = True):
    if use_color:
        message = message.replace("$RESET", RESET_SEQ).replace("$BOLD", BOLD_SEQ)
    else:
        message = message.replace("$RESET", "").replace("$BOLD", "")
    return message
 
COLORS = {
    'WARNING': YELLOW,
    'INFO': BLUE,
    'DEBUG': WHITE,
    'CRITICAL': GREEN,
    'ERROR': RED
}
 
class ColoredFormatter(logging.Formatter):
    def __init__(self, msg, use_color = True):
        logging.Formatter.__init__(self, msg)
        self.use_color = use_color
 
    def format(self, record):
        levelname = record.levelname
        msg = record.msg
        if self.use_color and levelname in COLORS:
            levelname_color = COLOR_SEQ % (30 + COLORS[levelname]) + levelname + RESET_SEQ
            msg_color = COLOR_SEQ % (30 + COLORS[levelname]) + msg + RESET_SEQ
            record.levelname = levelname_color
            record.msg = msg_color            
            
        return logging.Formatter.format(self, record)
 
 
 
# Custom logger class with multiple destinations
class ColoredLogger(logging.Logger):
    FORMAT = "[$BOLD%(name)-20s$RESET][%(levelname)-18s] $BOLD%(message)s$RESET (%(filename)s:%(lineno)d)"
    COLOR_FORMAT = formatter_message(FORMAT, True)
    def __init__(self, name):
        logging.Logger.__init__(self, name, logging.DEBUG)                
 
        color_formatter = ColoredFormatter(self.COLOR_FORMAT)
 
        console = logging.StreamHandler()
        console.setFormatter(color_formatter)
 
        self.addHandler(console)
        return


def log_start(module_name):
    logging.setLoggerClass(ColoredLogger)
    color_log = logging.getLogger(module_name)
    color_log.setLevel(logging.INFO)
    return color_log

if __name__ == '__main__':
    log = log_start('test')
    log.critical('dddd')

 

coloredlogs模块

xxxxxxxxxx
#####################################################################
import logging
logger = logging.getLogger('your-module')
 
# Initialize coloredlogs.
import coloredlogs
coloredlogs.install(level='INFO')
 
# Some examples.
logger.debug("this is a debugging message")
logger.info("this is an informational message")
logger.warning("this is a warning message")
logger.error("this is an error message")