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

0 vote263 views1 answer
Python dictionary get keys with condition on value
Python
1 vote312 views1 answer
Calculate pointbiseral correlation scipy
Python
39 votes276 views2 answers
Create text in python if not exists
Python
74 votes268 views1 answer
How to update pip in anaconda prompt
Python
0 vote273 views1 answer
Terminal commands for install python on cpanel
Python
0 vote270 views1 answer
How can I keep the pyqt GUI responsive when performing a cpu intensive task?
Python
0 vote285 views1 answer
Int to hex python without 0x
Python
0 vote283 views3 answers
Add a new button in the index of the page wagtail
Python
0 vote269 views1 answer
Numpy rolling
Python
0 vote264 views1 answer
Python default summary statistics for all columns
Python