ANSI colored Python logging — Gist

ANSI colored Python logging — Gist

import logging

from termcolor import colored

class ColorLog(object):

    colormap = dict(

        debug=dict(color='grey', attrs=['bold']),

        info=dict(color='white'),

        warn=dict(color='yellow', attrs=['bold']),

        warning=dict(color='yellow', attrs=['bold']),

        error=dict(color='red'),

        critical=dict(color='red', attrs=['bold']),

    )

    def __init__(self, logger):

        self._log = logger

    def __getattr__(self, name):

        if name in ['debug', 'info', 'warn', 'warning', 'error', 'critical']:

            return lambda s, *args: getattr(self._log, name)(

                colored(s, **self.colormap[name]), *args)

        return getattr(self._log, name)

log = ColorLog(logging.getLogger(__name__))

if __name__ == '__main__':

    log.setLevel(logging.DEBUG)

    stdout = logging.StreamHandler()

    stdout.setLevel(logging.DEBUG)

    log.addHandler(stdout)

    log.debug("booooring . . .")

    log.info("pleasing anecdote")

    log.warn("awkward utterance")

    log.error("drunken rudeness")