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