Gave all board game classes a type

This commit is contained in:
Yarne Coppens 2024-08-01 12:16:50 +02:00
parent d0fb364808
commit 77472e8ff4

View file

@ -3,11 +3,11 @@ from datetime import date
from enum import Enum from enum import Enum
class BoardgameType(Enum): class BoardgameType(Enum):
BOARDGAME = 1 BOARDGAME = 'boardgame'
BOARDGAMEEXPANSION = 2 BOARDGAMEEXPANSION = 'boardgameexpansion'
COLLECTIONBOARDGAME = 3 COLLECTIONBOARDGAME = 'collectionboardgame'
COLLECTIONBOARDGAMEEXPANSION = 4 COLLECTIONBOARDGAMEEXPANSION = 'collectionboardgameexpansion'
WISHLISTBOARDGAME = 5 WISHLISTBOARDGAME = 'wishlistboardgame'
class BoardGame(BaseModel): class BoardGame(BaseModel):
@ -23,17 +23,20 @@ class BoardGame(BaseModel):
max_playing_time: int max_playing_time: int
min_age: int min_age: int
all_expansion_ids: list[int] all_expansion_ids: list[int]
type: BoardgameType = BoardgameType.BOARDGAME
class BoardGameExpansion(BoardGame): class BoardGameExpansion(BoardGame):
pass type: BoardgameType = BoardgameType.BOARDGAMEEXPANSION
class CollectionBoardGame(BoardGame): class CollectionBoardGame(BoardGame):
price_paid: float price_paid: float
acquisition_date: date acquisition_date: date
acquired_from: str acquired_from: str
type: BoardgameType = BoardgameType.COLLECTIONBOARDGAME
class CollectionBoardGameExpansion(CollectionBoardGame): class CollectionBoardGameExpansion(CollectionBoardGame):
pass type: BoardgameType = BoardgameType.COLLECTIONBOARDGAMEEXPANSION
class WishlistBoardGame(BoardGame): class WishlistBoardGame(BoardGame):
wishlist_priority: int wishlist_priority: int
type: BoardgameType = BoardgameType.WISHLISTBOARDGAME