helper function set_color_formatter()

1 files changed, 15 insertions, 1 deletions

diff --git a/logging_ext.py b/logging_ext.py
index 08a1a97..dee73a6 100644
--- a/logging_ext.py
+++ b/logging_ext.py
@@ -1,4 +1,4 @@
-# -*- encoding: iso-8859-1 -*-
+# -*- coding: iso-8859-1 -*-
 # Copyright (c) 2006 LOGILAB S.A. (Paris, FRANCE).
 # http://www.logilab.fr/ -- mailto:contact@logilab.fr
 #
@@ -24,6 +24,10 @@ import logging
 
 from logilab.common.textutils import colorize_ansi
 
+def xxx_cyan(record):
+    if 'XXX' in record.message:
+        return 'cyan'
+
 class ColorFormatter(logging.Formatter):
     """
     A color Formatter for the logging standard module.
@@ -58,3 +62,13 @@ class ColorFormatter(logging.Formatter):
                 if color: 
                     return colorize_ansi(msg, color)
         return msg
+
+def set_color_formatter(logger=None):
+    if logger is None:
+        logger = logging.getLogger()
+        if not logger.handlers:
+            logging.basicConfig()
+    format_msg = logger.handlers[0].formatter._fmt
+    fmt = ColorFormatter(format_msg)
+    fmt.colorfilters.append(xxx_cyan)
+    logger.handlers[0].setFormatter(fmt)