众所众知在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文件到本地了，就是这么简单，感兴趣的小伙伴可以自己试试咯~
