import logging

class ColorFormatter(logging.Formatter):

    def color(self, level=None):
        codes = {\
            None:       (0, 0,   0),
            'DEBUG':    (0, 0,   2), # gris
            'INFO':     (0, 0,   0), # normal
            'WARNING':  (38,5, 208), # naranja
            'ERROR':    (0, 1,  31), # rojo
            'CRITICAL': (0, 1, 101), # negro, fondo rojo
            }
        return (chr(27)+'[%d;%d;%dm') % codes[level]

    def format(self, record):
        retval = logging.Formatter.format(self, record)
        return self.color(record.levelname) + retval + self.color()