sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-03-03 19:55:47 +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

13
lib/core/common.py
View File

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