Flask HttpResponse 响应体中返回 openpyxl 生成的 Excel 文档

Flask HttpResponse 响应体中返回 openpyxl 生成的 Excel 文档

现在我需要提供一个链接，访问时通过 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就可以了。

如果这篇文章对你有帮助的话，请点个赞，谢谢！