from enum import Enum
from datetime import time

from pydantic import BaseModel


class DayOfWeek(Enum):
    MONDAY = 1
    TUESDAY = 2
    WEDNESDAY = 3
    THURSDAY = 4
    FRIDAY = 5
    SATURDAY = 6
    SUNDAY = 7


class TimeSlot(BaseModel):
    startTime: time
    endTime: time


class DaySchedule(BaseModel):
    dayOfWeek: DayOfWeek
    slots: list[TimeSlot] | None = None