bgg_api/src/classes/play_classes.py

from sqlmodel import Field, SQLModel, Relationship
from typing import Union
from datetime import date

class PlayPlayerBase(SQLModel):
    name: str
    username: str
    score: Union[float, None]
    first_play : bool
    has_won : bool
    play_id : int = Field(default=None, foreign_key="play.id")


class PlayPlayer(PlayPlayerBase, table=True):
    id: int | None = Field(default=None, primary_key=True)
      
    play: "Play" = Relationship(back_populates="players")

class PlayPlayerPublic(PlayPlayerBase):
    id: int
    

class PlayPlayerPublicWithPlay(PlayPlayerPublic):
    play: "PlayPublic"


class PlayBase(SQLModel):
    boardgame_id: int | None = Field(default=None, foreign_key="boardgame.id")
    expansion_id: int | None = Field(default=None, foreign_key="boardgameexpansion.id")
    play_date: date
    duration: int #In minutes
    ignore_for_stats : bool
    location: str


class Play(PlayBase, table=True):
    id: int | None = Field(default=None, primary_key=True)
    
    players: list[PlayPlayer] = Relationship(back_populates="play")
    boardgame: Union["BoardGame", None] = Relationship(back_populates="plays")
    boardgameexpansion: Union["BoardGameExpansion", None] = Relationship(back_populates="plays")

    model_config = {    
        'validate_assignment':True
    }

class PlayPublic(PlayBase):
    id: int
    

class PlayPublicWithPlayers(PlayPublic):
    players: list[PlayPlayerPublic] = []
    boardgame: Union["BoardGameNoPlays", None]
    boardgameexpansion: Union["BoardGameExpansionNoPlays", None]


from src.classes.boardgame_classes import BoardGame, BoardGameExpansion, BoardGameNoPlays, BoardGameExpansionNoPlays
from src.classes.people_classes import Designer
Play.model_rebuild()
PlayPublicWithPlayers.model_rebuild()