from datetime import time
from pydantic import BaseModel
from typing import Optional
from uuid import UUID


class ShiftCreate(BaseModel):
    name: str
    start_time: str
    end_time: str
    enabled: bool = True
    remark: Optional[str] = None


class ShiftUpdate(BaseModel):
    name: Optional[str] = None
    start_time: Optional[str] = None
    end_time: Optional[str] = None
    enabled: Optional[bool] = None
    remark: Optional[str] = None


class ShiftResponse(BaseModel):
    id: UUID
    name: str
    start_time: time
    end_time: time
    enabled: bool
    remark: Optional[str] = None

    class Config:
        from_attributes = True