Spaces:
Sleeping
Sleeping
File size: 1,726 Bytes
b1eced0 3e8fd5d b1eced0 3e8fd5d b1eced0 3e8fd5d b1eced0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
from fastapi import Depends
from trauma.api.account.dto import AccountType
from trauma.api.account.model import AccountModel
from trauma.api.common.dto import Paging
from trauma.api.data import facility_router
from trauma.api.data.db_requests import get_facility_by_id, get_all_model_obj, search_facilities_obj
from trauma.api.data.model import EntityModel
from trauma.api.data.schemas import AllFacilitiesWrapper, AllFacilitiesResponse, SearchRequest
from trauma.core.security import PermissionDependency
from trauma.core.wrappers import TraumaResponseWrapper
@facility_router.get('/all')
async def get_all_countries(
_: AccountModel = Depends(PermissionDependency([AccountType.Admin, AccountType.User]))
) -> AllFacilitiesWrapper:
all_facilities = await get_all_model_obj()
response = AllFacilitiesResponse(
paging=Paging(pageSize=len(all_facilities), pageIndex=0, totalCount=len(all_facilities)),
data=all_facilities
)
return AllFacilitiesWrapper(data=response)
@facility_router.post('/search')
async def search_facilities(
data: SearchRequest,
) -> AllFacilitiesWrapper:
countries, total_count = await search_facilities_obj(data)
response = AllFacilitiesResponse(
paging=Paging(pageSize=data.pageSize, pageIndex=data.pageIndex, totalCount=total_count),
data=countries
)
return AllFacilitiesWrapper(data=response)
@facility_router.get('/{facilityId}')
async def get_country(
facilityId: str,
_: AccountModel = Depends(PermissionDependency([AccountType.Admin, AccountType.User]))
) -> TraumaResponseWrapper[EntityModel]:
facility = await get_facility_by_id(facilityId)
return TraumaResponseWrapper(data=facility)
|