add facility

trauma/__init__.py

@@ -24,6 +24,9 @@ def create_app() -> FastAPI:
     from trauma.api.security import security_router
     app.include_router(security_router, tags=['security'])
 
+    from trauma.api.data import facility_router
+    app.include_router(facility_router, tags=['facility'])
+
     app.add_middleware(
         CORSMiddleware,
         allow_origins=["*"],

trauma/api/data/__init__.py

@@ -0,0 +1,7 @@
+from fastapi.routing import APIRouter
+
+facility_router = APIRouter(
+    prefix="/api/facility", tags=["chat"]
+)
+
+from . import views

trauma/api/data/db_requests.py

@@ -0,0 +1,17 @@
+from fastapi import HTTPException
+
+from trauma.api.data.model import EntityModel
+from trauma.core.config import settings
+
+
+async def get_facility_by_id(facility_id: str) -> EntityModel:
+    facility = await settings.DB_CLIENT.entities.find_one({"id": facility_id})
+    if not facility:
+        raise HTTPException(status_code=404, detail="Country with specified id doesn't exists.")
+    return EntityModel.from_mongo(facility)
+
+
+async def get_all_model_obj() -> list[EntityModel]:
+    sort_v = -1
+    objects = await settings.DB_CLIENT.entities.find().sort("_id", sort_v).to_list(length=None)
+    return objects

trauma/api/data/schemas.py

@@ -0,0 +1,13 @@
+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

trauma/api/data/views.py

@@ -0,0 +1,32 @@
+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
+from trauma.api.data.model import EntityModel
+from trauma.api.data.schemas import AllFacilitiesWrapper, AllFacilitiesResponse
+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.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)