【Python】fastapi实现后端导出Excel

不念不忘少年蓝@ 2024-03-22 18:32 42阅读 0赞

### fastapi实现后端导出Excel ###

众所众知在CRM系统中常见的功能是查询，如果要把查询的结果本地化就需要导出数据至Excel了，导出前端后端都可以做，前端来做，就是接收后端数据加工成数据流形式再保存在Excel中；后端做的过程差不多，也是先将数据处理好，结果成数据流，借助第三方包来完成数据导出，今天主要谈谈后者，代码也比较简单，跟着代码去学习：

from fastapi import FastAPI, Response
    from fastapi.responses import StreamingResponse
    import pandas as pd
    import io
    import xlsxwriter
    
    app = FastAPI()
    
    @app.get("/")
    async def download_excel():
        # 生成一个简单的 DataFrame
        data = {"name": ["Alice", "Bob", "Charlie"], "age": [25, 30, 35]}
        df = pd.DataFrame(data)
    
        # 将 DataFrame 转换为 Excel 文件
        buffer = io.BytesIO()
        workbook = xlsxwriter.Workbook(buffer)
        worksheet = workbook.add_worksheet()
        for i, col in enumerate(df.columns):
            worksheet.write(0, i, col)
            for j, value in enumerate(df[col]):
                worksheet.write(j + 1, i, value)
        workbook.close()
    
        # 将 Excel 文件转换为字节流，作为响应体返回给客户端
        buffer.seek(0)
        excel_bytes = buffer.getvalue()
        return StreamingResponse(io.BytesIO(excel_bytes), media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", headers={"Content-Disposition": "attachment; filename=myfile.xlsx"})
    
    
    if __name__ == '__main__':
        import uvicorn
    
        # 官方推荐是用命令后启动 uvicorn main:app --host=127.0.0.1 --port=8000 --reload
        uvicorn.run(app='demo3:app', host="127.0.0.1", port=8010, reload=True)

浏览器请求：http://127.0.0.1:8010/

就会自动下载Excel文件到本地了，就是这么简单，感兴趣的小伙伴可以自己试试咯~