python-zipstream icon indicating copy to clipboard operation
python-zipstream copied to clipboard

Does it support zipping multiple S3 objects and sending to Django response stream without downloading locally?

Open aamironline opened this issue 4 years ago • 3 comments

If yes, can you provide an example? Any issues?

aamironline avatar Nov 02 '20 09:11 aamironline

I'm afraid that functionality would be outside of the scope of this package. However, zipstream does support streaming input files. So if you have an S3 library that can stream the data instead of storing it locally, you can probably use those streams as inputs for zipstream and then stream the zipped output to the browser. In that case, the data would never touch your server's disk.

arjan-s avatar Nov 02 '20 20:11 arjan-s

The approach below worked for me. It relies on using requests with signed URLs from S3, but I imagine the same can be achieved with S3.Client.download_fileobj in boto3.

some_list_of_files_to_get_from_s3 = [
    (<s3 signed url>, '<name of file>'),
    ...
]

z = zipstream.ZipFile(mode="w", compression=zipstream.ZIP_DEFLATED)

for each_s3_signed_url, file_name in some_list_of_files_to_get_from_s3:
    with requests.get(each_s3_signed_url, stream=True) as r:
        z.writestr(file_name, r.content) # this worked for me
        # tried the following but could not get it to write data to the files in the zip archive:
        # z.write_iter(file_name, r.iter_content())

With Django, it was then matter of passing z into StreamingHttpResponse to round out the view code:

response = StreamingHttpResponse(z, content_type='application/zip')
response['Content-Disposition'] = 'attachment; filename=archive.zip'
return response

Hope it helps.

gassc avatar Jul 21 '21 16:07 gassc

@gassc Thank you for your comment, I'm just solving a similar problem. Do you know if your approach can work the same way with Django-storages, I know it's using boto under the hood but not much else.

1oglop1 avatar Nov 23 '21 05:11 1oglop1