up
This commit is contained in:
@@ -1,56 +1,80 @@
|
||||
"""
|
||||
A simple log mechanism styled after PEP 282.
|
||||
"""A simple log mechanism styled after PEP 282."""
|
||||
|
||||
Retained for compatibility and should not be used.
|
||||
"""
|
||||
# The class here is styled after PEP 282 so that it could later be
|
||||
# replaced with a standard Python logging implementation.
|
||||
|
||||
import logging
|
||||
import warnings
|
||||
import sys
|
||||
|
||||
from ._log import log as _global_log
|
||||
DEBUG = 1
|
||||
INFO = 2
|
||||
WARN = 3
|
||||
ERROR = 4
|
||||
FATAL = 5
|
||||
|
||||
DEBUG = logging.DEBUG
|
||||
INFO = logging.INFO
|
||||
WARN = logging.WARN
|
||||
ERROR = logging.ERROR
|
||||
FATAL = logging.FATAL
|
||||
|
||||
class Log:
|
||||
def __init__(self, threshold=WARN):
|
||||
self.threshold = threshold
|
||||
|
||||
def _log(self, level, msg, args):
|
||||
if level not in (DEBUG, INFO, WARN, ERROR, FATAL):
|
||||
raise ValueError('%s wrong log level' % str(level))
|
||||
|
||||
if level >= self.threshold:
|
||||
if args:
|
||||
msg = msg % args
|
||||
if level in (WARN, ERROR, FATAL):
|
||||
stream = sys.stderr
|
||||
else:
|
||||
stream = sys.stdout
|
||||
try:
|
||||
stream.write('%s\n' % msg)
|
||||
except UnicodeEncodeError:
|
||||
# emulate backslashreplace error handler
|
||||
encoding = stream.encoding
|
||||
msg = msg.encode(encoding, "backslashreplace").decode(encoding)
|
||||
stream.write('%s\n' % msg)
|
||||
stream.flush()
|
||||
|
||||
def log(self, level, msg, *args):
|
||||
self._log(level, msg, args)
|
||||
|
||||
def debug(self, msg, *args):
|
||||
self._log(DEBUG, msg, args)
|
||||
|
||||
def info(self, msg, *args):
|
||||
self._log(INFO, msg, args)
|
||||
|
||||
def warn(self, msg, *args):
|
||||
self._log(WARN, msg, args)
|
||||
|
||||
def error(self, msg, *args):
|
||||
self._log(ERROR, msg, args)
|
||||
|
||||
def fatal(self, msg, *args):
|
||||
self._log(FATAL, msg, args)
|
||||
|
||||
|
||||
_global_log = Log()
|
||||
log = _global_log.log
|
||||
debug = _global_log.debug
|
||||
info = _global_log.info
|
||||
warn = _global_log.warning
|
||||
warn = _global_log.warn
|
||||
error = _global_log.error
|
||||
fatal = _global_log.fatal
|
||||
|
||||
|
||||
def set_threshold(level):
|
||||
orig = _global_log.level
|
||||
_global_log.setLevel(level)
|
||||
return orig
|
||||
# return the old threshold for use from tests
|
||||
old = _global_log.threshold
|
||||
_global_log.threshold = level
|
||||
return old
|
||||
|
||||
|
||||
def set_verbosity(v):
|
||||
if v <= 0:
|
||||
set_threshold(logging.WARN)
|
||||
set_threshold(WARN)
|
||||
elif v == 1:
|
||||
set_threshold(logging.INFO)
|
||||
set_threshold(INFO)
|
||||
elif v >= 2:
|
||||
set_threshold(logging.DEBUG)
|
||||
|
||||
|
||||
class Log(logging.Logger):
|
||||
"""distutils.log.Log is deprecated, please use an alternative from `logging`."""
|
||||
|
||||
def __init__(self, threshold=WARN):
|
||||
warnings.warn(Log.__doc__) # avoid DeprecationWarning to ensure warn is shown
|
||||
super().__init__(__name__, level=threshold)
|
||||
|
||||
@property
|
||||
def threshold(self):
|
||||
return self.level
|
||||
|
||||
@threshold.setter
|
||||
def threshold(self, level):
|
||||
self.setLevel(level)
|
||||
|
||||
warn = logging.Logger.warning
|
||||
set_threshold(DEBUG)
|
||||
|
||||
Reference in New Issue
Block a user