Merge pull request #239 from ayalash/force_colorize

ColorizingStreamHandlerMixin: Allow force/forbid colors

logbook/more.py

@@ -329,6 +329,17 @@ class ColorizingStreamHandlerMixin(object):
 
     .. _`colorama`: https://pypi.python.org/pypi/colorama
     """
+    _use_color = None
+
+    def force_color(self):
+        """Force colorizing the stream (`should_colorize` will return True)
+        """
+        self._use_color = True
+
+    def forbid_color(self):
+        """Forbid colorizing the stream (`should_colorize` will return False)
+        """
+        self._use_color = False
 
     def should_colorize(self, record):
         """Returns `True` if colorizing should be applied to this
@@ -341,6 +352,8 @@ def should_colorize(self, record):
                 import colorama
             except ImportError:
                 return False
+        if self._use_color is not None:
+            return self._use_color
         isatty = getattr(self.stream, 'isatty', None)
         return isatty and isatty()
 

tests/test_more.py

@@ -31,11 +31,16 @@ def test_colorizing_support(logger):
     from logbook.more import ColorizedStderrHandler
 
     class TestColorizingHandler(ColorizedStderrHandler):
+        def __init__(self, *args, **kwargs):
+            super(TestColorizingHandler, self).__init__(*args, **kwargs)
+            self._obj_stream = StringIO()
+
+        @property
+        def stream(self):
+            return self._obj_stream
 
-        def should_colorize(self, record):
-            return True
-        stream = StringIO()
     with TestColorizingHandler(format_string='{record.message}') as handler:
+        handler.force_color()
         logger.error('An error')
         logger.warn('A warning')
         logger.debug('A debug message')
@@ -45,6 +50,15 @@ def should_colorize(self, record):
             '\x1b[33;01mA warning\x1b[39;49;00m',
             '\x1b[37mA debug message\x1b[39;49;00m']
 
+    with TestColorizingHandler(format_string='{record.message}') as handler:
+        handler.forbid_color()
+        logger.error('An error')
+        logger.warn('A warning')
+        logger.debug('A debug message')
+        lines = handler.stream.getvalue().rstrip('\n').splitlines()
+        assert lines == ['An error', 'A warning', 'A debug message']
+
+
 
 def test_tagged(default_handler):
     from logbook.more import TaggingLogger, TaggingHandler