test/log.py

   1 #!/usr/bin/env python
   2
   3 import sys
   4 import os
   5 import logging
   6
   7 """ @var formatting delimiter consisting of '=' characters """
   8 double_line_delim = '=' * 78
   9 """ @var formatting delimiter consisting of '-' characters """
  10 single_line_delim = '-' * 78
  11
  12
  13 def colorize(msg, color):
  14     return color + msg + COLOR_RESET
  15
  16
  17 class ColorFormatter(logging.Formatter):
  18
  19     def init(self, fmt=None, datefmt=None):
  20         super(ColorFormatter, self).__init__(fmt, datefmt)
  21
  22     def format(self, record):
  23         message = super(ColorFormatter, self).format(record)
  24         if hasattr(record, 'color'):
  25             message = colorize(message, record.color)
  26         return message
  27 try:
  28     verbose = int(os.getenv("V", 0))
  29 except:
  30     verbose = 0
  31
  32 # 40 = ERROR, 30 = WARNING, 20 = INFO, 10 = DEBUG, 0 = NOTSET (all messages)
  33 if verbose >= 2:
  34     log_level = 10
  35 elif verbose == 1:
  36     log_level = 20
  37 else:
  38     log_level = 40
  39
  40 handler = logging.StreamHandler(sys.stdout)
  41 color_formatter = ColorFormatter(fmt='%(asctime)s,%(msecs)03d %(message)s',
  42                                  datefmt="%H:%M:%S")
  43 handler.setFormatter(color_formatter)
  44 handler.setLevel(log_level)
  45
  46 global_logger = logging.getLogger()
  47 global_logger.addHandler(handler)
  48
  49 scapy_logger = logging.getLogger("scapy.runtime")
  50 scapy_logger.setLevel(logging.ERROR)
  51
  52
  53 def get_logger(name):
  54     logger = logging.getLogger(name)
  55     logger.setLevel(logging.DEBUG)
  56     return logger
  57
  58
  59 def get_parallel_logger(stream):
  60     logger = logging.getLogger('parallel_logger_{}'.format(stream))
  61     logger.propagate = False
  62     logger.setLevel(logging.DEBUG)
  63     handler = logging.StreamHandler(stream)
  64     handler.setFormatter(color_formatter)
  65     handler.setLevel(log_level)
  66     logger.addHandler(handler)
  67     return logger
  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 = ''