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


class UserBase(BaseModel):
    username: str
    role: str
    is_active: bool = True


class UserCreate(UserBase):
    pin: str


class UserUpdate(BaseModel):
    username: Optional[str] = None
    role: Optional[str] = None
    is_active: Optional[bool] = None


class UserOut(UserBase):
    id: int
    created_at: datetime

    model_config = {"from_attributes": True}


class AssistantAssignmentOut(BaseModel):
    id: int
    primary_waiter_id: int
    assistant_waiter_id: int
    assigned_at: datetime

    model_config = {"from_attributes": True}