from datetime import datetime

from pydantic import BaseModel


class ScheduleCreate(BaseModel):
    type: str  # course | deadline | activity
    title: str
    start_time: datetime
    end_time: datetime | None = None
    location: str | None = None
    description: str | None = None


class ScheduleUpdate(BaseModel):
    type: str | None = None
    title: str | None = None
    start_time: datetime | None = None
    end_time: datetime | None = None
    location: str | None = None
    description: str | None = None


class ScheduleOut(BaseModel):
    id: int
    class_id: int
    type: str
    title: str
    start_time: datetime
    end_time: datetime | None
    location: str | None
    description: str | None
    created_at: datetime

    model_config = {"from_attributes": True}