From 5b87e4552324c13c138cc8982be2b751847225a1 Mon Sep 17 00:00:00 2001 From: Yarne Coppens Date: Fri, 2 Aug 2024 10:37:47 +0200 Subject: [PATCH] Created a wishlist expansion boardgame class --- src/classes/boardgame_classes.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/src/classes/boardgame_classes.py b/src/classes/boardgame_classes.py index 50ac0ec..9d7f8cc 100644 --- a/src/classes/boardgame_classes.py +++ b/src/classes/boardgame_classes.py @@ -8,6 +8,7 @@ class BoardgameType(Enum): COLLECTIONBOARDGAME = 'collectionboardgame' COLLECTIONBOARDGAMEEXPANSION = 'collectionboardgameexpansion' WISHLISTBOARDGAME = 'wishlistboardgame' + WISHLISTBOARDGAMEEXPANSION = 'wishlistboardgameexpansion' class BoardGame(BaseModel): @@ -28,15 +29,18 @@ class BoardGame(BaseModel): class BoardGameExpansion(BoardGame): type: BoardgameType = BoardgameType.BOARDGAMEEXPANSION -class CollectionBoardGame(BoardGame): +class OwnedBoardGame(BoardGame): price_paid: float acquisition_date: date acquired_from: str type: BoardgameType = BoardgameType.COLLECTIONBOARDGAME -class CollectionBoardGameExpansion(CollectionBoardGame): +class OwnedBoardGameExpansion(OwnedBoardGame): type: BoardgameType = BoardgameType.COLLECTIONBOARDGAMEEXPANSION class WishlistBoardGame(BoardGame): wishlist_priority: int - type: BoardgameType = BoardgameType.WISHLISTBOARDGAME \ No newline at end of file + type: BoardgameType = BoardgameType.WISHLISTBOARDGAME + +class WishlistBoardGameExpansion(WishlistBoardGame): + type: BoardgameType = BoardgameType.WISHLISTBOARDGAMEEXPANSION \ No newline at end of file