moduwusuowei 发表于 2023-5-17 17:50

【Python】fastapi实现后端导出Excel

## 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": }
    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):
            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文件到本地了,就是这么简单,感兴趣的小伙伴可以自己试试咯~

moduwusuowei 发表于 2023-5-17 17:50

uvicorn.run(app='demo3:app', host="127.0.0.1", port=8010, reload=True)中的demo3是你自己python文件的名字,别搞错了哈

losingstars 发表于 2023-5-17 20:31

谢谢分享。学习了。

fufuok 发表于 2023-5-17 20:37

{:17_1059:} 不错不错,技术贴支持~~

qzcwuaipojie 发表于 2024-7-1 13:01

来实践一下,fastapi是我工作常用的,必须好好学习成为大牛!
页: [1]
查看完整版本: 【Python】fastapi实现后端导出Excel