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 vote161 views1 answer
Jupyter notebook don't show plot
Python
0 vote173 views1 answer
Wifite subsystem
Python
0 vote173 views1 answer
Module 'cv2.cv2' has no attribute 'imWrite'
Python
0 vote163 views2 answers
Numpy int64
Python
10 votes176 views5 answers
Find nan values in a column pandas
Python
1 vote171 views1 answer
Add one to all elements in list python
Python
67 votes172 views4 answers
List slicing reverse python
Python
13 votes179 views4 answers
Even in python
Python
19 votes166 views3 answers
Debugger online for python
Python
18 votes183 views1 answer
Convert column in pandas to datetime
Python