Update src/services/cv_router.py

src/services/cv_router.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, Depends, HTTPException
+from fastapi import APIRouter, Depends, HTTPException, status
 from motor.motor_asyncio import AsyncIOMotorDatabase
 from src.core.database import mongo_db
 from src.models.mongo.cv_model import CVModel
@@ -27,26 +27,19 @@ class CVResponse(BaseModel):
 
 @router.post("/cvs", response_model=CVResponse)
 async def create_cv(cv: CVCreate, db: AsyncIOMotorDatabase = Depends(lambda: mongo_db)):
-    # Préparer les données pour l'insertion
-    cv_data = cv.model_dump(exclude_unset=True)
-    
-    # Insérer dans MongoDB et récupérer l'ID
-    cv_id = await CVModel.create(db, "cv", cv_data)
-    
-    # Créer la réponse avec l'ID
-    response_data = {
-        "id": cv_id,
-        "user_id": cv.user_id,
-        "parsed_data": cv.parsed_data,
-        "raw_text": cv.raw_text,
-        "upload_date": cv.upload_date
-    }
-    
-    return CVResponse(**response_data)
+    cv_entry = CVModel(**cv.model_dump(by_alias=True))
+    cv_id = await CVModel.create(db, CVModel.collection_name, cv_entry.model_dump(exclude_unset=True))
+    cv_entry.id = cv_id
+    return cv_entry
 
 @router.get("/cvs/{cv_id}", response_model=CVResponse)
 async def get_cv_by_id(cv_id: str, db: AsyncIOMotorDatabase = Depends(lambda: mongo_db)):
-    cv = await CVModel.get(db, "cv", {"_id": cv_id})
+    # Vérifie si l'ID est un ObjectId valide
+    if not ObjectId.is_valid(cv_id):
+        raise HTTPException(status_code=400, detail="ID de CV invalide")
+    
+    # MODIFICATION : Convertit cv_id en ObjectId avant d'interroger la base de données
+    cv = await CVModel.get(db, CVModel.collection_name, {"_id": ObjectId(cv_id)})
     if cv is None:
         raise HTTPException(status_code=404, detail="CV not found")
     return cv