from datetime import datetime

from pydantic import BaseModel


class ClassCreate(BaseModel):
    name: str
    cohort_year: int
    description: str | None = None


class ClassUpdate(BaseModel):
    name: str | None = None
    cohort_year: int | None = None
    description: str | None = None


class ClassOut(BaseModel):
    id: int
    name: str
    cohort_year: int
    description: str | None
    invite_code: str | None = None
    member_count: int = 0
    created_at: datetime

    model_config = {"from_attributes": True}