Update api/routes.py

api/routes.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, HTTPException, Depends, Body, UploadFile, File
+from fastapi import APIRouter, HTTPException, Depends, Body
 from fastapi.security import OAuth2PasswordRequestForm
 from models.schemas import SignupForm, TokenResponse, PatientCreate, DoctorCreate, AppointmentCreate
 from db.mongo import users_collection, patients_collection, appointments_collection
@@ -8,10 +8,7 @@ from bson import ObjectId
 from bson.errors import InvalidId
 from typing import Optional, List
 from pydantic import BaseModel
-from pathlib import Path
-import csv
-import io
-import json
+import httpx
 
 router = APIRouter()
 
@@ -134,176 +131,41 @@ async def list_patients(current_user: dict = Depends(get_current_user)):
         })
     return patients
 
-# --- COUNT PATIENTS ---
-@router.get("/patients-count")
-async def count_patients(current_user: dict = Depends(get_current_user)):
-    if current_user.get("role") != "doctor":
-        raise HTTPException(status_code=403, detail="Only doctors can count patients")
-    count = await patients_collection.count_documents({"created_by": current_user["email"]})
-    return {"count": count}
-
-# --- CREATE APPOINTMENT ---
-@router.post("/appointments")
-async def create_appointment(data: AppointmentCreate, current_user: dict = Depends(get_current_user)):
-    if current_user.get("role") != "patient":
-        raise HTTPException(status_code=403, detail="Only patients can book appointments")
-
-    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")
-
-    appointment_doc = {
-        "patient_id": patient_user["_id"],
-        "doctor_id": ObjectId(data.doctor_id),
-        "date": datetime.combine(data.date, datetime.min.time()),
-        "time": data.time.strftime("%H:%M:%S"),
-        "reason": data.reason,
-        "created_by": current_user["email"],
-        "created_at": datetime.utcnow()
-    }
-    await appointments_collection.insert_one(appointment_doc)
-
-    return {"message": "Appointment booked successfully"}
-
-# --- LIST DOCTOR'S APPOINTMENTS ---
-@router.get("/appointments/doctor")
-async def list_doctor_appointments(current_user: dict = Depends(get_current_user)):
-    if current_user.get("role") != "doctor":
-        raise HTTPException(status_code=403, detail="Only doctors can view this")
-
-    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
-
-        appointments.append({
-            "_id": str(a["_id"]),
-            "doctor_id": str(a["doctor_id"]),
-            "patient": {
-                "full_name": patient.get("full_name", "Unknown") if patient else "Unknown",
-                "email": patient.get("email", "") if patient else "",
-            },
-            "date": a.get("date").strftime("%Y-%m-%d") if isinstance(a.get("date"), datetime) else a.get("date"),
-            "time": a.get("time", ""),
-            "reason": a.get("reason", "")
-        })
-
-    return appointments
-
-# --- LIST PATIENT'S APPOINTMENTS ---
-@router.get("/appointments/patient")
-async def list_my_appointments(current_user: dict = Depends(get_current_user)):
-    if current_user.get("role") != "patient":
-        raise HTTPException(status_code=403, detail="Only patients can view their appointments")
-
-    user = await users_collection.find_one({"email": current_user["email"]})
-    if not user:
-        raise HTTPException(status_code=404, detail="User not found")
-
-    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")
+# --- 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"
 
     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))
+        async with httpx.AsyncClient() as client:
+            response = await client.get(fhir_api)
 
-# === 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
+        if response.status_code != 200:
+            raise HTTPException(status_code=502, detail="Failed to fetch from FHIR server.")
 
-# === 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")
-
-
-        
-@router.post("/ehr/auto-import")
-async def auto_import_synthea_patients():
-    csv_path = Path("data/synthea/patients.csv")
+        data = response.json()
+        entries = data.get("entry", [])
+        if not entries:
+            return {"message": "No patients found."}
 
-    if not csv_path.exists():
-        raise HTTPException(status_code=404, detail="Synthea patients.csv not found.")
-
-    try:
-        with open(csv_path, "r", encoding="utf-8") as f:
-            reader = csv.DictReader(f)
-            patients = []
+        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()
-                })
+        for entry in entries:
+            resource = entry.get("resource", {})
+            name = resource.get("name", [{}])[0]
+            full_name = f"{name.get('given', [''])[0]} {name.get('family', '')}".strip()
 
-        if not patients:
-            raise HTTPException(status_code=400, detail="No patients found in CSV.")
+            patient = {
+                "fhir_id": resource.get("id"),
+                "full_name": full_name,
+                "gender": resource.get("gender"),
+                "date_of_birth": resource.get("birthDate"),
+                "created_at": datetime.utcnow()
+            }
+            patients.append(patient)
 
         result = await patients_collection.insert_many(patients)
-        return {"message": "Imported successfully", "count": len(result.inserted_ids)}
+        return {"message": "Fetched and stored successfully", "count": len(result.inserted_ids)}
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))