使用zstandard在Python中压缩文件

Question

因此，我使用了zstandard python库，并编写了一个帮助器类和函数来使用上下文来解压缩文件。

class ZstdReader:
    def __init__(self, filename):
        self.filename = filename

    def __enter__(self):
        self.f = open(self.filename, 'rb')
        dctx = zstd.ZstdDecompressor()
        reader = dctx.stream_reader(self.f)
        return io.TextIOWrapper(reader, encoding='utf-8')

    def __exit__(self, *a):
        self.f.close()
        return False

def openZstd(filename, mode='rb'):
    if 'w' in mode:
        return ZstdWriter(filename)
    return ZstdReader(filename)

这真的很好用，并且允许我在使用文件f进行json转储和加载之前只使用with openZstd('filename.zst', 'rb') as f:。然而，我在将其推广到编写时遇到了问题，我尝试以与目前为止相同的方式遵循文档，但有些东西不起作用。这是我尝试过的：

class ZstdWriter:
    def __init__(self, filename):
        self.filename = filename

    def __enter__(self):
        self.f = open(self.filename, 'wb')
        ctx = zstd.ZstdCompressor()
        writer = ctx.stream_writer(self.f)
        return io.TextIOWrapper(writer, encoding='utf-8')

    def __exit__(self, *a):
        self.f.close()
        return False

当我使用这个类打开一个文件，并执行json.dump([], f)时，由于某种原因，该文件最终是空的。我猜其中一个步骤是接受我的输入，但不知道它可能是什么。

Answer 1

正如jasonharper在评论中建议的那样，您必须同时刷新io包装器和编写器本身，如下所示：

s = json.dumps({})
iw = io.TextIOWrapper(writer, encoding="utf-8")
iw.write(s)

iw.flush()
writer.flush(zstd.FLUSH_FRAME)
f.close()

这会导致数据在文件中，并且文件是完整的。