CODEWITHSUNDEEP
flaskpandas.excelwriter
Nickpick
Nickpick

Reputation: 6587

Flask response to excel file giving corrupt excel file

I have a website that has a button. When it's clicked, it returns a number of pandas dataframers into an excel file and returns that excel file automatically as download.

It seems to work ok, except when I open the file, it seems to be corrupted. It asks if some of the tabs should be recovered. I'm using the code below. Any suggestions are appreciated what could be the cause for this.

import io
from flask.helpers import make_response
from pandas.io.excel import ExcelWriter

output = io.BytesIO()
writer = ExcelWriter(output)

dfs = [df1,df2....]
tabs ['tab1','tab2',....]

for df, tab_name in zip(dfs, tab_names):
    df.to_excel(writer, tab_name)
writer.close()

resp = make_response(output.getvalue())
resp.headers['Content-Disposition'] = 'attachment; filename=output.xlsx'
resp.headers["Content-type"] = "text/csv"
return resp

Upvotes: 2

Views: 1047

Answers (1)

Dave W. Smith
Dave W. Smith

Reputation: 24966

You'll need to to add

output.seek(0)

after you close the writer.

You might also find it easier to write

return send_file(output, attachment_filename="output.xlsx", as_attachment=True)

(after importing send_file from flask)

Upvotes: 1

Related Questions