python logger 重写

#!/usr/bin/python
#encoding:utf-8


import logging  
import logging.handlers  


LOG_LEVEL = logging.DEBUG
class ColorFormatter(logging.Formatter):


    def __init__(self, msg):
        logging.Formatter.__init__(self, msg)
        
    def format(self, record):
        print record
        red, green, yellow, blue = range(4)
        colors = {'INFO': green, 'DEBUG': blue,
                  'WARNING': yellow, 'ERROR': red}
        msg = record.msg
        if msg[0] == "+":
            msg = "\33[01m" + msg[1:] + "\033[0m"
        elif msg[0] == "=":
            msg = "\33[07m" + msg + "\033[0m"
        levelname = record.levelname
        if levelname in colors:
            msg_color = "\033[0;%dm" % (
                31 + colors[levelname]) + msg + "\033[0m"
            record.msg = msg_color
        return logging.Formatter.format(self, record)






def main():
    global LOG
    LOG = logging.getLogger("pack-tool")
    LOG.setLevel(LOG_LEVEL)
    stream_handler = logging.StreamHandler()
    stream_handler.setLevel(LOG_LEVEL)
    
    stream_formatter = ColorFormatter("[%(asctime)s] %(message)s")


    stream_handler.setFormatter(stream_formatter)
    LOG.addHandler(stream_handler)


    LOG.info("+Building package ...")


if __name__ == "__main__":
    main()