def test_multiprocess_debug_mode():
"""get_debug_mode() and set_debug_mode() should preserve between processes"""
set_debug_mode(True)
assert get_debug_mode()
process = multiprocessing.Process(
target=assert_debug_mode, args=[True], daemon=True
)
process.start()
process.join()
assert process.exitcode == 0
set_debug_mode(False)
assert not get_debug_mode()
process = multiprocessing.Process(
target=assert_debug_mode, args=[False], daemon=True
)
process.start()
process.join()
assert process.exitcode == 0
del os.environ["BENTOML_DEBUG"]
def test_set_debug_mode():
set_debug_mode(True)
assert get_debug_mode()
set_debug_mode(False)
assert not get_debug_mode()
del os.environ["BENTOML_DEBUG"]
def wrapper(quiet, verbose, *args, **kwargs):
if quiet:
configure_logging(logging.ERROR)
if verbose:
logger.warning(
"The bentoml command option `--verbose/--debug` is ignored when"
"the `--quiet` flag is also in use")
elif verbose:
set_debug_mode(True)
return func(*args, **kwargs)
