Fastapi upload file save

import shutil
from pathlib import Path
from tempfile import NamedTemporaryFile
from typing import Callable

from fastapi import UploadFile


def save_upload_file(upload_file: UploadFile, destination: Path) -> None:
    try:
        with destination.open("wb") as buffer:
            shutil.copyfileobj(upload_file.file, buffer)
    finally:
        upload_file.file.close()


def save_upload_file_tmp(upload_file: UploadFile) -> Path:
    try:
        suffix = Path(upload_file.filename).suffix
        with NamedTemporaryFile(delete=False, suffix=suffix) as tmp:
            shutil.copyfileobj(upload_file.file, tmp)
            tmp_path = Path(tmp.name)
    finally:
        upload_file.file.close()
    return tmp_path


def handle_upload_file(
    upload_file: UploadFile, handler: Callable[[Path], None]
) -> None:
    tmp_path = save_upload_file_tmp(upload_file)
    try:
        handler(tmp_path)  # Do something with the saved temp file
    finally:
        tmp_path.unlink()  # Delete the temp file

Source: stackoverflow.com

Similar codes

5 votes368 views3 answers
Openpyxl delete rows
Python
1 vote308 views1 answer
Write a function to split a string and convert it into an array of words.
Python
7 votes374 views2 answers
Sort list by attribute python
Python
5 votes399 views4 answers
Discord.py find channel by id
Python
0 vote335 views1 answer
Python declare set
Python
4 votes325 views2 answers
Python wait until
Python
0 vote395 views1 answer
Resizing django embed player
Python Django
3 votes316 views1 answer
Tick labels vertical matplotlib
Python
0 vote286 views1 answer
Error: getaddrinfo ENOTFOUND www.python.org www.python.org:443 Downloading Python failed. Error: { Error: getaddrinfo ENOTFOUND www.python.org www.python.org:443
Python
0 vote360 views1 answer
Return object in python
Python