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 = '=' * 70
   9 """ @var formatting delimiter consisting of '-' characters """
  10 single_line_delim = '-' * 70
  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 handler.setFormatter(ColorFormatter(fmt='%(asctime)s,%(msecs)03d %(message)s',
  42                                     datefmt="%H:%M:%S"))
  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 getLogger(name):
  53     logger = logging.getLogger(name)
  54     logger.setLevel(logging.DEBUG)
  55     return logger
  56
  57 # Static variables to store color formatting strings.
  58 #
  59 # These variables (RED, GREEN, YELLOW and LPURPLE) are used to configure
  60 # the color of the text to be printed in the terminal. Variable COLOR_RESET
  61 # is used to revert the text color to the default one.
  62 if hasattr(sys.stdout, 'isatty') and sys.stdout.isatty():
  63     RED = '\033[91m'
  64     GREEN = '\033[92m'
  65     YELLOW = '\033[93m'
  66     LPURPLE = '\033[94m'
  67     COLOR_RESET = '\033[0m'
  68 else:
  69     RED = ''
  70     GREEN = ''
  71     YELLOW = ''
  72     LPURPLE = ''
  73     COLOR_RESET = ''