添加日志

common/utils/logger.py

+# _*_ coding:utf-8 _*_
+__author__ = 'fanxun'
+__data__ = "2021-05-10 15:47"
+
+import os
+import logging
+from logging.handlers import TimedRotatingFileHandler
+from config import BaseConfig
+
+# 日志级别
+CRITICAL = 50
+FATAL = CRITICAL
+ERROR = 40
+WARNING = 30
+WARN = WARNING
+INFO = 20
+DEBUG = 10
+NOTSET = 0
+
+LOG_PATH = BaseConfig.log_path
+
+class LogHandler(logging.Logger):
+    """
+    日志处理
+    """
+    def __init__(self, name, level=DEBUG, stream=True, file=True):
+        self.name = name
+        self.level = level
+        logging.Logger.__init__(self, self.name, level=level)
+        if stream:
+            self.__setStreamHandler__()
+        if file:
+            self.__setFileHandler__()
+
+    def __setFileHandler__(self, level=None):
+        """
+        输出到文件
+        """
+        file_name = os.path.join(LOG_PATH, '{name}.log'.format(name=self.name))
+        # 设置日志回滚, 保存在log目录, 一天保存一个文件, 保留15天
+        file_handler = TimedRotatingFileHandler(filename=file_name, when='D', interval=1, backupCount=15)
+        file_handler.suffix = '%Y%m%d.log'
+        if not level:
+            file_handler.setLevel(self.level)
+        else:
+            file_handler.setLevel(level)
+        formatter = logging.Formatter('%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s')
+
+        file_handler.setFormatter(formatter)
+        self.file_handler = file_handler
+        self.addHandler(file_handler)
+
+    def __setStreamHandler__(self, level=None):
+        """
+        输出到控制台
+        """
+        stream_handler = logging.StreamHandler()
+        formatter = logging.Formatter('%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s')
+        stream_handler.setFormatter(formatter)
+        if not level:
+            stream_handler.setLevel(self.level)
+        else:
+            stream_handler.setLevel(level)
+        self.addHandler(stream_handler)
+
+    def resetName(self, name):
+        self.name = name
+        self.removeHandler(self.file_handler)
+        self.__setFileHandler__()
+
+
+if __name__ == '__main__':
+    log = LogHandler('test')
+    log.info('this is a test msg')
+    log.error('hoo')
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+

config.py

@@ -16,6 +16,9 @@ class BaseConfig():
     # 钉钉相关
     webhook = ''
 
+    # 日志相关
+    log_path = r''  # 日志路径
+
 
 class BMCConfig(BaseConfig):
     """斑马信用app的配置类"""