From 77472e8ff49d60eda3912686e22b626ae10d28cf Mon Sep 17 00:00:00 2001
From: Yarne Coppens <git.w8x3v@simplelogin.com>
Date: Thu, 1 Aug 2024 12:16:50 +0200
Subject: [PATCH] Gave all board game classes a type

---
 classes/boardgame.py | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/classes/boardgame.py b/classes/boardgame.py
index ff32ccd..50ac0ec 100644
--- a/classes/boardgame.py
+++ b/classes/boardgame.py
@@ -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
\ No newline at end of file
+    wishlist_priority: int
+    type: BoardgameType = BoardgameType.WISHLISTBOARDGAME
\ No newline at end of file