单例模式实现 logging 日志实例化

淩亂°似流年 2022-11-04 05:28 140阅读 0赞

#### 单例模式实现 logging 日志实例化 ####

#!/usr/bin/python
    # -*- coding: UTF-8 -*-
    
    import logging
    import logging.handlers
    import os
    
    
    class Logger(object):
        __flag = None
    
        def __new__(cls, *args, **kwargs):
            if not cls.__flag:
                cls.__flag = super().__new__(cls)
            return cls.__flag
    
        def __init__(self, log_path):
            if 'logger' not in self.__dict__:
                logger = logging.getLogger()
                logger.setLevel(logging.INFO)
                formatter = logging.Formatter('%(asctime)s - %(process)d-%(threadName)s - '
                                              '%(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s')
                console_handler = logging.StreamHandler()
                console_handler.setFormatter(formatter)
                logger.addHandler(console_handler)
                file_handler = logging.handlers.RotatingFileHandler(
                    log_path, maxBytes=10485760, backupCount=5, encoding="utf-8")
                file_handler.setFormatter(formatter)
                logger.addHandler(file_handler)
                self.logger = logger
    
        def return_logger(self):
            return self.logger
    
    
    def get_logger(log_dir=None, log_filename=None):
        log_dir = log_dir if log_dir else os.path.join(os.getcwd(), "log")
        if not os.path.exists(log_dir):
            os.makedirs(log_dir, exist_ok=True)
        log_filename = log_filename if log_filename else 'log.log'
        log_path = os.path.join(log_dir, log_filename)
        return Log(log_path).return_logger()
    
    
    logger = get_logger(log_filename='shzdsylog.log')
    
    if __name__ == '__main__':
        logger.info('测试成功')
        logger.warning('测试成功')
        logger.error('测试成功')
        logger.critical('测试成功')