Python:使用 "with"打开压缩或解压缩文件

Question

我正在寻找使用“with”打开文件的解决方案。然而，我想检查文件是否被 gzip 压缩，然后使用 gzip.open 或只是打开类似的内容:

if f.endswith(".gz"):
    with gzip.open() as f:
else:
    with open() as f:
for line in f:

有什么想法吗？

Answer 1

实现此目的的一种方法是创建自己的上下文管理器来为您完成此操作。

from contextlib import contextmanager

@contextmanager
def open_file(filename):
    if filename.endswith(".gz"):
        with gzip.open(filename) as f:
            yield f
    else:
        with open(filename) as f:
            yield f

然后您就可以使用它来打开文件。

with open_file("filename") as f:
    # f will be either the result of `gzip.open` or `open`, depending on its filename.
    for line in f:
        pass  

这使您能够保留使用 with 子句的优点，同时使您能够灵活地选择要调用的函数。