Yarninator/bgg_api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Response models can now properly return expansions as well

Browse source
This commit is contained in:
Yarne Coppens 2024-08-08 18:13:01 +02:00
parent a7f9b63556
commit c73d7b31ce
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/main.py
View file

@ -1,3 +1,5 @@
from typing import Union
from fastapi import FastAPI
from contextlib import asynccontextmanager
@ -23,13 +25,13 @@ def get_boardgame_by_id(boardgame_id: int):
requested_boardgame: boardgame_classes.BoardGame = data_connection.get_boardgame(boardgame_id)
return requested_boardgame
@app.get("/owned", response_model=list[boardgame_classes.OwnedBoardGame])
@app.get("/owned", response_model=Union[list[boardgame_classes.OwnedBoardGame], list[boardgame_classes.OwnedBoardGameExpansion]])
def get_owned_collection():
requested_collection: list[boardgame_classes.OwnedBoardGame] = data_connection.get_user_owned_collection()
return requested_collection
@app.get("/wishlist", response_model=list[boardgame_classes.WishlistBoardGame])
@app.get("/wishlist", response_model=Union[list[boardgame_classes.WishlistBoardGame], list[boardgame_classes.WishlistBoardGameExpansion]])
def get_wishlist_collection():
requested_collection: list[boardgame_classes.WishlistBoardGame] = data_connection.get_user_wishlist_collection()
return requested_collection