Skip to content

Instantly share code, notes, and snippets.

@vsajip

vsajip/ansistrm.py

Created December 29, 2010 11:14
Download ZIP

Revisions

  1. vsajip revised this gist Apr 15, 2012. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion ansistrm.py
    View file
    Original file line number Diff line number Diff line change
    @@ -1,5 +1,5 @@
    #
    # Copyright (C) 2010-2012 Vinay Sajip. All rights reserved.
    # Copyright (C) 2010-2012 Vinay Sajip. All rights reserved. Licensed under the new BSD license.
    #
    import ctypes
    import logging
  2. vsajip revised this gist Apr 8, 2012. 1 changed file with 0 additions and 1 deletion.
    1 change: 0 additions & 1 deletion ansistrm.py
    View file
    Original file line number Diff line number Diff line change
    @@ -62,7 +62,6 @@ def emit(self, record):
    def output_colorized(self, message):
    self.stream.write(message)
    else:
    import ctypes
    import re
    ansi_esc = re.compile(r'\x1b\[((?:\d+)(?:;(?:\d+))*)m')

  3. vsajip revised this gist Apr 8, 2012. 1 changed file with 2 additions and 1 deletion.
    3 changes: 2 additions & 1 deletion ansistrm.py
    View file
    Original file line number Diff line number Diff line change
    @@ -1,6 +1,7 @@
    #
    # Copyright (C) 2010, 2011 Vinay Sajip. All rights reserved.
    # Copyright (C) 2010-2012 Vinay Sajip. All rights reserved.
    #
    import ctypes
    import logging
    import os

  4. vsajip revised this gist Jun 23, 2011. 1 changed file with 2 additions and 2 deletions.
    4 changes: 2 additions & 2 deletions ansistrm.py
    View file
    Original file line number Diff line number Diff line change
    @@ -1,7 +1,6 @@
    #
    # Copyright (C) 2010 Vinay Sajip. All rights reserved.
    # Copyright (C) 2010, 2011 Vinay Sajip. All rights reserved.
    #
    import ctypes
    import logging
    import os

    @@ -62,6 +61,7 @@ def emit(self, record):
    def output_colorized(self, message):
    self.stream.write(message)
    else:
    import ctypes
    import re
    ansi_esc = re.compile(r'\x1b\[((?:\d+)(?:;(?:\d+))*)m')

  5. vsajip revised this gist Dec 30, 2010. 1 changed file with 3 additions and 5 deletions.
    8 changes: 3 additions & 5 deletions ansistrm.py
    View file
    Original file line number Diff line number Diff line change
    @@ -63,7 +63,7 @@ def output_colorized(self, message):
    self.stream.write(message)
    else:
    import re
    ansi_esc = re.compile(r'\x1b\[((\d+)(;(\d+))*)m')
    ansi_esc = re.compile(r'\x1b\[((?:\d+)(?:;(?:\d+))*)m')

    nt_color_map = {
    0: 0x00, # black
    @@ -89,10 +89,9 @@ def output_colorized(self, message):
    text = parts.pop(0)
    if text:
    write(text)
    if len(parts) > 4:
    params = parts[0]
    if parts:
    params = parts.pop(0)
    if h is not None:
    parts = parts[4:]
    params = [int(p) for p in params.split(';')]
    color = 0
    for p in params:
    @@ -144,4 +143,3 @@ def main():

    if __name__ == '__main__':
    main()

  6. vsajip created this gist Dec 29, 2010.
    147 changes: 147 additions & 0 deletions ansistrm.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,147 @@
    #
    # Copyright (C) 2010 Vinay Sajip. All rights reserved.
    #
    import ctypes
    import logging
    import os

    class ColorizingStreamHandler(logging.StreamHandler):
    # color names to indices
    color_map = {
    'black': 0,
    'red': 1,
    'green': 2,
    'yellow': 3,
    'blue': 4,
    'magenta': 5,
    'cyan': 6,
    'white': 7,
    }

    #levels to (background, foreground, bold/intense)
    if os.name == 'nt':
    level_map = {
    logging.DEBUG: (None, 'blue', True),
    logging.INFO: (None, 'white', False),
    logging.WARNING: (None, 'yellow', True),
    logging.ERROR: (None, 'red', True),
    logging.CRITICAL: ('red', 'white', True),
    }
    else:
    level_map = {
    logging.DEBUG: (None, 'blue', False),
    logging.INFO: (None, 'black', False),
    logging.WARNING: (None, 'yellow', False),
    logging.ERROR: (None, 'red', False),
    logging.CRITICAL: ('red', 'white', True),
    }
    csi = '\x1b['
    reset = '\x1b[0m'

    @property
    def is_tty(self):
    isatty = getattr(self.stream, 'isatty', None)
    return isatty and isatty()

    def emit(self, record):
    try:
    message = self.format(record)
    stream = self.stream
    if not self.is_tty:
    stream.write(message)
    else:
    self.output_colorized(message)
    stream.write(getattr(self, 'terminator', '\n'))
    self.flush()
    except (KeyboardInterrupt, SystemExit):
    raise
    except:
    self.handleError(record)

    if os.name != 'nt':
    def output_colorized(self, message):
    self.stream.write(message)
    else:
    import re
    ansi_esc = re.compile(r'\x1b\[((\d+)(;(\d+))*)m')

    nt_color_map = {
    0: 0x00, # black
    1: 0x04, # red
    2: 0x02, # green
    3: 0x06, # yellow
    4: 0x01, # blue
    5: 0x05, # magenta
    6: 0x03, # cyan
    7: 0x07, # white
    }

    def output_colorized(self, message):
    parts = self.ansi_esc.split(message)
    write = self.stream.write
    h = None
    fd = getattr(self.stream, 'fileno', None)
    if fd is not None:
    fd = fd()
    if fd in (1, 2): # stdout or stderr
    h = ctypes.windll.kernel32.GetStdHandle(-10 - fd)
    while parts:
    text = parts.pop(0)
    if text:
    write(text)
    if len(parts) > 4:
    params = parts[0]
    if h is not None:
    parts = parts[4:]
    params = [int(p) for p in params.split(';')]
    color = 0
    for p in params:
    if 40 <= p <= 47:
    color |= self.nt_color_map[p - 40] << 4
    elif 30 <= p <= 37:
    color |= self.nt_color_map[p - 30]
    elif p == 1:
    color |= 0x08 # foreground intensity on
    elif p == 0: # reset to default color
    color = 0x07
    else:
    pass # error condition ignored
    ctypes.windll.kernel32.SetConsoleTextAttribute(h, color)

    def colorize(self, message, record):
    if record.levelno in self.level_map:
    bg, fg, bold = self.level_map[record.levelno]
    params = []
    if bg in self.color_map:
    params.append(str(self.color_map[bg] + 40))
    if fg in self.color_map:
    params.append(str(self.color_map[fg] + 30))
    if bold:
    params.append('1')
    if params:
    message = ''.join((self.csi, ';'.join(params),
    'm', message, self.reset))
    return message

    def format(self, record):
    message = logging.StreamHandler.format(self, record)
    if self.is_tty:
    # Don't colorize any traceback
    parts = message.split('\n', 1)
    parts[0] = self.colorize(parts[0], record)
    message = '\n'.join(parts)
    return message

    def main():
    root = logging.getLogger()
    root.setLevel(logging.DEBUG)
    root.addHandler(ColorizingStreamHandler())
    logging.debug('DEBUG')
    logging.info('INFO')
    logging.warning('WARNING')
    logging.error('ERROR')
    logging.critical('CRITICAL')

    if __name__ == '__main__':
    main()