fastapi upload file save

1import shutil
2from pathlib import Path
3from tempfile import NamedTemporaryFile
4from typing import Callable
5
6from fastapi import UploadFile
7
8
9def save_upload_file(upload_file: UploadFile, destination: Path) -> None:
10    try:
11        with destination.open("wb") as buffer:
12            shutil.copyfileobj(upload_file.file, buffer)
13    finally:
14        upload_file.file.close()
15
16
17def save_upload_file_tmp(upload_file: UploadFile) -> Path:
18    try:
19        suffix = Path(upload_file.filename).suffix
20        with NamedTemporaryFile(delete=False, suffix=suffix) as tmp:
21            shutil.copyfileobj(upload_file.file, tmp)
22            tmp_path = Path(tmp.name)
23    finally:
24        upload_file.file.close()
25    return tmp_path
26
27
28def handle_upload_file(
29    upload_file: UploadFile, handler: Callable[[Path], None]
30) -> None:
31    tmp_path = save_upload_file_tmp(upload_file)
32    try:
33        handler(tmp_path)  # Do something with the saved temp file
34    finally:
35        tmp_path.unlink()  # Delete the temp file
36