Ответный ZIP-файл от Django

Question

Я пытаюсь скачать img из URL, добавить его в zip-архив и затем ответить на этот архив Django HttpResponse.

import os
import requests
import zipfile
from django.http import HttpResponse

url = 'http://some.link/img.jpg' 
file = requests.get(url)
data = file.content
rf = open('tmp/pic1.jpg', 'wb')
rf.write(data)
rf.close()
zipf = zipfile.ZipFile('tmp/album.zip', 'w') # This file is ok
filename = os.path.basename(os.path.normpath('tmp/pic1.jpg'))
zipf.write('tmp/pic1.jpg', filename)
zipf.close()
resp = HttpResponse(open('tmp/album.zip', 'rb'))
resp['Content-Disposition'] = 'attachment; filename=album.zip'
resp['Content-Type'] = 'application/zip'
return resp # Got corrupted zip file

Когда я сохраняю файл в папку tmp - ничего страшного, я могу его распаковать.Но когда я отвечаю на этот файл, я получаю «Ошибка 1/2/21» на MacOS или неожиданный EOF, если я пытаюсь открыть в редакторе Atom (только для проверки).Я также использовал StringIO вместо сохранения zip-файла, но это не влияет на результат.

Answer 1

Если вы используете Python 3, вы должны сделать это так:

import os, io, zipfile, requests
from django.http import HttpResponse

# Get file
url = 'https://some.link/img.jpg'
response = requests.get(url)
# Get filename from url
filename = os.path.split(url)[1]
# Create zip
buffer = io.BytesIO()
zip_file = zipfile.ZipFile(buffer, 'w')
zip_file.writestr(filename, response.content)
zip_file.close()
# Return zip
response = HttpResponse(buffer.getvalue())
response['Content-Type'] = 'application/x-zip-compressed'
response['Content-Disposition'] = 'attachment; filename=album.zip'

return response

Это без сохранения файла.Загруженный файл идет непосредственно к io.

. Чтобы ответить на сохраненный файл, используйте следующий синтаксис:

response = HttpResponse(open('path/to/file', 'rb').read())