CERBERUS/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2026-08-04 06:50:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Minor patch for colorization (multiple quoted strings in same line)

Browse source
This commit is contained in:
Miroslav Stampar 2018-08-09 16:21:04 +02:00
parent 9e00202823
commit 75a64245c5
3 changed files with 14 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

20
thirdparty/ansistrm/ansistrm.py vendored
View file

@ -181,16 +181,16 @@ class ColorizingStreamHandler(logging.StreamHandler):
reset = self.reset
message = message.replace(counter, ''.join((self.csi, str(self.color_map["yellow"] + 30), 'm', counter, reset)), 1)
match = re.search(r"'([^']+)'", message) # single-quoted
if level != "PAYLOAD" and match:
string = match.group(1)
if not message.endswith(self.reset):
reset = self.reset
elif self.bold in message: # bold
reset = self.reset + self.bold
else:
reset = self.reset
message = message.replace(string, ''.join((self.csi, str(self.color_map["white"] + 30), 'm', string, reset)), 1)
if level != "PAYLOAD":
for match in re.finditer(r"'([^']+)'", message): # single-quoted
string = match.group(1)
if not message.endswith(self.reset):
reset = self.reset
elif self.bold in message: # bold
reset = self.reset + self.bold
else:
reset = self.reset
message = message.replace(string, ''.join((self.csi, str(self.color_map["white"] + 30), 'm', string, reset)), 1)
else:
message = ''.join((self.csi, ';'.join(params), 'm', message, self.reset))