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


class DutyCreate(BaseModel):
    date: str
    staff_id: UUID
    duty_type: str
    contact: Optional[str] = None
    note: Optional[str] = None


class DutyUpdate(BaseModel):
    date: Optional[str] = None
    staff_id: Optional[UUID] = None
    duty_type: Optional[str] = None
    contact: Optional[str] = None
    note: Optional[str] = None


class DutyResponse(BaseModel):
    id: UUID
    date: date
    staff_id: UUID
    duty_type: str
    contact: Optional[str] = None
    note: Optional[str] = None

    class Config:
        from_attributes = True