Update api/routes/patients.py

api/routes/patients.py

@@ -1,9 +1,9 @@
-from fastapi import APIRouter, HTTPException, Depends, Query, status
+from fastapi import APIRouter, HTTPException, Depends, Query, status, Body
 from db.mongo import patients_collection
 from core.security import get_current_user
 from utils.db import create_indexes
 from utils.helpers import calculate_age, standardize_language
-from models.entities import Note
+from models.entities import Note, PatientCreate
 from datetime import datetime
 from bson import ObjectId
 from bson.errors import InvalidId
@@ -33,6 +33,76 @@ BASE_DIR = Path(__file__).resolve().parent.parent.parent
 SYNTHEA_DATA_DIR = BASE_DIR / "output" / "fhir"
 os.makedirs(SYNTHEA_DATA_DIR, exist_ok=True)
 
+@router.post("/", status_code=status.HTTP_201_CREATED)
+async def create_patient(
+    patient_data: PatientCreate,
+    current_user: dict = Depends(get_current_user)
+):
+    """
+    Create a new patient in the database.
+    
+    Args:
+        patient_data: Patient data to be created
+        current_user: Authenticated user making the request
+        
+    Returns:
+        The created patient data with generated IDs
+    """
+    logger.info(f"Creating new patient by user {current_user.get('email')}")
+    
+    if current_user.get('role') not in ['admin', 'doctor']:
+        logger.warning(f"Unauthorized create attempt by {current_user.get('email')}")
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Only administrators and doctors can create patients"
+        )
+    
+    try:
+        # Prepare the patient document
+        patient_doc = patient_data.dict()
+        now = datetime.utcnow().isoformat()
+        
+        # Add system-generated fields
+        patient_doc.update({
+            "fhir_id": str(uuid.uuid4()),
+            "import_date": now,
+            "last_updated": now,
+            "source": "manual"
+        })
+        
+        # Ensure arrays exist even if empty
+        for field in ['conditions', 'medications', 'encounters', 'notes']:
+            if field not in patient_doc:
+                patient_doc[field] = []
+        
+        # Insert the patient document
+        result = await patients_collection.insert_one(patient_doc)
+        
+        # Return the created patient with the generated ID
+        created_patient = await patients_collection.find_one(
+            {"_id": result.inserted_id}
+        )
+        
+        if not created_patient:
+            logger.error("Failed to retrieve created patient")
+            raise HTTPException(
+                status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+                detail="Failed to retrieve created patient"
+            )
+        
+        created_patient["id"] = str(created_patient["_id"])
+        del created_patient["_id"]
+        
+        logger.info(f"Successfully created patient {created_patient['fhir_id']}")
+        return created_patient
+        
+    except Exception as e:
+        logger.error(f"Failed to create patient: {str(e)}")
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"Failed to create patient: {str(e)}"
+        )
+
 async def process_synthea_patient(bundle: dict, file_path: str) -> Optional[dict]:
     logger.debug(f"Processing patient from file: {file_path}")
     patient_data = {}
@@ -276,7 +346,7 @@ async def import_patients(
             detail=f"Import failed: {str(e)}"
         )
 
-@router.get("/patients", response_model=List[dict])
+@router.get("/", response_model=List[dict])
 async def list_patients(
     search: Optional[str] = Query(None),
     min_notes: int = Query(0, ge=0),
@@ -285,7 +355,7 @@ async def list_patients(
     skip: int = Query(0, ge=0)
 ):
     logger.info(f"Listing patients with search: {search}, limit: {limit}, skip: {skip}")
-    query = {"source": "synthea"}
+    query = {}
     
     if search:
         query["$or"] = [
@@ -310,7 +380,8 @@ async def list_patients(
         "conditions": 1,
         "medications": 1,
         "encounters": 1,
-        "notes": 1
+        "notes": 1,
+        "source": 1
     }
     
     try:
@@ -330,6 +401,7 @@ async def list_patients(
                 "medications": patient.get("medications", []),
                 "encounters": patient.get("encounters", []),
                 "notes": patient.get("notes", []),
+                "source": patient.get("source", "unknown"),
                 "age": calculate_age(patient.get("date_of_birth")),
                 "stats": {
                     "notes": len(patient.get("notes", [])),
@@ -348,7 +420,7 @@ async def list_patients(
             detail=f"Failed to retrieve patients: {str(e)}"
         )
 
-@router.get("/patients/{patient_id}", response_model=dict)
+@router.get("/{patient_id}", response_model=dict)
 async def get_patient(patient_id: str):
     logger.info(f"Retrieving patient: {patient_id}")
     try:
@@ -413,7 +485,7 @@ async def get_patient(patient_id: str):
             detail=f"Failed to retrieve patient: {str(e)}"
         )
 
-@router.post("/patients/{patient_id}/notes", status_code=status.HTTP_201_CREATED)
+@router.post("/{patient_id}/notes", status_code=status.HTTP_201_CREATED)
 async def add_note(
     patient_id: str,
     note: Note,