main.py

from fastapi import FastAPI,Request,HTTPException, status
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from typing import List
from models import Episode, Season
import json

# template directory
templates = Jinja2Templates(directory="templates")

# instatiate app
app = FastAPI()
app.mount("/static", StaticFiles(directory="static"), name="static")
# read in json data
with open('data/outlander.json') as f:
    show_data: dict = json.load(f)

# homepage endpoint
@app.get("/",include_in_schema=False)
async def get_homepage(request: Request):
    image_path = request.url_for("static", path="jamie_claire.jpg")
    return templates.TemplateResponse("index.html",{"request":request, "image_path":image_path})

# endpoint for all seasons
@app.get("/seasons", response_model=List[Season])
async def get_seasons():
    return show_data["seasons"]

# endpoint for a single season
@app.get("/seasons/{season_number}", response_model=Season)
async def get_season(season_number: int):
    for season in show_data["seasons"]:
        if season["season_number"] == season_number:
            return season
    raise HTTPException(status_code= status.HTTP_404_NOT_FOUND, detail= "Season not found.")

# endpoint for all episodes from a specific season
@app.get("/seasons/{season_number}/all_episodes", response_model= List[Episode])
async def get_episodes(season_number: int):
    for season in show_data["seasons"]:
        if season["season_number"] == season_number:
            return season["episodes"]
    raise HTTPException(status_code= status.HTTP_404_NOT_FOUND, detail= "Season not found.")

# endpoint for single episode form a specific season
@app.get("/seasons/{season_number}/{episode_number}", response_model=Episode)
async def get_episode(season_number: int, episode_number: int):
    for season in show_data["seasons"]:
        if season["season_number"] == season_number:
            for episode in season["episodes"]:
                if episode["episode_number"] == episode_number:
                    return episode
            raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail = "Episode not found.")
    raise HTTPException(status_code= status.HTTP_404_NOT_FOUND, detail= "Season not found.")