Yarninator/bgg_api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Created Play and PlayPlayer tables with link table

Browse source
This commit is contained in:
Yarne Coppens 2024-08-11 20:21:54 +02:00
parent bb1f255c6c
commit 2b7fcdbae3
2 changed files with 21 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/classes/play_classes.py
View file

@ -1,18 +1,27 @@
from pydantic import BaseModel from sqlmodel import Field, SQLModel, Relationship
from typing import Union from typing import Union
from datetime import date from datetime import date
class PlayPlayer(BaseModel): class PlayerPlayLink(SQLModel, table=True):
name: str player_name: str | None = Field(default=None, foreign_key="playplayer.name", primary_key=True)
play_id: int | None = Field(default=None, foreign_key="play.id", primary_key=True)
class PlayPlayer(SQLModel, table=True):
name: str | None = Field(default=None, primary_key=True)
username: str username: str
score: Union[float, None] score: Union[float, None]
first_play : bool first_play : bool
has_won : bool has_won : bool
class Play(BaseModel): plays: list["Play"] = Relationship(back_populates="players", link_model=PlayerPlayLink)
class Play(SQLModel, table=True):
id: int| None = Field(default=None, primary_key=True)
boardgame_id: int boardgame_id: int
players: list[PlayPlayer]
play_date: date play_date: date
duration: int #In minutes duration: int #In minutes
ignore_for_stats : bool ignore_for_stats : bool
location: str location: str
players: list[PlayPlayer] = Relationship(back_populates="plays", link_model=PlayerPlayLink)

7
src/modules/db_connection.py
View file

@ -2,7 +2,7 @@ from sqlmodel import create_engine, SQLModel, Session, select
from src.config import definitions from src.config import definitions
from typing import Union from typing import Union
from src.classes import boardgame_classes from src.classes import boardgame_classes, play_classes
sqlite_url = definitions.SQLITE_URL sqlite_url = definitions.SQLITE_URL
@ -53,6 +53,11 @@ def delete_database():
boardgame_classes.WishlistBoardGame.__table__.drop(engine) boardgame_classes.WishlistBoardGame.__table__.drop(engine)
boardgame_classes.WishlistBoardGameExpansion.__table__.drop(engine) boardgame_classes.WishlistBoardGameExpansion.__table__.drop(engine)
play_classes.Play.__table__.drop(engine)
play_classes.PlayPlayer.__table__.drop(engine)
play_classes.PlayerPlayLink.__table__.drop(engine)
def create_db_and_tables(): def create_db_and_tables():
SQLModel.metadata.create_all(engine) SQLModel.metadata.create_all(engine)