From b320dc118ded0e8325ea37a75a83630d32cc9ed7 Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Thu, 12 Jul 2012 14:55:54 +0200 Subject: [PATCH] Minor fix (recognizing if it's colorizing handler or not) --- lib/core/common.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/lib/core/common.py b/lib/core/common.py index 21672ea5a..268609f96 100644 --- a/lib/core/common.py +++ b/lib/core/common.py @@ -702,14 +702,15 @@ def setColor(message, bold=False): if level: kb.currentMessage = level - if bold: - retVal = colored(message, color=None, on_color=None, attrs=("bold",)) - elif hasattr(LOGGER_HANDLER, "level_map") and hasattr(kb, "currentMessage") and kb.currentMessage: - _ = LOGGER_HANDLER.level_map.get(logging.getLevelName(kb.currentMessage)) + if hasattr(LOGGER_HANDLER, "level_map"): # colorizing handler + if bold: + retVal = colored(message, color=None, on_color=None, attrs=("bold",)) + elif hasattr(kb, "currentMessage") and kb.currentMessage: + _ = LOGGER_HANDLER.level_map.get(logging.getLevelName(kb.currentMessage)) - if _: - background, foreground, bold = _ - retVal = colored(message, color=foreground, on_color="on_%s" % background if background else None, attrs=("bold",) if bold else None) + if _: + background, foreground, bold = _ + retVal = colored(message, color=foreground, on_color="on_%s" % background if background else None, attrs=("bold",) if bold else None) return retVal