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