itrnsjsonapi.py

import os
import json
import base64
from fastapi import FastAPI, Request, HTTPException, Response
from fastapi.responses import HTMLResponse, FileResponse
from fastapi.templating import Jinja2Templates

data_root = "/home/snehring/projects/itrnsjsonimages/data"

app = FastAPI()


@app.get("/")
def hello():
    return {"msg": "Hello, World!"}


@app.get("/images/{camera_id}", response_class=HTMLResponse)
def read_id_dir(camera_id: str, request: Request):
    try:
        entries = os.listdir(os.path.join(data_root, camera_id))
        return Jinja2Templates(directory="templates").TemplateResponse(
            name="dir_list.j2",
            request=request,
            context={"files": entries, "camera": camera_id},
        )
    except FileNotFoundError as e:
        raise HTTPException(status_code=404) from e


@app.get("/images/{camera_id}/{date}", response_class=HTMLResponse)
def read_camera_date_dir(camera_id: str, date: str, request: Request):
    try:
        with open(
            os.path.join(data_root, camera_id, date, f"{camera_id}-{date}.json"),
            encoding="utf-8",
            mode="r",
        ) as json_file:
            entries = json.load(json_file).keys()
        return Jinja2Templates(directory="templates").TemplateResponse(
            name="dir_list.j2",
            request=request,
            context={"files": entries, "camera": camera_id, "date": date},
        )
    except FileNotFoundError as e:
        raise HTTPException(status_code=404) from e


@app.get("/images/{camera_id}/{date}/{image_id}", response_class=FileResponse)
def get_image_from_path(camera_id: str, date: str, image_id: str, request: Request):
    try:
        with open(
            os.path.join(data_root, camera_id, date, f"{camera_id}-{date}.json"),
            encoding="utf-8",
            mode="r",
        ) as json_file:
            jpeg = json.load(json_file)[image_id]
        return Response(media_type="image/jpeg", content=base64.b64decode(jpeg))
    except FileNotFoundError as e:
        print("file not found")
        raise HTTPException(status_code=404) from e
    except KeyError as e:
        print("key errro")
        raise HTTPException(status_code=404) from e