From f219b3998084224f4fdd9afba2cb4d4a693399a5 Mon Sep 17 00:00:00 2001 From: Bernardo Damele Date: Tue, 10 Jul 2012 13:08:37 +0100 Subject: [PATCH] minor fix in case ctypes is not installed on Windows --- lib/core/log.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/lib/core/log.py b/lib/core/log.py index 7add4431e..1e3a883a7 100644 --- a/lib/core/log.py +++ b/lib/core/log.py @@ -17,8 +17,13 @@ logging.addLevelName(CUSTOM_LOGGING.TRAFFIC_OUT, "TRAFFIC OUT") logging.addLevelName(CUSTOM_LOGGING.TRAFFIC_IN, "TRAFFIC IN") LOGGER = logging.getLogger("sqlmapLog") -#LOGGER_HANDLER = logging.StreamHandler(sys.stdout) -LOGGER_HANDLER = ColorizingStreamHandler(sys.stdout) + +try: + import ctypes + LOGGER_HANDLER = ColorizingStreamHandler(sys.stdout) +except ImportError: + LOGGER_HANDLER = logging.StreamHandler(sys.stdout) + FORMATTER = logging.Formatter("\r[%(asctime)s] [%(levelname)s] %(message)s", "%H:%M:%S") LOGGER_HANDLER.setFormatter(FORMATTER)