Changeset 190

Timestamp:

08/13/08 20:14:52 (4 months ago)

Author:

apdavison

Message:

Added some stuff to make logging easier to setup (NeuroTools.init_logging(filename)) and with fancier formatting (colours, indents), to make it easier to read on the console.

Files:

• trunk/src/__init__.py (modified) (1 diff)

trunk/src/__init__.py

@@ -9 +9 @@
-check_pytables_version()
-"""
+
+# Setup fancy logging
+
+red     = 0010; green  = 0020; yellow = 0030; blue = 0040;
+magenta = 0050; cyan   = 0060; bright = 0100
+try:
+    import ll.ansistyle
+    def colour(col,text):
+        return str(ll.ansistyle.Text(col,str(text)))
+except ImportError:
+    def colour(col,text):
+            return text
+        
+import logging
+
+# Add a header() level to logging
+logging.HEADER = 60
+logging.addLevelName(logging.HEADER, 'HEADER')
+
+root = logging.getLogger()
+
+def root_header(msg, *args, **kwargs):
+    if len(root.handlers) == 0:
+        basicConfig()
+    apply(root.header, (msg,)+args, kwargs)
+
+def logger_header(self, msg, *args, **kwargs):
+    if self.manager.disable >= logging.HEADER:
+        return
+    if logging.HEADER >= self.getEffectiveLevel():
+        apply(self._log, (logging.HEADER, msg, args), kwargs)
+
+logging.Logger.header = logger_header
+logging.header = root_header
+
+class FancyFormatter(logging.Formatter):
+    """
+    A log formatter that colours and indents the log message depending on the level.
+    """
+    
+    DEFAULT_COLOURS = {
+        'CRITICAL': bright+red,
+        'ERROR': red,
+        'WARNING': magenta,
+        'HEADER': bright+yellow,
+        'INFO': cyan,
+        'DEBUG': green
+    }
+    
+    DEFAULT_INDENTS = {
+        'CRITICAL': "",
+        'ERROR': "",
+        'WARNING': "",
+        'HEADER': "",
+        'INFO': "  ",
+        'DEBUG': "    ",
+    }
+    
+    def __init__(self, fmt=None, datefmt=None, colours=DEFAULT_COLOURS):
+        logging.Formatter.__init__(self, fmt, datefmt)
+        self._colours = colours
+        self._indents = FancyFormatter.DEFAULT_INDENTS
+    
+    def format(self, record):
+        s = logging.Formatter.format(self, record)
+        if record.levelname == "HEADER":
+            s = "=== %s ===" % s
+        if self._colours:
+            s = colour(self._colours[record.levelname], s)
+        return self._indents[record.levelname] + s
+
+
+def init_logging(filename, file_level=logging.INFO, console_level=logging.WARNING):
+    logging.basicConfig(level=file_level,
+                        format='%(asctime)s %(levelname)s %(message)s', # %(pathname)s %(module)s %(funcName)s',
+                        filename=filename,
+                        filemode='w')
+    console = logging.StreamHandler()
+    console.setLevel(console_level)
+    console.setFormatter(FancyFormatter('%(message)s'))
+    logging.getLogger('').addHandler(console)