From c73d7b31ced312e36fec21753ebff180ee15db06 Mon Sep 17 00:00:00 2001 From: Yarne Coppens Date: Thu, 8 Aug 2024 18:13:01 +0200 Subject: [PATCH] Response models can now properly return expansions as well --- src/main.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/main.py b/src/main.py index 5cbd483..9edf181 100644 --- a/src/main.py +++ b/src/main.py @@ -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