Gave all board game classes a type
This commit is contained in:
parent
d0fb364808
commit
77472e8ff4
1 changed files with 11 additions and 8 deletions
|
|
@ -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
|
||||
Loading…
Reference in a new issue