From 655b1b5cf0fc9dfa8c5efe3b4e5bf20ee5517415 Mon Sep 17 00:00:00 2001
From: Yarne Coppens <git.w8x3v@simplelogin.com>
Date: Thu, 1 Aug 2024 12:05:51 +0200
Subject: [PATCH] Created board game type Enum

---
 bgg_connection.py    | 2 +-
 classes/boardgame.py | 9 +++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/bgg_connection.py b/bgg_connection.py
index a58e20b..2ef407a 100644
--- a/bgg_connection.py
+++ b/bgg_connection.py
@@ -3,7 +3,7 @@ import xml.etree.ElementTree as ET
 from pydantic import HttpUrl
 import requests
 
-from classes.boardgame import BoardGame, BoardGameExpansion
+from classes.boardgame import BoardGame, BoardGameExpansion, CollectionBoardGame, WishlistBoardGame, BoardgameType
 import auth_manager
 
 authenticated_session: requests.Session = requests.Session()
diff --git a/classes/boardgame.py b/classes/boardgame.py
index fb0b9c7..ff32ccd 100644
--- a/classes/boardgame.py
+++ b/classes/boardgame.py
@@ -1,5 +1,14 @@
 from pydantic import BaseModel, HttpUrl
 from datetime import date
+from enum import Enum
+
+class BoardgameType(Enum):
+    BOARDGAME = 1
+    BOARDGAMEEXPANSION = 2
+    COLLECTIONBOARDGAME = 3
+    COLLECTIONBOARDGAMEEXPANSION = 4
+    WISHLISTBOARDGAME = 5
+
 
 class BoardGame(BaseModel):
     id: int