现在我需要提供一个链接,访问时通过 openpyxl 动态生成一个 Excel 文档,并作为响应体返回。
在这里将动态生成的 Excel 文档保存到服务器,并重定向到静态文件链接显然不是一个合适的做法。
所以我们需要直接将 openpyx 生成的 Excel 文档写入到 Flask 的 HttpResponse 对象响应体中。
研究源码后发现,openpyxl在保存excel文件时会使用save_workbook 这个函数,这个函数在 openpyxl.writer.excel 模块中定义。
打开这个文件,这里还定义了一个叫做 save_virtual_workbook 的函数,看来这就是我们要找的东西。
def save_virtual_workbook(workbook,):
"""Return an in-memory workbook, suitable for a Django response."""
temp_buffer = BytesIO()
archive = ZipFile(temp_buffer, 'w', ZIP_DEFLATED, allowZip64=True)
writer = ExcelWriter(workbook, archive)
try:
writer.write_data()
finally:
archive.close()
virtual_workbook = temp_buffer.getvalue()
temp_buffer.close()
return virtual_workbook
这个函数将 Excel 文档写入到内存,返回一个字节数组。
Flask 的 HttpResponse 类构造方法可以接受一个字节数组作为响应体,所以我们可以这样写。
from openpyxl import Workbook
from openpyxl.writer.excel import save_virtual_workbook
from flask import make_response
wb = Workbook()
ws = wb.create_sheet('test1')
ws.append('这是一个测试')
content = save_virtual_workbook(wb)
resp = make_response(content)
resp.headers["Content-Disposition"] = 'attachment; filename=finance_info.xlsx'
resp.headers['Content-Type'] = 'application/x-xlsx'
之后在返回resp就可以了。
如果这篇文章对你有帮助的话,请点个赞,谢谢!