Объект XlsxWriter сохранить как http-ответ для создания загрузки в Django

объект XlsxWriter сохранить как http-ответ для создания загрузки в Django?

27
django excel httpresponse python xlsxwriter

3 ответов

Я думаю, вы спрашиваете о том, как создать файл excel в памяти с помощью xlsxwriter и верните его через HttpResponse. Вот пример:

try:
    import cStringIO as StringIO
except ImportError:
    import StringIO

from django.http import HttpResponse

from xlsxwriter.workbook import Workbook


def your_view(request):
    # your view logic here

    # create a workbook in memory
    output = StringIO.StringIO()

    book = Workbook(output)
    sheet = book.add_worksheet('test')       
    sheet.write(0, 0, 'Hello, world!')
    book.close()

    # construct response
    output.seek(0)
    response = HttpResponse(output.read(), mimetype="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
    response['Content-Disposition'] = "attachment; filename=test.xlsx"

    return response

надеюсь, что это поможет.

46

небольшое обновление ответа @alecxe для Python 3 (io.BytesIO вместо StringIO.StringIO) и Django >= 1.5 (content_type вместо mimetype), С полностью в памяти файл сборки, которая с тех пор была реализована @jmcnamara ({'in_memory': True}) !
Вот полный пример :

import io

from django.http.response import HttpResponse

from xlsxwriter.workbook import Workbook


def your_view(request):

    output = io.BytesIO()

    workbook = Workbook(output, {'in_memory': True})
    worksheet = workbook.add_worksheet()
    worksheet.write(0, 0, 'Hello, world!')
    workbook.close()

    output.seek(0)

    response = HttpResponse(output.read(), content_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
    response['Content-Disposition'] = "attachment; filename=test.xlsx"

    output.close()

    return response
46

когда дело доходит до Джанго, вы даже можете обойтись без целого StringIO проделки. HttpResponse ведет себя так же, как StringIO в этом отношении:

from django.http import HttpResponse
from xlsxwriter.workbook import Workbook

def your_view(request):
    # your view logic here

    # create the HttpResponse object ...
    response = HttpResponse(content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
    response['Content-Disposition'] = "attachment; filename=test.xlsx"

    # .. and pass it into the XLSXWriter
    book = Workbook(response, {'in_memory': True})
    sheet = book.add_worksheet('test')       
    sheet.write(0, 0, 'Hello, world!')
    book.close()

    return response

добавление: вам нужно указать {'in_memory': True} или вы могли бы сделать HttpResponse has no attribute seek(). Спасибо @Jeb

13