Better colors on stack traces

This commit is contained in:
derv82
2018-04-19 02:50:25 -04:00
parent 4ef3236e4d
commit 1bbc7fefaf

View File

@@ -149,8 +149,8 @@ class Wifite(object):
from traceback import format_exc from traceback import format_exc
Color.p('\n{!} ') Color.p('\n{!} ')
err = format_exc().strip() err = format_exc().strip()
err = err.replace('\n', '\n{!} {C} ') err = err.replace('\n', '\n{W}{!} {W} ')
err = err.replace(' File', '{W}File') err = err.replace(' File', '{W}{D}File')
err = err.replace(' Exception: ', '{R}Exception: {O}') err = err.replace(' Exception: ', '{R}Exception: {O}')
Color.pl(err) Color.pl(err)
except KeyboardInterrupt: except KeyboardInterrupt:
@@ -181,8 +181,8 @@ class Wifite(object):
from traceback import format_exc from traceback import format_exc
Color.p('\n{!} ') Color.p('\n{!} ')
err = format_exc().strip() err = format_exc().strip()
err = err.replace('\n', '\n{!} {C} ') err = err.replace('\n', '\n{W}{!} {W} ')
err = err.replace(' File', '{W}File') err = err.replace(' File', '{W}{D}File')
err = err.replace(' Exception: ', '{R}Exception: {O}') err = err.replace(' Exception: ', '{R}Exception: {O}')
Color.pl(err) Color.pl(err)
except KeyboardInterrupt: except KeyboardInterrupt:
@@ -243,8 +243,8 @@ def run():
from traceback import format_exc from traceback import format_exc
Color.p('\n{!} ') Color.p('\n{!} ')
err = format_exc().strip() err = format_exc().strip()
err = err.replace('\n', '\n{!} {C} ') err = err.replace('\n', '\n{W}{!} {W} ')
err = err.replace(' File', '{W}File') err = err.replace(' File', '{W}{D}File')
err = err.replace(' Exception: ', '{R}Exception: {O}') err = err.replace(' Exception: ', '{R}Exception: {O}')
Color.pl(err) Color.pl(err)