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