能否用单个上下文管理器替代两个处理Gzip与普通日志文件？

当然可以实现！你只需要封装一个简单的辅助函数，根据文件后缀返回对应的文件对象，就能用单个with语句处理两种格式的日志文件，既避免了重复代码，又能保证文件被正确关闭。

具体实现方案 #

先定义一个辅助函数，它会判断文件是否为.gz格式，返回对应的上下文管理器对象：

import gzip
from io import TextIOWrapper

def open_log(file_path):
    # 如果是gzip压缩文件，打开二进制流并转成文本模式（适配文本日志场景）
    if file_path.endswith('.gz'):
        gz_file = gzip.open(file_path, 'rb')
        return TextIOWrapper(gz_file, encoding='utf-8')
    # 非压缩文件直接以文本模式打开
    else:
        return open(file_path, 'r', encoding='utf-8')

然后你就可以用统一的with语句处理所有文件了：

file_path = "your_log_file.log"  # 也可以是 "your_log_file.log.gz"
with open_log(file_path) as f:
    # 这里执行你原来的逻辑，比如逐行处理日志
    for line in f:
        process_log_line(line)  # 替换成你的实际处理代码

适配二进制日志的变体 #

如果你的日志是二进制格式，不需要转文本的话，可以去掉TextIOWrapper，统一用二进制模式打开：

def open_log(file_path):
    if file_path.endswith('.gz'):
        return gzip.open(file_path, 'rb')
    else:
        return open(file_path, 'rb')

这个方法的核心是利用Python的上下文管理器协议：gzip.open()和open()返回的对象都实现了__enter__和__exit__方法，所以都能在with语句中使用，确保文件操作完成后自动关闭，不会出现资源泄漏问题。

内容的提问来源于stack exchange，提问作者snoob dogg