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):
21 def init(self, fmt=None, datefmt=None):
22 super(ColorFormatter, self).__init__(fmt, datefmt)
24 def format(self, record):
25 message = super(ColorFormatter, self).format(record)
26 if hasattr(record, 'color'):
27 message = colorize(message, record.color)
31 # 40 = ERROR, 30 = WARNING, 20 = INFO, 10 = DEBUG, 0 = NOTSET (all messages)
32 if config.verbose >= 2:
34 elif config.verbose == 1:
39 handler = logging.StreamHandler(sys.stdout)
40 color_formatter = ColorFormatter(fmt='%(asctime)s,%(msecs)03d %(message)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'