def set_loggers(config):
from rqalpha.utils.logger import user_log, user_system_log, system_log
from rqalpha.utils.logger import init_logger
from rqalpha.utils import logger
extra_config = config.extra
init_logger()
for log in [system_log, user_system_log]:
log.level = getattr(logbook, config.extra.log_level.upper(),
logbook.NOTSET)
user_log.level = logbook.DEBUG
if extra_config.log_level.upper() != "NONE":
if extra_config.user_log_disabled:
user_log.disable()
else:
user_log.enable()
if extra_config.user_system_log_disabled:
user_system_log.disable()
else:
user_system_log.enable()
for logger_name, level in extra_config.logger:
getattr(logger, logger_name).level = getattr(logbook, level.upper())
def run_with_user_log_disabled(disabled=True):
from rqalpha.utils.logger import user_log
if disabled:
user_log.disable()
try:
yield
finally:
if disabled:
user_log.enable()
def run_with_user_log_disabled(disabled=True):
from rqalpha.utils.logger import user_log
if disabled:
user_log.disable()
try:
yield
finally:
if disabled:
user_log.enable()
