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">

  
    <title>Download File</title>
  
  
    <a href="%7B%7B%20url_for('text2speech')%20%7D%7D">Download File</a>
  
</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 중국어 웹사이트의 기타 관련 기사를 참조하세요!