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