在Django中使用Python3-ZipStream获取损坏的zip

Question

我从这里使用zipstream，并且有一个Django视图，它返回所有文件附件的zip文件，这些附件都托管在AmazonS3上。但当我下载zip文件时，它们都会被损坏，也就是说，我无法打开它们。

import io
import zipstream
s = io.BytesIO()
with zipstream.ZipFile(s,"w", compression=zipstream.ZIP_DEFLATED) as zf:
for file_path in file_paths:
file_dir, file_name = os.path.split(file_path)
zf.writestr(file_name, urllib.urlopen(file_path).read())
response = StreamingHttpResponse(s.getvalue(), content_type='application/octet-stream')
response['Content-Disposition'] = 'attachment; filename={}'.format('files.zip')
return response

Answer 1

安装aiozipstream包而不是zipstream包。如果您已经安装了zipstream软件包，请先卸载它。pip卸载zipstream，然后执行pip安装aiozipstream

#Do the import in the same way
from zipstream import ZipStream
from django.http import StreamingHttpResponse
def s3_content_generator(key):
#s3_bucket - your s3 bucket name
infile_object = s3.get_object(Bucket=s3_bucket, Key= key)
data = str(infile_object.get('Body', '').read())
yield bytes(data, 'utf-8')

files = []
#filepath - list of s3 keys
for keyin filepath:
files.append({'stream': s3_content_generator(key),'name': 'filename'})
#large chunksize fasten the download process
zf = ZipStream(files, chunksize=32768)
response = StreamingHttpResponse(zf.stream(), content_type='application/zip')
response['Content-Disposition'] = 'attachment; filename={}'.format('files.zip')
return response