7 from config import config
9 """ @var formatting delimiter consisting of '=' characters """
10 double_line_delim = "=" * 78
11 """ @var formatting delimiter consisting of '-' characters """
12 single_line_delim = "-" * 78
15 def colorize(msg, color):
16 return f"{color}{msg}{COLOR_RESET}"
19 class ColorFormatter(logging.Formatter):
20 def init(self, fmt=None, datefmt=None):
21 super(ColorFormatter, self).__init__(fmt, datefmt)
23 def format(self, record):
24 message = super(ColorFormatter, self).format(record)
25 if hasattr(record, "color"):
26 message = colorize(message, record.color)
30 # 40 = ERROR, 30 = WARNING, 20 = INFO, 10 = DEBUG, 0 = NOTSET (all messages)
31 if config.verbose >= 2:
33 elif config.verbose == 1:
38 handler = logging.StreamHandler(sys.stdout)
39 color_formatter = ColorFormatter(
40 fmt="%(asctime)s,%(msecs)03d %(message)s", datefmt="%H:%M:%S"
42 handler.setFormatter(color_formatter)
43 handler.setLevel(log_level)
45 global_logger = logging.getLogger()
46 global_logger.addHandler(handler)
48 scapy_logger = logging.getLogger("scapy.runtime")
49 scapy_logger.setLevel(logging.ERROR)
53 logger = logging.getLogger(name)
54 logger.setLevel(logging.DEBUG)
58 def get_parallel_logger(stream):
59 logger = logging.getLogger("parallel_logger_{!s}".format(stream))
60 logger.propagate = False
61 logger.setLevel(logging.DEBUG)
62 handler = logging.StreamHandler(stream)
63 handler.setFormatter(color_formatter)
64 handler.setLevel(log_level)
65 logger.addHandler(handler)
69 # Static variables to store color formatting strings.
71 # These variables (RED, GREEN, YELLOW and LPURPLE) are used to configure
72 # the color of the text to be printed in the terminal. Variable COLOR_RESET
73 # is used to revert the text color to the default one.
74 if hasattr(sys.stdout, "isatty") and sys.stdout.isatty():
79 COLOR_RESET = "\033[0m"