Получение поврежденных архивов с помощью Python ZipStream в Django

Question

Я немного новичок в Джанго, поэтому, пожалуйста, потерпите меня.
Отсюда я использую zipstream и имею представление Django, которое возвращает zip-файл всех вложенных файлов, которые все размещены на Amazon S3. Но при загрузке все zip-файлы выглядят испорченными, то есть я не могу их открыть. Я попытался проверить файлы с unzip -t но ошибки не очень полезны.

file_paths = [fa.file.url for fa in file_attachments.all()]

zf = zipstream.ZipFile(mode='w', compression=zipstream.ZIP_DEFLATED)

zip_subdir = "Attachments-%s" % (request_id)

for file_path in file_paths:
    file_dir, file_name = os.path.split(file_path)

    zf.writestr(file_name, urllib.urlopen(file_path).read())

zip_filename = "%s.zip" % (zip_subdir)

response = StreamingHttpResponse(zf, mimetype='application/zip')
response['Content-Disposition'] = \
    'attachment; filename={}'.format(zip_filename)
return response

Есть идеи?

Решил это.

s = StringIO.StringIO()
with zipstream.ZipFile(s, mode='w', compression=zipstream.ZIP_DEFLATED) as zf:
    #could fail on url open.
    for file_path in file_paths:
        file_dir, file_name = os.path.split(file_path)

        try:
            file_contents = urllib.urlopen(file_path).read()
            zf.writestr(file_name, file_contents)
        except IOError: #connection cannot be made
            logging.error()

response = StreamingHttpResponse(s.getvalue(), mimetype='application/octet-stream')
response['Content-Disposition'] = \
    'attachment; filename={}'.format("%s" % (request_id))
return response

Answer 1

Вы должны закрыть ZipFile когда вы закончите писать к нему. В противном случае, чтобы процитировать документацию, "важные записи не будут записаны", пока вы этого не сделаете.

Самый чистый способ сделать это - использовать with заявление:

with zipstream.ZipFile(mode='w', compression=zipstream.ZIP_DEFLATED) as zf:
    # ...write to zf...