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