Update api/routes.py

api/routes.py

@@ -9,6 +9,7 @@ from bson.errors import InvalidId
 from typing import Optional, List
 from pydantic import BaseModel
 import httpx
+from pymongo import UpdateOne
 
 router = APIRouter()
 
@@ -132,6 +133,7 @@ async def list_patients(current_user: dict = Depends(get_current_user)):
     return patients
 
 # --- IMPORT FROM PUBLIC FHIR API ---
+
 @router.post("/ehr/fetch-from-api")
 async def fetch_and_store_patients_from_fhir():
     fhir_api = "https://hapi.fhir.org/baseR4/Patient?_count=50"
@@ -148,29 +150,42 @@ async def fetch_and_store_patients_from_fhir():
         if not entries:
             return {"message": "No patients found."}
 
-        patients = []
+        operations = []
 
         for entry in entries:
             resource = entry.get("resource", {})
             name = resource.get("name", [{}])[0]
             full_name = f"{name.get('given', [''])[0]} {name.get('family', '')}".strip()
 
-            patient = {
-                "fhir_id": resource.get("id"),
-                "full_name": full_name,
-                "gender": resource.get("gender"),
-                "date_of_birth": resource.get("birthDate"),
-                "created_at": datetime.utcnow()
+            fhir_id = resource.get("id")
+            gender = resource.get("gender")
+            birth_date = resource.get("birthDate")
+
+            operations.append(UpdateOne(
+                {"fhir_id": fhir_id},
+                {"$set": {
+                    "fhir_id": fhir_id,
+                    "full_name": full_name,
+                    "gender": gender,
+                    "date_of_birth": birth_date,
+                    "created_at": datetime.utcnow()
+                }},
+                upsert=True  # 👈 This avoids duplicates
+            ))
+
+        if operations:
+            result = await patients_collection.bulk_write(operations)
+            return {
+                "message": "Fetched and stored successfully",
+                "inserted": result.upserted_count,
+                "modified": result.modified_count
             }
-            patients.append(patient)
 
-        result = await patients_collection.insert_many(patients)
-        return {"message": "Fetched and stored successfully", "count": len(result.inserted_ids)}
+        return {"message": "No patients to store"}
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-
 @router.get("/ehr/fhir-patients")
 async def list_fhir_patients():
     cursor = patients_collection.find({"fhir_id": {"$exists": True}})