from datetime import datetime
from typing import Optional
from pydantic import BaseModel


class ScheduleResponse(BaseModel):
    id: int
    name: str
    cron_timer: str
    status: str
    last_run_at: Optional[datetime] = None
    next_run_at: Optional[datetime] = None
    last_status: Optional[str] = None
    recorded_at: datetime

    class Config:
        from_attributes = True


class ScheduleUpdate(BaseModel):
    cron_timer: str


class ScheduleLogResponse(BaseModel):
    id: int
    schedule_id: int
    schedule_name: str
    started_at: datetime
    ended_at: Optional[datetime] = None
    status: str

    class Config:
        from_attributes = True