Python logging模块实现日志饶接 按照时间命名

import os
import zipfile
from datetime import datetime
from logging.handlers import RotatingFileHandler


class CompressedRotatingFileHandler(RotatingFileHandler):
    """
    自定义的 RotatingFileHandler，支持在日志轮转时压缩旧日志文件，
    并管理压缩文件的最大数量。
    """

    def __init__(self, filename, mode='a', maxBytes=10 * 1024 * 1024, backupCount=5, encoding=None, delay=False):
        """
        初始化处理器。

        :param filename: 日志文件名。
        :param mode: 打开文件的模式，默认是追加模式 'a'。
        :param maxBytes: 日志文件的最大字节数，达到后进行轮转。
        :param backupCount: 保留的压缩日志文件的最大数量。
        :param encoding: 文件编码。
        :param delay: 如果为 True，则文件将在第一次写入时打开。
        """
        super().__init__(filename, mode, maxBytes, backupCount, encoding, delay)
        self.backupCount = backupCount

    def doRollover(self):
        """
        执行日志轮转：关闭当前日志文件，重命名并压缩旧日志文件，
        然后删除多余的压缩日志文件。
        """
        if self.stream:
            self.stream.close()
            self.stream = None

        # 获取当前时间戳
        timestamp = datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
        # 定义压缩后的文件名
        zip_filename = f"{os.path.splitext(self.baseFilename)[0]}_{timestamp}"
        file_name = os.path.basename(zip_filename)
        # 创建 zip 压缩包并添加日志文件
        with zipfile.ZipFile(zip_filename + '.zip', 'w', zipfile.ZIP_DEFLATED) as zipf:
            zipf.write(self.baseFilename, arcname=file_name + '.log')

        # 清空当前日志文件
        with open(self.baseFilename, 'w'):
            pass

        # 管理备份文件
        self._cleanup_old_logs()

    def _cleanup_old_logs(self):
        """
        删除多余的压缩日志文件，保留最新的 backupCount 个文件。
        """
        dir_name, base_name = os.path.split(self.baseFilename)
        prefix = os.path.splitext(base_name)[0] + "_"
        suffix = ".zip"
        # 获取所有符合命名规则的压缩日志文件
        compressed_files = [
            os.path.join(dir_name, f) for f in os.listdir(dir_name)
            if f.startswith(prefix) and f.endswith(suffix)
        ]
        # 按修改时间排序，最新的在后面
        compressed_files.sort(key=lambda x: os.path.getmtime(x))
        # 删除多余的文件
        while len(compressed_files) > self.backupCount:
            oldest = compressed_files.pop(0)
            os.remove(oldest)


import logging


def setup_logger(log_file, max_bytes=10 * 1024 * 1024, backup_count=5, log_level=logging.INFO):
    """
    配置并返回一个日志记录器。

    :param log_file: 日志文件路径。
    :param max_bytes: 日志文件的最大字节数，超过后进行轮转和压缩。
    :param backup_count: 保留的压缩日志文件的最大数量。
    :param log_level: 日志级别。
    :return: 配置好的 Logger 对象。
    """
    logger = logging.getLogger("CompressedLogger")
    logger.setLevel(log_level)
    logger.propagate = False  # 防止日志传递到根记录器

    if not logger.handlers:
        handler = CompressedRotatingFileHandler(
            filename=log_file,
            maxBytes=max_bytes,
            backupCount=backup_count,
            encoding='utf-8'
        )
        formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
        handler.setFormatter(formatter)
        logger.addHandler(handler)

    return logger


# 使用示例
if __name__ == "__main__":
    logger = setup_logger("logs/application.log", max_bytes=0.5 * 1024 * 1024, backup_count=2000,
                          log_level=logging.DEBUG)

    for i in range(100000):
        logger.info(f"这是第 {i} 条日志记录")