Генерування файлу для завантаження за допомогою Django

Чи можна створити zip-архів і запропонувати його завантажити, але при цьому не зберегти файл на жорсткий диск?

Для запуску завантаження потрібно встановити Content-Dispositionзаголовок:

from django.http import HttpResponse
from wsgiref.util import FileWrapper

# generate the file
response = HttpResponse(FileWrapper(myfile.getvalue()), content_type='application/zip')
response['Content-Disposition'] = 'attachment; filename=myfile.zip'
return response

Якщо ви не хочете файл на диску, вам потрібно скористатися StringIO

import cStringIO as StringIO

myfile = StringIO.StringIO()
while not_finished:
    # generate chunk
    myfile.write(chunk)

За бажанням ви також можете встановити Content-Lengthзаголовок:

response['Content-Length'] = myfile.tell()

Вам буде приємніше створювати тимчасовий файл. Це економить багато пам’яті. Якщо одночасно у вас є більше одного або двох користувачів, ви виявите, що економія пам'яті дуже, дуже важлива.

Тим не менш, ти можеш написати в об'єкт StringIO .

>>> import zipfile
>>> import StringIO
>>> buffer= StringIO.StringIO()
>>> z= zipfile.ZipFile( buffer, "w" )
>>> z.write( "idletest" )
>>> z.close()
>>> len(buffer.getvalue())
778

Об'єкт "буфер" схожий на файл із 778-байтовим архівом ZIP.

Чому б замість цього не створити файл tar? Так:

def downloadLogs(req, dir):
    response = HttpResponse(content_type='application/x-gzip')
    response['Content-Disposition'] = 'attachment; filename=download.tar.gz'
    tarred = tarfile.open(fileobj=response, mode='w:gz')
    tarred.add(dir)
    tarred.close()

    return response

Так, ви можете використовувати ZipFile модуль , модуль ZLIB або інші модулі стиснення для створення архіву поштового індексу в пам'яті. Ви можете змусити свій вигляд писати zip-архів до HttpResponseоб’єкта, який повертає подання Django, замість того, щоб надсилати контекст шаблону. Нарешті, вам потрібно буде встановити mimeype у відповідному форматі, щоб повідомити браузеру обробляти відповідь як файл .

models.py

from django.db import models

class PageHeader(models.Model):
    image = models.ImageField(upload_to='uploads')

views.py

from django.http import HttpResponse
from StringIO import StringIO
from models import *
import os, mimetypes, urllib

def random_header_image(request):
    header = PageHeader.objects.order_by('?')[0]
    image = StringIO(file(header.image.path, "rb").read())
    mimetype = mimetypes.guess_type(os.path.basename(header.image.name))[0]

    return HttpResponse(image.read(), mimetype=mimetype)

Приклад коду можна знайти за адресою http://djangosnippets.org/snippets/365/

def download_zip(request,file_name):
    filePath = '<path>/'+file_name
    fsock = open(file_name_with_path,"rb")
    response = HttpResponse(fsock, content_type='application/zip')
    response['Content-Disposition'] = 'attachment; filename=myfile.zip'
    return response

Ви можете замінити zip та тип вмісту відповідно до ваших вимог.

Те саме з архівом tgz в пам'яті:

import tarfile
from io import BytesIO


def serve_file(request):
    out = BytesIO()
    tar = tarfile.open(mode = "w:gz", fileobj = out)
    data = 'lala'.encode('utf-8')
    file = BytesIO(data)
    info = tarfile.TarInfo(name="1.txt")
    info.size = len(data)
    tar.addfile(tarinfo=info, fileobj=file)
    tar.close()

    response = HttpResponse(out.getvalue(), content_type='application/tgz')
    response['Content-Disposition'] = 'attachment; filename=myfile.tgz'
    return response