bgg_api/src/main.py

from typing import Union
from fastapi import FastAPI

from src.classes import boardgame_classes
from src.modules import bgg_connection

app = FastAPI()


@app.get("/")
def read_root():
    return {"Hello": "World"}

@app.get("/boardgame/{boardgame_id}", response_model=boardgame_classes.BoardGame)
def get_boardgame_by_id(boardgame_id: int):
    requested_boardgame: boardgame_classes.BoardGame = bgg_connection.get_boardgame(boardgame_id)
    return requested_boardgame

@app.get("/collection", response_model=list[boardgame_classes.CollectionBoardGame])
def get_owned_collection():
    requested_collection: list[boardgame_classes.CollectionBoardGame] = bgg_connection.get_user_owned_collection()
    return requested_collection


@app.get("/wishlist", response_model=list[boardgame_classes.WishlistBoardGame])
def get_wishlist_collection():
    requested_collection: list[boardgame_classes.WishlistBoardGame] = bgg_connection.get_user_wishlist_collection()
    return requested_collection