59 lines
No EOL
1.7 KiB
Python
59 lines
No EOL
1.7 KiB
Python
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
|
|
Play.model_rebuild()
|
|
PlayPublicWithPlayers.model_rebuild() |