coloredformatter.py

## 
##  coloredformatter.py
## 
##   Created on: Oct 21, 2012
##       Author: martinez
##

import copy
import logging

CODE={
  'ENDC':0,  # RESET COLOR
  'BOLD':1,
  'UNDERLINE':4,
  'BLINK':5,
  'INVERT':7,
  'CONCEALD':8,
  'STRIKE':9,
  'GREY30':90,
  'GREY40':2,
  'GREY65':37,
  'GREY70':97,
  'GREY20_BG':40,
  'GREY33_BG':100,
  'GREY80_BG':47,
  'GREY93_BG':107,
  'DARK_RED':31,
  'RED':91,
  'RED_BG':41,
  'LIGHT_RED_BG':101,
  'DARK_YELLOW':33,
  'YELLOW':93,
  'YELLOW_BG':43,
  'LIGHT_YELLOW_BG':103,
  'DARK_BLUE':34,
  'BLUE':94,
  'BLUE_BG':44,
  'LIGHT_BLUE_BG':104,
  'DARK_MAGENTA':35,
  'PURPLE':95,
  'MAGENTA_BG':45,
  'LIGHT_PURPLE_BG':105,
  'DARK_CYAN':36,
  'AUQA':96,
  'CYAN_BG':46,
  'LIGHT_AUQA_BG':106,
  'DARK_GREEN':32,
  'GREEN':92,
  'GREEN_BG':42,
  'LIGHT_GREEN_BG':102,
  'BLACK':30,
}

def termcode(num):
  return '\033[%sm'%num

def colorstr(astr,color):
  return termcode(CODE[color])+astr+termcode(CODE['ENDC'])

class ColoredFormatter(logging.Formatter):
  # A variant of code found at http://stackoverflow.com/questions/384076/how-can-i-make-the-python-logging-output-to-be-colored
  LEVELCOLOR = {
    'DEBUG': 'BLUE',
    'INFO': 'GREEN',
    'WARNING': 'PURPLE',
    'ERROR': 'RED',
    'CRITICAL': 'RED_BG',
  }

  def __init__(self, msg):
    logging.Formatter.__init__(self, msg)

  def format(self, record):
    record = copy.copy(record)
    levelname = record.levelname
    if levelname in self.LEVELCOLOR:
      record.levelname = colorstr(levelname,self.LEVELCOLOR[levelname])
      record.name      = colorstr(record.name,'BOLD')
      record.msg       = colorstr(record.msg,self.LEVELCOLOR[levelname])
    return logging.Formatter.format(self, record)