trauma/api/message/schemas.py

from pydantic import BaseModel

from trauma.api.common.dto import Paging
from trauma.api.message.model import MessageModel
from trauma.core.wrappers import TraumaResponseWrapper


class CreateMessageRequest(BaseModel):
    text: str
    fileUrl: str | None = None


class MessageWrapper(TraumaResponseWrapper[MessageModel]):
    pass


class AllMessageResponse(BaseModel):
    paging: Paging
    data: list[MessageModel]


class AllMessageWrapper(TraumaResponseWrapper[AllMessageResponse]):
    pass