diff options
Diffstat (limited to '')
| -rw-r--r-- | portato/__init__.py | 48 |
1 files changed, 47 insertions, 1 deletions
diff --git a/portato/__init__.py b/portato/__init__.py index 4b1f42f..0166170 100644 --- a/portato/__init__.py +++ b/portato/__init__.py @@ -3,9 +3,55 @@ # File: portato/__init__.py # This file is part of the Portato-Project, a graphical portage-frontend. # -# Copyright (C) 2006 René 'Necoro' Neumann +# Copyright (C) 2006-2007 René 'Necoro' Neumann # This is free software. You may redistribute copies of it under the terms of # the GNU General Public License version 2. # There is NO WARRANTY, to the extent permitted by law. # # Written by René 'Necoro' Neumann <necoro@necoro.net> + +import logging +import sys +import os + +class OutputFormatter (logging.Formatter): + + colors = { + "blue" : 34, + "green" : 32, + "red" : 31, + "yellow": 33 + } + + def __init__(self, *args, **kwargs): + logging.Formatter.__init__(self, *args, **kwargs) + + for key, value in self.colors.iteritems(): + self.colors[key] = "\x1b[01;%02dm*\x1b[39;49;00m" % value + + def format (self, record): + string = logging.Formatter.format(self, record) + color = None + + if os.isatty(sys.stderr.fileno()): + if record.levelno <= logging.DEBUG: + color = self.colors["blue"] + elif record.levelno <= logging.INFO: + color = self.colors["green"] + elif record.levelno <= logging.WARNING: + color = self.colors["yellow"] + else: + color = self.colors["red"] + else: + color = "%s:" % record.levelname + + return "%s %s" % (color, string) + +# set the whole logging stuff +formatter = OutputFormatter("%(message)s (%(filename)s:%(lineno)s)") + +handler = logging.StreamHandler() +handler.setFormatter(formatter) +logging.getLogger().addHandler(handler) + +logging.getLogger().setLevel(logging.DEBUG) |