log-formater.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
import logging
import logging.handlers

class ColorFormatter(logging.Formatter):
    """Format logging with nice colors"""
    FORMAT_CON = ("[$BOLD%(asctime)-20s$RESET]"
                  " [%(levelname)-19s]  %(message)s ")

    FORMAT_LOG = ("$BOLD[%(asctime)-9s]$RESET"
                  " [%(levelname)-8s] [%(process)d]"
                  " [%(name)s:%(lineno)d]  %(message)s ")

    BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = range(8)

    RESET_SEQ = "\033[0m"
    COLOR_SEQ = "\033[1;%dm"
    BOLD_SEQ = "\033[1m"

    COLORS = {
        'WARNING': MAGENTA,
        'INFO': GREEN,
        'DEBUG': BLUE,
        'CRITICAL': RED,
        'ERROR': RED,
        'RED': RED,
        'GREEN': GREEN,
        'YELLOW': YELLOW,
        'BLUE': BLUE,
        'MAGENTA': MAGENTA,
        'CYAN': CYAN,
        'WHITE': WHITE,
    }

    def __init__(self, use_color=True, print_to_console=True):
        if print_to_console:
            msg = self.formatter_msg(
                self.FORMAT_CON, use_color, print_to_console)
        else:
            msg = self.formatter_msg(
                self.FORMAT_LOG, use_color, print_to_console)
        logging.Formatter.__init__(self, msg, datefmt='%m-%d-%Y %I:%M:%S %p')
        self.use_color = use_color

    def formatter_msg(self, msg, use_color=True, print_to_console=True):
        if use_color:
            msg = msg.replace(
                "$RESET", self.RESET_SEQ).replace("$BOLD", self.BOLD_SEQ)
        else:
            msg = msg.replace("$RESET", "").replace("$BOLD", "")
        return msg

    def format(self, record):
        message_text = record.msg
        levelname = record.levelname
        if self.use_color and levelname in self.COLORS:
            fore_color = 30 + self.COLORS[levelname]
            levelname_color = self.COLOR_SEQ % fore_color \
                + levelname + self.RESET_SEQ
            message_color = self.COLOR_SEQ % fore_color \
                + message_text + self.RESET_SEQ
            record.levelname = levelname_color
            record.msg = message_color
        return logging.Formatter.format(self, record)

if __name__ == '__main__':
    # create logger
    log = logging.getLogger(__file__)
    log.setLevel(logging.DEBUG)
    
    # create console file handler and set level to debug
    console_handler = logging.StreamHandler()
    console_handler.setLevel(logging.DEBUG)
    
    # create log file handler and set level to debug
    # the logfile_handler is autorotating the log file
    logfile_handler = logging.handlers.RotatingFileHandler(opt.log_file,
                                                           maxBytes=1024*1024,
                                                           backupCount=1)
    logfile_handler.setLevel(logging.DEBUG)

    # format the handlers output
    console_handler.setFormatter(ColorFormatter())
    logfile_handler.setFormatter(ColorFormatter(use_color=False,
                                                print_to_console=False))
    # add handlers to the logger
    log.addHandler(logfile_handler)
    log.addHandler(console_handler)