trauma/api/data/db_requests.py

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}, {"embedding": 0})
    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({}, {"embedding": 0}).sort("_id", sort_v).to_list(length=None)
    return objects