"""Pydantic schemas for mock WhatsApp endpoint."""

from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel


class MockWhatsAppRequest(BaseModel):
    command: str
    from_number: str | None = None
    ticket_id: int | None = None


class MockWhatsAppResponse(BaseModel):
    status: str = "received"
    message: str = "Command logged successfully"


class MockWhatsAppLogEntry(BaseModel):
    id: int
    command: str
    from_number: str | None
    ticket_id: int | None
    received_at: datetime

    model_config = {"from_attributes": True}