python 封装日志,并删除过期日志

• 生成的日志结构
  
• 封装日志模块,并实现clean_log函数,删除上月日志

import logging
import os
import shutil
import time


class Log:
    def __init__(self, name='apipro'):
        # 日志文件基本配置
        # 设置日志记录器名称为 apipro，并返回日志对象
        self.logger = logging.getLogger(name)
        # 设置日志级别 debug
        self.logger.setLevel(logging.DEBUG)

    def __get_logs_dir(self):
        """
        根据当前时间，创建以当前时间为依据的目录
        :return:
        """
        # 拼接根目录和logs  #ptSamples/logs
        logs = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'logs')
        # print(logs)
        if not os.path.exists(logs):  # 判断项目下是否存在logs文件夹
            os.mkdir(logs)  # 不存在就创建该文件夹

        # 判断是否存在当天时间的日志目录
        date_dir = os.path.join(logs, time.strftime('%Y_%m_%d'))  # ptSamples/logs/2021_06_15/
        if not os.path.exists(date_dir):
            os.mkdir(date_dir)

        return date_dir

    def __log_handler(self):

        current_date_dir = self.__get_logs_dir()  # 获取日志文件目录
        # 添加日志文件名
        logfile = os.path.join(current_date_dir, time.strftime("%H_%M_%S"))

        # 创建命令行输出流，并将文件处理流对象赋值给变量stream
        stream_handler = logging.StreamHandler()
        # 设置stream的日志级别
        stream_handler.setLevel(logging.DEBUG)
        # 创建日志自定义的格式化对象
        formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(name)s : %(message)s")
        # 将格式赋值给 stream 文件流
        stream_handler.setFormatter(formatter)

        # 创建文件输入流 可以将日志保存到文件中
        file_handler = logging.FileHandler(filename=logfile + '.log', mode='w', encoding='utf-8')
        # 设置文件输入流的日志级别
        file_handler.setLevel(logging.DEBUG)
        # 设置文件输入流的格式
        file_handler.setFormatter(formatter)

        return stream_handler, file_handler

    @property
    def get_log(self):
        stream_handler, file_handler = self.__log_handler()
        # 将是stream添加到日志记录器中
        self.logger.addHandler(stream_handler)
        # 将文件输入流添加到自定义的日志记录器中
        self.logger.addHandler(file_handler)
        return self.logger

    # 清理上个月的日志
    def clean_log(self):
        path = os.path.dirname(self.__get_logs_dir())
        today_date = time.strftime('%Y_%m_%d')
        print(f'开始检测是否存在上月日志:{path}')

        state = False

        # 遍历目录下的所有日志文件 i是文件名
        for i in os.listdir(path):
            file_path = path + '/' + i  # 生成日志文件的路径
            # 获取日志的年月，和今天的年月
            today_m = int(today_date[5:7])  # 今天的月份
            today_y = int(today_date[0:4])  # 今天的年份

            # print(f'今天的年份{today_y},今天的月份{today_m}')

            log_m = int(i[5:7])  # 日志的月份
            log_y = int(i[0:4])  # 日志的年份
            # print(f'日志文件的年份{log_y},日志的月份{log_m}')

            # 对上个月的日志进行清理，即删除。
            if os.path.exists(file_path):  # 判断文件是否存在
                # print(file_path)
                if log_m < today_m:  # 判断日志文件的月份是否小于当月
                    shutil.rmtree(file_path)
                    state = True
                    print(f'日志文件月份小于当月,删除:{i}')
                elif log_y < today_y:  # 判断日志的文件年份是否小于今年
                    shutil.rmtree(file_path)
                    state = True
                    print(f'日志年份小于今年,删除:{i}')

        if state:
            print('上月日志删除完成')
        else:
            print('不存在可删除日志')

logger = Log().get_log
logger.info("test信息")  # 2021-06-15 14:58:31,233 - DEBUG - apipro : test信息

if __name__ == "__main__":
    Log().clean_log()