FastApiTesting/files.py at main · msaigla/FastApiTesting · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from fastapi import FastAPI, File, UploadFile

from fastapi.responses import StreamingResponse, FileResponse


app = FastAPI()

@app.post("/files")
async def upload_file(uploaded_file: UploadFile):
    file = uploaded_file.file
    filename = uploaded_file.filename
    with open(filename, 'wb') as f:
        f.write(file.read())


@app.post("/multiple_files")
async def upload_file(uploaded_files: list[UploadFile]):
    for uploaded_file in uploaded_files:
        file = uploaded_file.file
        filename = uploaded_file.filename
        with open(filename, 'wb') as f:
            f.write(file.read())


@app.get("/files/{filename}")
async def get_file(filename: str):
    return FileResponse(filename)


def iterfile(filename: str):
    with open(filename, 'rb') as f:
        while chunk := f.read(1024 * 1024):
            yield chunk


@app.get("/files/steaming/{filename}")
async def get_streaming_file(filename: str):
    return StreamingResponse(iterfile(filename), media_type="video/mp4")