Lib/distutils/log.py

   1 """A simple log mechanism styled after PEP 282."""
   2
   3 # The class here is styled after PEP 282 so that it could later be
   4 # replaced with a standard Python logging implementation.
   5
   6 DEBUG = 1
   7 INFO = 2
   8 WARN = 3
   9 ERROR = 4
  10 FATAL = 5
  11
  12 import sys
  13
  14 class Log:
  15
  16     def __init__(self, threshold=WARN):
  17         self.threshold = threshold
  18
  19     def _log(self, level, msg, args):
  20         if level not in (DEBUG, INFO, WARN, ERROR, FATAL):
  21             raise ValueError('%s wrong log level' % str(level))
  22
  23         if level >= self.threshold:
  24             if args:
  25                 msg = msg % args
  26             if level in (WARN, ERROR, FATAL):
  27                 stream = sys.stderr
  28             else:
  29                 stream = sys.stdout
  30             stream.write('%s\n' % msg)
  31             stream.flush()
  32
  33     def log(self, level, msg, *args):
  34         self._log(level, msg, args)
  35
  36     def debug(self, msg, *args):
  37         self._log(DEBUG, msg, args)
  38
  39     def info(self, msg, *args):
  40         self._log(INFO, msg, args)
  41
  42     def warn(self, msg, *args):
  43         self._log(WARN, msg, args)
  44
  45     def error(self, msg, *args):
  46         self._log(ERROR, msg, args)
  47
  48     def fatal(self, msg, *args):
  49         self._log(FATAL, msg, args)
  50
  51 _global_log = Log()
  52 log = _global_log.log
  53 debug = _global_log.debug
  54 info = _global_log.info
  55 warn = _global_log.warn
  56 error = _global_log.error
  57 fatal = _global_log.fatal
  58
  59 def set_threshold(level):
  60     # return the old threshold for use from tests
  61     old = _global_log.threshold
  62     _global_log.threshold = level
  63     return old
  64
  65 def set_verbosity(v):
  66     if v <= 0:
  67         set_threshold(WARN)
  68     elif v == 1:
  69         set_threshold(INFO)
  70     elif v >= 2:
  71         set_threshold(DEBUG)