首页 > 文章 > 后端开发 > 如何使用 FastAPI 发布数据后下载文件？

如何使用 FastAPI 发布数据后下载文件？

Barbara Streisand原创: 2024-10-31 20:25:17995浏览

How to Download a File after POSTing Data Using FastAPI?

如何使用 FastAPI 在发布数据后下载文件？

使用 FastAPI 和文件处理时，一项常见任务是让用户在提交后下载文件通过 POST 请求获取数据。这可以通过利用 FileResponse 类并确保标头和内容类型的正确配置来实现。

在下面提供的代码中，我们将演示如何设置处理数据并返回音频的 POST 端点文件（MP3）供下载。我们将使用表单数据类型来捕获用户输入并生成音频文件。

选项 1：使用单独的下载端点

在此方法中，我们将为处理文件下载。

<code class="python">from fastapi import FastAPI, Request, Form, File, UploadFile
from fastapi.responses import FileResponse, HTMLResponse
from fastapi.templating import Jinja2Templates
from gtts import gTTS

app = FastAPI()
templates = Jinja2Templates(directory="templates")


def text_to_speech(language: str, text: str) -> str:
    tts = gTTS(text=text, lang=language, slow=False)
    filepath = "temp/welcome.mp3"
    tts.save(filepath)
    return f"Text to speech conversion successful to {filepath}"


@app.get("/")
def home(request: Request):
    return templates.TemplateResponse("index.html", {"request": request})


@app.post("/text2speech")
def text2speech(request: Request, message: str = Form(...), language: str = Form(...)):
    if message and language:
        output = text_to_speech(language, message)
    path = "temp/welcome.mp3"
    filename = os.path.basename(path)
    headers = {"Content-Disposition": f"attachment; filename={filename}"}
    return FileResponse(path, headers=headers, media_type="audio/mp3")</code>

GET 端点用作主页，而 POST 端点处理用户输入并生成音频文件。 FileResponse 负责为文件下载设置适当的标头。

在模板中，您可以添加一个链接来触发文件下载：

<code class="html"><!DOCTYPE html>
<html>
  <head>
    <title>Download File</title>
  </head>
  <body>
    <a href="{{ url_for('text2speech') }}">Download File</a>
  </body>
</html></code>

选项 2：端点内文件下载

或者，您可以在处理数据的同一端点内处理文件下载：

<code class="python">from fastapi import FastAPI, Request, Form

app = FastAPI()


def text_to_speech(language: str, text: str) -> str:
    tts = gTTS(text=text, lang=language, slow=False)
    filepath = "temp/welcome.mp3"
    tts.save(filepath)
    return f"Text to speech conversion successful to {filepath}"


@app.post("/text2speech")
def text2speech(request: Request, message: str = Form(...), language: str = Form(...)):
    output = text_to_speech(language, message)
    return {"filepath": filepath}</code>

在模板中，您可以使用 JavaScript 使用从返回的文件路径触发文件下载端点：

<code class="html"><script>
  const filepath = "{{ filepath }}";
  window.location.href = filepath;
</script></code>

处理大文件

对于超出内存限制的文件，可以使用 StreamingResponse：

<code class="python">from fastapi.responses import StreamingResponse

@app.post("/text2speech")
def text2speech(request: Request, message: str = Form(...), language: str = Form(...)):
    ...
    def iterfile():
        with open(filepath, "rb") as f:
            yield from f

    headers = {"Content-Disposition": f"attachment; filename={filename}"}
    return StreamingResponse(iterfile(), headers=headers, media_type="audio/mp3")</code>

下载后删除文件

为了防止磁盘空间累积，最好在下载文件后将其删除。使用BackgroundTasks，您可以将文件删除安排为后台任务：

<code class="python">from fastapi import BackgroundTasks

@app.post("/text2speech")
def text2speech(request: Request, background_tasks: BackgroundTasks, ...):
    ...
    background_tasks.add_task(os.remove, path=filepath)
    return FileResponse(filepath, headers=headers, media_type="audio/mp3")</code>

以上是如何使用 FastAPI 发布数据后下载文件？的详细内容。更多信息请关注PHP中文网其他相关文章！

JavaScript fastapi for while using class delete this background input

声明：

本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系admin@php.cn

上一篇：How can I obtain IP addresses of visitors using Flask?下一篇：How do the \'levels\', \'keys\', and \'names\' arguments in Pandas\' concat function work to create a MultiIndex?

查看更多