from datetime import datetime

from sqlmodel import Field, Relationship, SQLModel


class Event(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    title: str = Field(nullable=False)
    event_time: datetime = Field(nullable=False)
    registration_deadline: datetime = Field(nullable=False)
    description: str
    recipe_link: str

    registrations: list["Registration"] = Relationship()


class Household(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    name: str = Field(nullable=False)


class Registration(SQLModel, table=True):
    event_id: int | None = Field(default=None, foreign_key="event.id", primary_key=True)
    household_id: int | None = Field(
        default=None, foreign_key="household.id", primary_key=True
    )
    num_adult_meals: int
    num_children_meals: int
    num_small_children_meals: int
    comment: str | None

    household: Household = Relationship()