sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2024-11-26 03:23:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

colourize the whole log message

Browse Source
This commit is contained in:
Bernardo Damele 2012-07-11 12:18:48 +01:00
parent 31571e6e2d
commit ff6ca6fb1a
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
extra/ansistrm/ansistrm.py
View File

@ -125,12 +125,13 @@ class ColorizingStreamHandler(logging.StreamHandler):
def format(self, record): def format(self, record):
message = logging.StreamHandler.format(self, record) message = logging.StreamHandler.format(self, record)
if self.is_tty: #if self.is_tty:
# Don't colorize any traceback # # Don't colorize any traceback
parts = message.split('\n', 1) # parts = message.split('\n', 1)
parts[0] = self.colorize(parts[0], record) # parts[0] = self.colorize(parts[0], record)
message = '\n'.join(parts) # message = '\n'.join(parts)
return message #return message
return self.colorize(message, record)
def main(): def main():
root = logging.getLogger() root = logging.getLogger()