mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-17 17:34:35 +08:00
Update qlib logger
This commit is contained in:
22
qlib/log.py
22
qlib/log.py
@@ -11,6 +11,26 @@ from contextlib import contextmanager
|
|||||||
|
|
||||||
from .config import C
|
from .config import C
|
||||||
|
|
||||||
|
class QlibLogger(Logger,meta=):
|
||||||
|
'''
|
||||||
|
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):
|
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)
|
module_name = "qlib.{}".format(module_name)
|
||||||
# Get logger.
|
# Get logger.
|
||||||
module_logger = logging.getLogger(module_name)
|
module_logger = QlibLogger(module_name)
|
||||||
module_logger.setLevel(level)
|
module_logger.setLevel(level)
|
||||||
return module_logger
|
return module_logger
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user