liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-16 09:11:00 +08:00
Code Activity

Modify set_global_logger_level use of contextmanager

Browse Source
This commit is contained in:
zhupr
2021-05-12 17:58:39 +08:00
parent 9bd77bd89f
commit b8e64dc526
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
qlib/log.py
View File

@@ -165,8 +165,16 @@ class LogFilter(logging.Filter):
return allow return allow
@contextmanager
def set_global_logger_level(level: int): def set_global_logger_level(level: int):
_handler_level_map = {}
qlib_logger = logging.root.manager.loggerDict.get("qlib", None) qlib_logger = logging.root.manager.loggerDict.get("qlib", None)
if qlib_logger is not None: if qlib_logger is not None:
for _handler in qlib_logger.handlers: for _handler in qlib_logger.handlers:
_handler_level_map[_handler] = _handler.level
_handler.level = level _handler.level = level
try:
yield
finally:
for _handler, _level in _handler_level_map.items():
_handler.level = _level