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

 
Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐