def test_decrease_verbosity(self):
"""Make sure decrease_verbosity() respects default and custom levels."""
# Start from a known state.
set_level(logging.INFO)
assert get_level() == logging.INFO
# INFO -> WARNING.
decrease_verbosity()
assert get_level() == logging.WARNING
# WARNING -> ERROR.
decrease_verbosity()
assert get_level() == logging.ERROR
# ERROR -> CRITICAL.
decrease_verbosity()
assert get_level() == logging.CRITICAL
# CRITICAL -> CRITICAL.
decrease_verbosity()
assert get_level() == logging.CRITICAL
def test_increase_verbosity(self):
"""Make sure increase_verbosity() respects default and custom levels."""
# Start from a known state.
set_level(logging.INFO)
assert get_level() == logging.INFO
# INFO -> VERBOSE.
increase_verbosity()
assert get_level() == logging.VERBOSE
# VERBOSE -> DEBUG.
increase_verbosity()
assert get_level() == logging.DEBUG
# DEBUG -> NOTSET.
increase_verbosity()
assert get_level() == logging.NOTSET
# NOTSET -> NOTSET.
increase_verbosity()
assert get_level() == logging.NOTSET
def log_level(self):
"""
The logging level to configure in child processes (an integer).
Defaults to the current log level in the parent process at the point
when the worker processes are created.
"""
return coloredlogs.get_level()
def __init__(self, **options):
"""
Initialize a :class:`Promise` object.
The initializer arguments are the same as for
:class:`multiprocessing.Process`. The child
process is started automatically.
"""
super(Promise, self).__init__(**options)
self.log_level = coloredlogs.get_level()
self.queue = multiprocessing.Queue()
self.start()
def init_logger(self):
"""
get properly log warnning and log error function
"""
if self.args.log_level:
log_level = getattr(logging, self.args.log_level)
if coloredlogs:
coloredlogs.install(level=log_level, fmt=LOG_FMT)
else:
logging.basicConfig(level=log_level)
ch = logging.StreamHandler()
formatter = logging.Formatter(LOG_FMT)
ch.setFormatter(formatter)
elif coloredlogs:
coloredlogs.install(level='INFO', fmt=LOG_FMT)
if coloredlogs:
effective_level = coloredlogs.get_level()
else:
effective_level = logger.getEffectiveLevel()
# make sure warning and error display at any effective level
if effective_level > logging.WARNING:
self.warning = logger.critical
else:
self.warning = logger.warning
if effective_level > logging.ERROR:
self.error = logger.critical
else:
self.error = logger.error
self.info = logger.info
self.debug = logger.debug
self.exception = logger.exception
self.critical = logger.critical
def __enter__(self):
"""Customize the logging verbosity when entering the :keyword:`with` block."""
if self.original_level is None:
self.original_level = get_level()
set_level(self.level)
