如何在Flet-FastAPI应用中实现文件下载功能

flet-fastapi
flet
uvicorn
pydantic
fastapi

python -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt

from contextlib import asynccontextmanager
import flet as ft
import flet_fastapi
from fastapi import FastAPI
from fastapi.responses import FileResponse, StreamingResponse
import os # 用于文件路径操作

# 假设要下载的文件位于项目根目录
# 为了演示，我们先创建一个虚拟文件
with open("test.txt", "w") as f:
    f.write("This is a test file for download.")

@asynccontextmanager
async def lifespan(app: FastAPI):
    # Flet-FastAPI管理器启动
    await flet_fastapi.app_manager.start()
    yield
    # Flet-FastAPI管理器关闭
    await flet_fastapi.app_manager.shutdown()

# 初始化FastAPI应用，并指定生命周期管理器
app = FastAPI(lifespan=lifespan)

@app.get('/download')
async def download_file_endpoint():
    file_path = "test.txt" # 替换为你的文件实际路径
    file_name = "example_document.txt" # 用户下载时看到的文件名

    # 检查文件是否存在
    if not os.path.exists(file_path):
        # 可以返回一个错误响应，例如 404 Not Found
        return {"message": "File not found"}, 404

    # 设置Content-Disposition头部，强制浏览器下载
    headers = {'Content-Disposition': f'attachment; filename="{file_name}"'}

    # 返回FileResponse
    return FileResponse(
        file_path,
        media_type="text/plain", # 根据文件类型设置正确的MIME类型
        filename=file_name, # 再次指定文件名，FileResponse会自动处理Content-Disposition
        headers=headers # 也可以通过headers参数传递
    )

async def main(page: ft.Page):
    page.title = "Flet-FastAPI 文件下载示例"

    async def download_button_clicked(e):
        # 触发浏览器访问FastAPI的下载端点
        # _self 表示在当前窗口/标签页打开，从而触发下载
        await page.launch_url_async(url='/download', web_window_name='_self')

    await page.add_async(
        ft.FilledButton(text="下载文件", on_click=download_button_clicked)
    )
    await page.update_async()

# 将Flet应用挂载到FastAPI的根路径
app.mount('/', flet_fastapi.app(main))

from fastapi import FastAPI
from fastapi.responses import FileResponse
from contextlib import asynccontextmanager
import flet as ft
import flet_fastapi
import os

# 为了演示，创建一个虚拟文件
file_to_download = "test.txt"
with open(file_to_download, "w") as f:
    f.write("这是一个用于Flet-FastAPI下载功能的示例文件内容。")
    f.write("\n第二行内容。")

@asynccontextmanager
async def lifespan(app: FastAPI):
    """
    FastAPI应用的生命周期管理器，用于启动和关闭Flet-FastAPI管理器。
    """
    await flet_fastapi.app_manager.start()
    print("Flet-FastAPI管理器已启动。")
    yield
    await flet_fastapi.app_manager.shutdown()
    print("Flet-FastAPI管理器已关闭。")
    # 清理演示文件
    if os.path.exists(file_to_download):
        os.remove(file_to_download)
        print(f"已删除演示文件: {file_to_download}")


app = FastAPI(lifespan=lifespan)

@app.get('/download')
async def download_file_endpoint():
    """
    FastAPI端点，负责返回文件供下载。
    """
    # 假设文件位于当前工作目录
    file_path = file_to_download 
    user_filename = "我的下载文件.txt" # 用户下载时看到的文件名

    if not os.path.exists(file_path):
        return {"message": "文件不存在！"}, 404

    # 设置Content-Disposition头部，强制浏览器下载文件
    # FileResponse的filename参数通常会自动设置此头部，但明确指定更佳
    headers = {
        'Content-Disposition': f'attachment; filename="{user_filename}"'
    }

    return FileResponse(
        path=file_path,
        media_type="text/plain", # 根据文件类型设置正确的MIME类型
        filename=user_filename, # 再次指定，确保兼容性
        headers=headers
    )

async def main(page: ft.Page):
    """
    Flet应用的UI主函数。
    """
    page.title = "Flet-FastAPI 文件下载示例"
    page.vertical_alignment = ft.MainAxisAlignment.CENTER
    page.horizontal_alignment = ft.CrossAxisAlignment.CENTER

    async def download_button_clicked(e):
        """
        按钮点击事件处理函数，触发文件下载。
        """
        print("下载按钮被点击...")
        # 使用launch_url_async触发浏览器访问FastAPI下载端点
        # _self 参数确保在当前窗口/标签页进行操作，从而触发下载
        await page.launch_url_async(url='/download', web_window_name='_self')
        print("已触发文件下载请求。")

    await page.add_async(
        ft.FilledButton(text="点击下载文件", on_click=download_button_clicked)
    )
    await page.update_async()

# 将Flet应用挂载到FastAPI的根路径
app.mount('/', flet_fastapi.app(main))

uvicorn --reload main:app