1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-07 13:00:58 +08:00

Update qlib logger

This commit is contained in:
Jactus
2021-04-16 09:58:55 +08:00
parent a3a2b5ae0b
commit 848d953226

View File

@@ -11,6 +11,26 @@ from contextlib import contextmanager
from .config import C
class QlibLogger(Loggermeta=):
'''
Customized logger for Qlib.
'''
def __init__(self, module_name):
self.module_name = module_name
self.level = 0
@property
def logger(self):
logger = logging.getLogger(self.module_name)
logger.setLevel(self.level)
return logger
def setLevel(self, level):
self.level = level
def __getattr__(self, name):
return self.logger.__getattribute__(name)
def get_module_logger(module_name, level: Optional[int] = None):
"""
@@ -27,7 +47,7 @@ def get_module_logger(module_name, level: Optional[int] = None):
module_name = "qlib.{}".format(module_name)
# Get logger.
module_logger = logging.getLogger(module_name)
module_logger = QlibLogger(module_name)
module_logger.setLevel(level)
return module_logger