Adding traffic in and out to color_map

This commit is contained in:
Miroslav Stampar
2012-07-11 20:42:18 +02:00
parent 43cac2212b
commit 15ee5310d9
2 changed files with 3 additions and 1 deletions

View File

@@ -21,6 +21,8 @@ LOGGER = logging.getLogger("sqlmapLog")
try:
import ctypes
LOGGER_HANDLER = ColorizingStreamHandler(sys.stdout)
LOGGER_HANDLER.level_map[logging.getLevelName("TRAFFIC OUT")] = (None, "cyan", True)
LOGGER_HANDLER.level_map[logging.getLevelName("TRAFFIC IN")] = (None, "grey", True)
except ImportError:
LOGGER_HANDLER = logging.StreamHandler(sys.stdout)