如何对多个 zip 文件的 BytesIO 进行上下文管理?

Question

我正在尝试在创建多个zip文件时对BytesIO流使用上下文管理器。在写入第一个 zip 文件后，我找不到"重置"BytesIO 对象的方法，因此我可以使用相同的 BytesIO 对象来创建下一个 zip 文件。我总是得到一个"无法打开文件...作为存档"错误，在将第二个 zip 文件写入磁盘后尝试打开该文件。第一个zip文件打开就好了。我已经搜索过，但找不到解决方案。将模式从写入更改为追加也无济于事。当然，我可以重新初始化为新的 BytesIO 对象，但这会破坏上下文管理器。下面是我认为应该可以工作的代码。我在Windows 10上使用Anaconda Python 3.6.6。

import io
import os
import zipfile
with io.BytesIO() as bytes_io:
with zipfile.ZipFile(bytes_io, mode='w') as zf:
filecount = 0
for item in os.scandir(r'C:UsersstephenDocuments'):
if not item.is_dir():
zf.write(item.path, item.name)
filecount += 1
if filecount % 3 == 0:
with open(r'C:UsersstephenDocuments\' + str(filecount // 3) + '.zip', 'wb') as f:
f.write(bytes_io.getvalue())
bytes_io.seek(0)
bytes_io.truncate()

Answer 1

您可以重用相同的BytesIO对象，但应为每个要创建的 zip 文件创建一个新的ZipFile对象：

with io.BytesIO() as bytes_io:
filecount = 0
for item in os.scandir(r'C:UsersstephenDocuments'):
if not item.is_dir():
with zipfile.ZipFile(bytes_io, mode='w') as zf:
zf.write(item.path, item.name)
filecount += 1
if filecount % 3 == 0:
with open(r'C:UsersstephenDocuments\' + str(filecount // 3) + '.zip', 'wb') as f:
f.write(bytes_io.getvalue())
bytes_io.seek(0)
bytes_io.truncate()