test/log.py

   1 #!/usr/bin/env python3
   2
   3 import sys
   4 import os
   5 import logging
   6
   7 from config import config
   8
   9 """ @var formatting delimiter consisting of '=' characters """
  10 double_line_delim = "=" * 78
  11 """ @var formatting delimiter consisting of '-' characters """
  12 single_line_delim = "-" * 78
  13
  14
  15 def colorize(msg, color):
  16     return f"{color}{msg}{COLOR_RESET}"
  17
  18
  19 class ColorFormatter(logging.Formatter):
  20     def init(self, fmt=None, datefmt=None):
  21         super(ColorFormatter, self).__init__(fmt, datefmt)
  22
  23     def format(self, record):
  24         message = super(ColorFormatter, self).format(record)
  25         if hasattr(record, "color"):
  26             message = colorize(message, record.color)
  27         return message
  28
  29
  30 # 40 = ERROR, 30 = WARNING, 20 = INFO, 10 = DEBUG, 0 = NOTSET (all messages)
  31 if config.verbose >= 2:
  32     log_level = 10
  33 elif config.verbose == 1:
  34     log_level = 20
  35 else:
  36     log_level = 40
  37
  38 handler = logging.StreamHandler(sys.stdout)
  39 color_formatter = ColorFormatter(
  40     fmt="%(asctime)s,%(msecs)03d %(message)s", datefmt="%H:%M:%S"
  41 )
  42 handler.setFormatter(color_formatter)
  43 handler.setLevel(log_level)
  44
  45 global_logger = logging.getLogger()
  46 global_logger.addHandler(handler)
  47
  48 scapy_logger = logging.getLogger("scapy.runtime")
  49 scapy_logger.setLevel(logging.ERROR)
  50
  51
  52 def get_logger(name):
  53     logger = logging.getLogger(name)
  54     logger.setLevel(logging.DEBUG)
  55     return logger
  56
  57
  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)
  66     return logger
  67
  68
  69 # Static variables to store color formatting strings.
  70 #
  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():
  75     RED = "\033[91m"
  76     GREEN = "\033[92m"
  77     YELLOW = "\033[93m"
  78     LPURPLE = "\033[94m"
  79     COLOR_RESET = "\033[0m"
  80 else:
  81     RED = ""
  82     GREEN = ""
  83     YELLOW = ""
  84     LPURPLE = ""
  85     COLOR_RESET = ""