sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-02-16 19:40:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix for an Issue #140

Browse Source
This commit is contained in:
Miroslav Stampar 2012-08-07 10:57:29 +02:00
parent fec8a5cc9d
commit 31ceb0cb6c
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/core/common.py
View File

@ -701,7 +701,7 @@ def setColor(message, bold=False):
retVal = message retVal = message
level = extractRegexResult(r"\[(?P<result>[A-Z ]+)\]", message) or kb.get("stickyLevel") level = extractRegexResult(r"\[(?P<result>[A-Z ]+)\]", message) or kb.get("stickyLevel")
if message and hasattr(LOGGER_HANDLER, "level_map"): # colorizing handler if message and getattr(LOGGER_HANDLER, "is_tty", False): # colorizing handler
if bold: if bold:
retVal = colored(message, color=None, on_color=None, attrs=("bold",)) retVal = colored(message, color=None, on_color=None, attrs=("bold",))
elif level: elif level:

4
thirdparty/ansistrm/ansistrm.py vendored
View File

@ -51,6 +51,8 @@ class ColorizingStreamHandler(logging.StreamHandler):
stream = self.stream stream = self.stream
if not self.is_tty: if not self.is_tty:
if message and message[0] == "\r":
message = message[1:]
stream.write(message) stream.write(message)
else: else:
self.output_colorized(message) self.output_colorized(message)
@ -122,7 +124,7 @@ class ColorizingStreamHandler(logging.StreamHandler):
ctypes.windll.kernel32.SetConsoleTextAttribute(h, color) ctypes.windll.kernel32.SetConsoleTextAttribute(h, color)
def colorize(self, message, record): def colorize(self, message, record):
if record.levelno in self.level_map: if record.levelno in self.level_map and self.is_tty:
bg, fg, bold = self.level_map[record.levelno] bg, fg, bold = self.level_map[record.levelno]
params = [] params = []