from typing import Optional

from sqlmodel import SQLModel, Field

from backend.app.db.base_class import Base


class MatchBase(SQLModel):
    team_1_result: int | None
    team_2_result: int | None


class Match(MatchBase, Base, table=True):
    team_1: Optional[int] = Field(default=None, foreign_key="team.id")
    team_2: Optional[int] = Field(default=None, foreign_key="team.id")
    winner: Optional[int] = Field(default=None, foreign_key="team.id")
    day:  Optional[int] = Field(default=None, foreign_key="matchday.id")


class MatchCreate(MatchBase):
    team_1: Optional[int]
    team_2: Optional[int]
    winner: Optional[int]
    day: Optional[int]

class MatchUpdate(MatchBase):
    team_1: Optional[int]
    team_2: Optional[int]
    winner: Optional[int]
    day: Optional[int]