pgtoolkit/coloredformatter.py
author Radek Brich <radek.brich@devl.cz>
Sat, 29 Sep 2012 13:53:54 +0200
changeset 49 08e4dfe1b0cb
parent 13 16dc5dec9c36
child 78 64c62ac8f65d
permissions -rw-r--r--
Add test for MyManager (enable only when MySQLdb is available). Configure tests using pgtoolkit.conf (same as used by other executables).

import logging

from pgtoolkit import highlight


class ColoredFormatter(logging.Formatter):
    def __init__(self, fmt, datefmt):
        logging.Formatter.__init__(self, fmt, datefmt)

    def format(self, record):
        color = ()
        if record.levelno == 21:
            color = (2, None)
        if record.levelno == logging.ERROR and not record.exc_info is None:
            color = (11, 1)
        if color:
            origmsg = record.msg
            f,b = color
            record.msg = highlight(1, f, b) + record.msg + highlight(0)
            res = logging.Formatter.format(self, record)
            record.msg = origmsg
        else:
            res = logging.Formatter.format(self, record)
        return res