Skip to content

yavuzakyazici/fastapi_video

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

An example of streaming file/video for FastAPI. This example demonstrates how to add a streaming file support to FastAPI with some extra capabilities. This mini example also shows how to declare static files directory for the app,

app.mount("/assets", StaticFiles(directory="assets"), name="assets")

Folder structure: Screenshot 2024-03-10 at 16 51 20

how to upload file to that directory,

@app.post("/video")
async def create_upload_file(file: UploadFile):
    f = open(f'{video_folder_path+file.filename}', 'wb')
    content = await file.read()
    f.write(content)
    # return the url to the uploaded file relative to the root of the domain
    video_url=fastapi_base_url+"get_video_by_name/"
    return {"video_url": urljoin(video_url, file.filename)}

how to stream from that directory

@app.get("/get_video_by_name/{video_name}")
def get_video(video_name:str, request: Request):
    if video_name is not None:
        video_url=video_folder_path+video_name
        return range_requests_response(
            request, file_path=video_url, content_type="video/mp4"
        )
    else:
        raise HTTPException(status_code=404, detail="No videos found!")

and... how to send out 304 unmodified header once the file is cached.

To get it working on your machine you can clone the repository. You can use the included requirements.txt to install dependencies with pip.

Here is our main.py with "/video" route for uploading video to assets/video directory main.py has also a route for getting the video. Here is what it looks like.

from fastapi import FastAPI, Request, UploadFile, HTTPException
from fastapi.staticfiles import StaticFiles
from urllib.parse import urljoin
from file_stream import range_requests_response

video_folder_path = "assets/video/"
fastapi_base_url = "http://127.0.0.1:8000/"

app = FastAPI()

app.mount("/assets", StaticFiles(directory="assets"), name="assets")

@app.get("/")
def greet():
    return {"message": "Welcome to FastAPI video exmaple."} 

@app.post("/video")
async def create_upload_file(file: UploadFile):
    f = open(f'{video_folder_path+file.filename}', 'wb')
    content = await file.read()
    f.write(content)
    # return the url to the uploaded file relative to the root of the domain
    video_url=fastapi_base_url+"get_video_by_name/"
    return {"video_url": urljoin(video_url, file.filename)}


@app.get("/get_video_by_name/{video_name}")
def get_video(video_name:str, request: Request):
    if video_name is not None:
        video_url=video_folder_path+video_name
        return range_requests_response(
            request, file_path=video_url, content_type="video/mp4"
        )
    else:
        raise HTTPException(status_code=404, detail="No videos found!")

Here is our file_stream.py Thanks to angel-langdon for the original reusable video streamer code. I added streaming with partial 206 header and eTag support. eTag was generated from from last_modified and file_size with md5_hexdigest function. This way, if file size or last_modified changes, so does the eTag.

import os
from typing import BinaryIO
from starlette._compat import md5_hexdigest
from fastapi import HTTPException, Request, status
from fastapi.responses import StreamingResponse, Response
from datetime import datetime

def send_bytes_range_requests(
    file_obj: BinaryIO, start: int, end: int, chunk_size: int = 8*1024
):
    """Send a file in chunks using Range Requests specification RFC7233

    `start` and `end` parameters are inclusive due to specification
    """
    with file_obj as f:
        f.seek(start)
        while (pos := f.tell()) <= end:
            read_size = min(chunk_size, end + 1 - pos)
            yield f.read(read_size)


def _get_range_header(range_header: str, file_size: int) -> tuple[int, int]:
    def _invalid_range():
        return HTTPException(
            status.HTTP_416_REQUESTED_RANGE_NOT_SATISFIABLE,
            detail=f"Invalid request range (Range:{range_header!r})",
        )

    try:
        h = range_header.replace("bytes=", "").split("-")
        start = int(h[0]) if h[0] != "" else 0
        end = int(h[1]) if h[1] != "" else file_size - 1
    except ValueError:
        raise _invalid_range()

    if start > end or start < 0 or end > file_size - 1:
        raise _invalid_range()
    return start, end


def range_requests_response(
    request: Request, file_path: str, content_type: str
):
    """Returns StreamingResponse using Range Requests of a given file"""

    file_size = os.stat(file_path).st_size
    range_header = request.headers.get("range")

    """Compose etag from last_modified and file_size"""
    last_modified = datetime.fromtimestamp(os.stat(file_path).st_mtime).strftime("%a, %d %b %Y %H:%M:%S")
    etag_base = str(last_modified) + "-" + str(file_size)
    etag = f'"{md5_hexdigest(etag_base.encode(), usedforsecurity=False)}"'

    """Check if the browser sent etag matches the videos etag"""
    request_if_non_match_etag = request.headers.get("if-none-match")

    """if there is a match return 304 unmodified instead of 206 response without video file"""
    if request_if_non_match_etag == etag:
        headers = {
            "cache-control": "public, max-age=86400, stale-while-revalidate=2592000",
            "etag" : etag,
            "last-modified":str(last_modified),
        }
        status_code = status.HTTP_304_NOT_MODIFIED
        return Response(None, status_code=status_code, headers=headers)

    headers = {
        "etag" : etag,
        "content-type": content_type,
        "accept-ranges": "bytes",
        "content-encoding": "identity",
        "content-length": str(file_size),
        "access-control-expose-headers": (
        "content-type, accept-ranges, content-length, "
        "content-range, content-encoding"
        ),
    }
    start = 0
    end = file_size - 1
    status_code = status.HTTP_200_OK


    if range_header is not None:
        start, end = _get_range_header(range_header, file_size)
        size = end - start + 1
        headers["content-length"] = str(size)
        headers["content-range"] = f"bytes {start}-{end}/{file_size}"
        status_code = status.HTTP_206_PARTIAL_CONTENT

    return StreamingResponse(
        send_bytes_range_requests(open(file_path, mode="rb"), start, end),
        headers=headers,
        status_code=status_code,
    )

First request gets 200 response for the file.. 1

If you inspect the video on Google Chrome Dev tools when you first time you download the video file, network tab clearly show 206 partial header meaning streaming with smaller chunks. 2

Server also sends an eTag along with 206 header.. 3

Second time, if the browser finds a matching eTag, server sends back 304 unmodified header. this is done via the if-none-match header 4 5

You can see it in the code clearly here:

 """Check if the browser sent etag matches the videos etag"""
    request_if_non_match_etag = request.headers.get("if-none-match")

    """if there is a match return 304 unmodified instead of 206 response without video file"""
    if request_if_non_match_etag == etag:
        headers = {
            "cache-control": "public, max-age=86400, stale-while-revalidate=2592000",
            "etag" : etag,
            "last-modified":str(last_modified),
        }
        status_code = status.HTTP_304_NOT_MODIFIED
        return Response(None, status_code=status_code, headers=headers)

If there is no matching eTag found this means that the file is modified since eTag was and MD5 based on last-modified and the file size. Which you can also see clearly in the code here:

   """Compose etag from last_modified and file_size"""
    last_modified = datetime.fromtimestamp(os.stat(file_path).st_mtime).strftime("%a, %d %b %Y %H:%M:%S")
    etag_base = str(last_modified) + "-" + str(file_size)
    etag = f'"{md5_hexdigest(etag_base.encode(), usedforsecurity=False)}"'

However, if the eTag is the same ( look at the code above ) 304 unmodified header is sent and cached video is used.

Unfortunately does not work on Safari meaning Safari does not send if-none-match header. However, I tested on Chrome, Opera and Mozilla Developer edition and it works :)

I hope this helps someone :)

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages