sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-03-03 11:45:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix for a case when ColorizingStreamHandler is not used

Browse Source
This commit is contained in:
Miroslav Stampar 2012-07-11 20:36:32 +02:00
parent 72378d4f61
commit 43cac2212b
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lib/core/common.py
View File

@ -696,15 +696,14 @@ def singleTimeLogMessage(message, level=logging.INFO, flag=None):
logger.log(level, message)
def setColor(message, bold=False):
message = "[TRAFFIC IN] " + message
retVal = message
level = extractRegexResult(r"\A\s*\[(?P<result>[A-Z ]+)\]", message)
if hasattr(LOGGER_HANDLER, "level_map"):
_ = LOGGER_HANDLER.level_map.get(logging._levelNames.get(level))
if _:
background, foreground, bold = _
retVal = colored(message, color=foreground, on_color="on_%s" % background if background else None, attrs=("bold",) if bold else None)
else:
retVal = message
return retVal