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 that ugly red blank line in CRITICAL messages

Browse Source
This commit is contained in:
Miroslav Stampar 2012-07-11 20:49:22 +02:00
parent 15ee5310d9
commit e673a57311
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
extra/ansistrm/ansistrm.py
View File

@ -3,6 +3,7 @@
# #
import logging import logging
import os import os
import re
class ColorizingStreamHandler(logging.StreamHandler): class ColorizingStreamHandler(logging.StreamHandler):
# color names to indices # color names to indices
@ -119,8 +120,13 @@ class ColorizingStreamHandler(logging.StreamHandler):
if bold: if bold:
params.append('1') params.append('1')
if params: if params:
message = ''.join((self.csi, ';'.join(params), if message.lstrip() != message:
'm', message, self.reset)) prefix = re.search(r"\s+", message).group(0)
message = message[len(prefix):]
else:
prefix = ""
message = "%s%s" % (prefix, ''.join((self.csi, ';'.join(params),
'm', message, self.reset)))
return message return message
def format(self, record): def format(self, record):