Update api/routes.py

api/routes.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, HTTPException, Depends, Body
+from fastapi import APIRouter, HTTPException, Depends, Body, UploadFile, File
 from fastapi.security import OAuth2PasswordRequestForm
 from models.schemas import SignupForm, TokenResponse, PatientCreate, DoctorCreate, AppointmentCreate
 from db.mongo import users_collection, patients_collection, appointments_collection
@@ -6,8 +6,12 @@ from core.security import hash_password, verify_password, create_access_token, g
 from datetime import datetime
 from bson import ObjectId
 from bson.errors import InvalidId
-from typing import Optional
+from typing import Optional, List
 from pydantic import BaseModel
+from pathlib import Path
+import csv
+import io
+import json
 
 router = APIRouter()
 
@@ -88,7 +92,7 @@ async def get_me(current_user: dict = Depends(get_current_user)):
         raise HTTPException(status_code=404, detail="User not found")
 
     return {
-        "id": str(user["_id"]),  # ✅ needed in frontend to submit appointment
+        "id": str(user["_id"]),
         "email": user["email"],
         "full_name": user.get("full_name", ""),
         "role": user.get("role", "unknown"),
@@ -118,15 +122,15 @@ async def list_patients(current_user: dict = Depends(get_current_user)):
     if current_user.get("role") != "doctor":
         raise HTTPException(status_code=403, detail="Only doctors can view patients")
 
-    patients_cursor = patients_collection.find({"created_by": current_user["email"]})
+    cursor = patients_collection.find({"created_by": current_user["email"]})
     patients = []
-    async for patient in patients_cursor:
+    async for p in cursor:
         patients.append({
-            "id": str(patient["_id"]),
-            "full_name": patient.get("full_name", ""),
-            "date_of_birth": patient.get("date_of_birth"),
-            "gender": patient.get("gender", ""),
-            "notes": patient.get("notes", "")
+            "id": str(p["_id"]),
+            "full_name": p.get("full_name", ""),
+            "date_of_birth": p.get("date_of_birth"),
+            "gender": p.get("gender", ""),
+            "notes": p.get("notes", "")
         })
     return patients
 
@@ -144,16 +148,14 @@ async def create_appointment(data: AppointmentCreate, current_user: dict = Depen
     if current_user.get("role") != "patient":
         raise HTTPException(status_code=403, detail="Only patients can book appointments")
 
-    # Get patient user info
     patient_user = await users_collection.find_one({"email": current_user["email"]})
     if not patient_user:
         raise HTTPException(status_code=404, detail="Patient user not found")
 
-    # Insert appointment
     appointment_doc = {
         "patient_id": patient_user["_id"],
         "doctor_id": ObjectId(data.doctor_id),
-        "date": data.date,"date": datetime.combine(data.date, datetime.min.time()),
+        "date": datetime.combine(data.date, datetime.min.time()),
         "time": data.time.strftime("%H:%M:%S"),
         "reason": data.reason,
         "created_by": current_user["email"],
@@ -161,30 +163,8 @@ async def create_appointment(data: AppointmentCreate, current_user: dict = Depen
     }
     await appointments_collection.insert_one(appointment_doc)
 
-    # Auto-add to doctor's patient list if not already
-    existing = await patients_collection.find_one({
-        "user_email": current_user["email"],
-        "created_by": await get_doctor_email_by_id(data.doctor_id)
-    })
-    if not existing:
-        await patients_collection.insert_one({
-            "full_name": patient_user.get("full_name", ""),
-            "user_email": patient_user["email"],
-            "gender": "",
-            "created_by": await get_doctor_email_by_id(data.doctor_id),
-            "created_at": datetime.utcnow()
-        })
-
     return {"message": "Appointment booked successfully"}
 
-# --- Helper function ---
-async def get_doctor_email_by_id(doctor_id: str) -> Optional[str]:
-    try:
-        doc = await users_collection.find_one({"_id": ObjectId(doctor_id), "role": "doctor"})
-        return doc["email"] if doc else None
-    except Exception:
-        return None
-
 # --- LIST DOCTOR'S APPOINTMENTS ---
 @router.get("/appointments/doctor")
 async def list_doctor_appointments(current_user: dict = Depends(get_current_user)):
@@ -193,13 +173,11 @@ async def list_doctor_appointments(current_user: dict = Depends(get_current_user
 
     cursor = appointments_collection.find({"doctor_id": ObjectId(current_user["_id"])})
     appointments = []
-
     async for a in cursor:
         try:
             patient_id = a.get("patient_id")
             if not isinstance(patient_id, ObjectId):
                 patient_id = ObjectId(patient_id)
-
             patient = await users_collection.find_one({"_id": patient_id})
         except Exception:
             patient = None
@@ -231,3 +209,65 @@ async def list_my_appointments(current_user: dict = Depends(get_current_user)):
     patient_id = user.get("_id")
     cursor = appointments_collection.find({"patient_id": patient_id})
     return [{**a, "_id": str(a["_id"]), "patient_id": str(a["patient_id"]), "doctor_id": str(a["doctor_id"])} async for a in cursor]
+
+# === UTIL: Parse Synthea CSV ===
+def parse_synthea_patient_csv(file: UploadFile) -> List[dict]:
+    decoded = file.file.read().decode("utf-8")
+    reader = csv.DictReader(io.StringIO(decoded))
+    patients = []
+
+    for row in reader:
+        patients.append({
+            "synthea_id": row.get("Id"),
+            "full_name": f"{row.get('FIRST', '')} {row.get('LAST', '')}".strip(),
+            "gender": row.get("GENDER"),
+            "date_of_birth": datetime.strptime(row.get("BIRTHDATE"), "%Y-%m-%dT%H:%M:%SZ"),
+            "address": row.get("ADDRESS"),
+            "city": row.get("CITY"),
+            "state": row.get("STATE"),
+            "zip": row.get("ZIP"),
+            "created_at": datetime.utcnow()
+        })
+
+    return patients
+
+# === API: Upload and import patients.csv ===
+@router.post("/ehr/import-patients")
+async def import_synthea_patients(file: UploadFile = File(...)):
+    if not file.filename.endswith(".csv"):
+        raise HTTPException(status_code=400, detail="Only CSV files are supported")
+
+    try:
+        patients = parse_synthea_patient_csv(file)
+        result = await patients_collection.insert_many(patients)
+        return {"inserted_count": len(result.inserted_ids)}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+# === GET all Synthea patients ===
+@router.get("/ehr/synthea-patients")
+async def list_synthea_patients():
+    cursor = patients_collection.find({"synthea_id": {"$exists": True}})
+    patients = []
+    async for p in cursor:
+        patients.append({
+            "id": str(p["_id"]),
+            "synthea_id": p.get("synthea_id"),
+            "full_name": p.get("full_name"),
+            "gender": p.get("gender"),
+            "date_of_birth": p.get("date_of_birth"),
+        })
+    return patients
+
+# === GET individual Synthea patient ===
+@router.get("/ehr/synthea-patient/{id}")
+async def get_synthea_patient(id: str):
+    try:
+        patient = await patients_collection.find_one({"_id": ObjectId(id)})
+        if not patient:
+            raise HTTPException(status_code=404, detail="Patient not found")
+        patient["id"] = str(patient["_id"])
+        del patient["_id"]
+        return patient
+    except Exception:
+        raise HTTPException(status_code=400, detail="Invalid patient ID")