2

所以我正在使用zstandardpython库，并且我编写了一个帮助类和函数来使用上下文来解压缩文件。

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)

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

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

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

4

1 回答 1

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()

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

于 2020-10-24T14:40:05.033 回答