使用烧瓶返回创建的 excel 文件

Question

I am creating a excel file using openpyxl which I would like to return as a file download (so not saving locally).

I can create the excel file fine and save it to disk. However, I cannot get this file to download.

Attempt 1:

import flask_excel as excel

...

create_excel_sheet(data) # internally save the sheet with name sheet.xlsx

output = excel.make_response()
output.headers["Content-Disposition"] = "attachment; filename=" + \
                                        'sheet.xlsx'
output.headers["Content-type"] = "application/vnd.openxmlformats-\
officedocument.spreadsheetml.sheet"

return output

This returns an empty text file with name sheet.xlsx

Attempt 2: wb = create_excel_sheet(data) # return openpyxl workbook

output = excel.make_response(wb)
output.headers["Content-Disposition"] = "attachment; filename=" + \
                                        'sheet.xlsx'
output.headers["Content-type"] = "application/vnd.openxmlformats-\
officedocument.spreadsheetml.sheet"

return output

I don't want to use pyexcel for the data as I need openpyxl to create a fancy excel sheet. Obviously if pyexcel and openpyxl communicated that would be fine.

any thoughts?

Cheers, Mike

Answer 1

Based on Charlie Clark's hint , I finally settled on the following solution.

output = make_response(create_sheet(data))
output.headers["Content-Disposition"] = "attachment; filename=" + \
                                        "sheet.xlsx"
output.headers["Content-type"] = \
    "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"

where

def create_sheet(data):

returns

return save_virtual_workbook(wb)

Answer 2

Since I struggled with ambiguity to reassemble fragmented and a little bit old-fashioned code snippets, I want to leave another answer here. This is technically the same but a quite complete code snippet which is a little bit more up to date.

from flask import Response
from openpyxl import Workbook
from openpyxl.writer.excel import save_virtual_workbook

...

@app.route("/download")
def download():
    wb = Workbook()
    
    ...

    return Response(
        save_virtual_workbook(wb),
        headers={
            'Content-Disposition': 'attachment; filename=sheet.xlsx',
            'Content-type': 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
        }
    )

Answer 3

What I did when I faced the same problem is that I wrote a temporary file on the server, I made my create_excel_sheet(data) function to return file name and then send the file back with the flask send_file() function:

send_file( create_excel_sheet_and_return_filename(data) )

You can use python modules that create temporary files which are deleted when the process exists and use authorization if security is an issue.

Answer 4

That all looks correct. Are you actually returning the response from your view? That wasn't clear in the question, and would explain the problem, I think.

For example:

@app.route('/download_sheet')
def download():
    create_excel_sheet(data)
    output = excel.make_response()
    output.headers["Content-Disposition"] = "attachment; filename=sheet.xlsx"
    output.headers["Content-type"] = "application/vnd.openxmlformats-\
officedocument.spreadsheetml.sheet"
    return output  # Send the response to the user

...

<a href="{{ url_for('app.download') }}">Click here for Sheet</a>