atgfilter.py

#!/usr/bin/env python
import sys
import logging
import re

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"

# patterns = {'(Include)' : BLUE,
#             '(<IfModule.*>)': MAGENTA,
#             '(configuration\s\W?)': RED,
#             '(?<=ErrorDocument)\s(\d\d\d)': CYAN,
#             '(?<=ErrorDocument\s\d\d\d)\s(.*)': GREEN,
#             '(ErrorDocument)': YELLOW,
#             }

patterns = {'(\d\d\:\d\d\:\d\d\,\d\d\d)' : (BLUE,None),
            '(?<=\d\d\:\d\d\:\d\d\,\d\d\d)\s+(ERROR)\s+' : (RED,None),
            '(?<=\d\d\:\d\d\:\d\d\,\d\d\d)\s+(DEBUG)\s+' : (GREEN,None),
            '(?<=\d\d\:\d\d\:\d\d\,\d\d\d)\s+(INFO)\s+' : (BLUE,None),
            '(?<=\d\d\:\d\d\:\d\d\,\d\d\d)\s+(WARN)\s+' : (MAGENTA,None),
            '(\[.*?\])' : (YELLOW,None),
            '\s+(at\s.*)' : (RED,None),
            }

def colorize(str,color,bg_color=None,bold = False):
    color_code = COLOR_SEQ % (30 + color)
    if bg_color:
        bg_color_code = COLOR_SEQ % (40 + bg_color)
    colored = "%s%s%s%s%s" % (BOLD_SEQ if bold else '', 
                              color_code,
                              bg_color_code if bg_color else '', 
                              str,
                              RESET_SEQ)
    return colored


if __name__ == '__main__':
    while 1:
        line = sys.stdin.readline()
        per_line_matches = {}
        for p in patterns:
            match = re.search(p,line)
            if match:
                try:
                    group = match.groups()[0]
                    per_line_matches[group] = colorize(group, patterns[p][0], bg_color=patterns[p][1])
                except IndexError:
                    pass
        for m in per_line_matches.keys():
            line = line.replace(m,per_line_matches[m])
        if not line:
            break
        sys.stdout.write(line)