diff options
Diffstat (limited to 'logutils/colorize.py')
-rw-r--r-- | logutils/colorize.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/logutils/colorize.py b/logutils/colorize.py index 3dfaad0..2c39639 100644 --- a/logutils/colorize.py +++ b/logutils/colorize.py @@ -1,10 +1,15 @@ # -# Copyright (C) 2010-2011 Vinay Sajip. All rights reserved. +# Copyright (C) 2010-2013 Vinay Sajip. All rights reserved. # import ctypes import logging import os +try: + unicode +except NameError: + unicode = None + class ColorizingStreamHandler(logging.StreamHandler): """ A stream handler which supports colorizing of console streams @@ -58,6 +63,9 @@ class ColorizingStreamHandler(logging.StreamHandler): try: message = self.format(record) stream = self.stream + if unicode and isinstance(message, unicode): + enc = getattr(stream, 'encoding', 'utf-8') + message = message.encode(enc, 'replace') if not self.is_tty: stream.write(message) else: |