fastapi返回xlsx文件

fastapi返回xlsx文件，代码如下：

# 这里是导入部分包
from fastapi import FastAPI
from fastapi.responses import StreamingResponse
from io import BytesIO
import xlsxwriter
import time


# 创建一个app
app = FastAPI()


# 这个就是创建一个小的api，和我们这个文章的主题没啥关系
@app.get("/")
async def root():
    return {"message": "Hello World"}


# 这个就是从stack overflow上看到的答案
# 答案的第一句是这样写的：
# @app.get("/testxlsx", , response_description='xlsx')
# 但是我尝试了一下，将response_description=‘xlsx'丢掉也是没有任何问题的

@app.get("/testxlsx")  # , response_description='xlsx'
async def payments():

    for i in range(5):
        print(i)
        time.sleep(1)

    output = BytesIO() # 在内存中创建一个缓存流
    workbook = xlsxwriter.Workbook(output)
    worksheet = workbook.add_worksheet()
    worksheet.write(0, 0, 'ISBN')
    worksheet.write(0, 1, 'Name')
    worksheet.write(0, 2, 'Takedown date')
    worksheet.write(0, 3, 'Last updated')
    workbook.close()
    output.seek(0)

    headers = {
        'Content-Disposition': 'attachment; filename="filename.xlsx"'
    }
    return StreamingResponse(output, headers=headers)


# uvicorn main:app --reload

总结就是使用io的BytesIo来作为保存的一个载体，在StreamingResponse里面设置好类型；

然后在终端输入：

uvicorn main:app --reload

然后在浏览器打开链接：

http://127.0.0.1:8000/testxlsx

这个主要是记录一下，fastapi返回excel这样的类型，我还没有遇到过，在fastapi的官网上也没找到（官网上说的比较深入，但是没看懂）但是这个人写的就比较清楚；

参考链接：

https://stackoverflow.com/questions/63465155/how-to-return-xlsx-file-from-memory-in-fastapi