from datetime import datetime

from sqlmodel import SQLModel, Field, Relationship


class Submission(SQLModel, table=True):
    __tablename__ = "submissions"

    id: int | None = Field(default=None, primary_key=True)
    team_id: int = Field(foreign_key="teams.id")
    task_id: int = Field(foreign_key="tasks.id")
    description: str
    file_url: str | None = None
    submitted_at: datetime = Field(default_factory=datetime.utcnow)

    # Relationships
    team: "Team" = Relationship(back_populates="submissions")
    task: "Task" = Relationship(back_populates="submissions")