QuentinL52 commited on
Commit
e435e58
·
verified ·
1 Parent(s): 237ed98

Update src/services/cv_router.py

Browse files
Files changed (1) hide show
  1. src/services/cv_router.py +6 -2
src/services/cv_router.py CHANGED
@@ -26,12 +26,16 @@ class CVResponse(BaseModel):
26
  json_encoders = {ObjectId: str}
27
  from_attributes = True
28
 
29
- @router.post("/cvs", response_model=CVResponse)
30
  async def create_cv(cv: CVCreate, db: AsyncIOMotorDatabase = Depends(lambda: mongo_db)):
31
  cv_entry = cv.model_dump(by_alias=True, exclude_unset=True)
32
  cv_id = await CVModel.create(db, CVModel.collection_name, cv_entry)
 
 
33
 
34
- cv_entry['id'] = cv_id
 
 
35
  return CVResponse(**cv_entry)
36
 
37
  @router.get("/cvs/{cv_id}", response_model=CVResponse)
 
26
  json_encoders = {ObjectId: str}
27
  from_attributes = True
28
 
29
+
30
  async def create_cv(cv: CVCreate, db: AsyncIOMotorDatabase = Depends(lambda: mongo_db)):
31
  cv_entry = cv.model_dump(by_alias=True, exclude_unset=True)
32
  cv_id = await CVModel.create(db, CVModel.collection_name, cv_entry)
33
+
34
+ cv_entry['id'] = str(cv_id)
35
 
36
+ if '_id' in cv_entry:
37
+ cv_entry.pop('_id')
38
+
39
  return CVResponse(**cv_entry)
40
 
41
  @router.get("/cvs/{cv_id}", response_model=CVResponse)