日志记录类

日志记录类

import logging
import os


class LogRecorder:
    """日志记录类 单例"""
    _instance = None

    def __new__(cls, *args, **kwargs):
        if not cls._instance:
            cls._instance = object.__new__(cls)
        return cls._instance

    def __init__(self, module_name, log_path):
        self.logger = logging.getLogger(module_name)
        self.logger.setLevel(logging.DEBUG)
        format_str = '%(asctime)s - %(levelname)s - %(filename)s:%(lineno)s - %(funcName)s - %(message)s'
        self.formatter = logging.Formatter(format_str)
        self.add_file_handler(log_path)
        self.add_console_handler()

    def add_file_handler(self, log_path):
        if os.path.exists(log_path):
            os.remove(log_path)
        # 文件记录
        file_handler = logging.FileHandler(log_path)
        file_handler.setLevel(logging.DEBUG)
        file_handler.setFormatter(self.formatter)
        self.logger.addHandler(file_handler)

    def add_console_handler(self):
        # 控制台记录
        console_handler = logging.StreamHandler()
        console_handler.setLevel(logging.DEBUG)
        console_handler.setFormatter(self.formatter)
        self.logger.addHandler(console_handler)

    def get_logger(self):
        return self.logger


log_recorder = LogRecorder("cnblogs", "/root/cnblogs.log")
logger = log_recorder.get_logger()