From 848d953226cedc782f8949838698801458b1a829 Mon Sep 17 00:00:00 2001
From: Jactus <dw1920@nyu.edu>
Date: Fri, 16 Apr 2021 09:58:55 +0800
Subject: [PATCH] Update qlib logger

---
 qlib/log.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/qlib/log.py b/qlib/log.py
index 126acb9d2..ed050f6c9 100644
--- a/qlib/log.py
+++ b/qlib/log.py
@@ -11,6 +11,26 @@ from contextlib import contextmanager
 
 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):
     """
@@ -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