trauma/api/data/schemas.py

from pydantic import BaseModel

from trauma.api.common.dto import Paging
from trauma.api.data.model import EntityModel
from trauma.core.wrappers import TraumaResponseWrapper


class AllFacilitiesResponse(BaseModel):
    paging: Paging
    data: list[EntityModel]


class AllFacilitiesWrapper(TraumaResponseWrapper[AllFacilitiesResponse]):
    pass


class SearchRequest(BaseModel):
    name: str
    pageSize: int
    pageIndex: int