Nattyboi commited on
Commit
37d0d50
·
1 Parent(s): 73b6185

fixed updating questionaire

Browse files
Files changed (1) hide show
  1. utils.py +8 -7
utils.py CHANGED
@@ -170,14 +170,15 @@ def create_questionaire(db_uri: str, db_name: str, collection_name: str, documen
170
  collection = db[collection_name]
171
 
172
  # Insert the document
173
- try:
174
- result= collection.find_one_and_replace(filter={"user_id":document.get("user_id")},replacement=document)
175
- except:
176
  result = collection.insert_one(document)
177
- finally:
178
- client.close()
 
179
 
180
- return str(result.inserted_id)
181
 
182
 
183
  # Close the connection
@@ -282,7 +283,7 @@ def user_details_func(db_uri: str, document: Dict) -> Optional[Dict]:
282
  print(f"in questionaire retrieval:")
283
  try:
284
  questionaire_doc.pop("_id", None)
285
- questionaire_doc.pop("user_id", None)
286
  user_doc['career_questions'] = questionaire_doc
287
  except Exception as e:
288
  # If questionnaire fails, continue with what we have
 
170
  collection = db[collection_name]
171
 
172
  # Insert the document
173
+
174
+ result= collection.find_one_and_replace(filter={"user_id":document.get("user_id")},replacement=document)
175
+ if result:
176
  result = collection.insert_one(document)
177
+ return str(result.inserted_id)
178
+
179
+ client.close()
180
 
181
+ return str(result)
182
 
183
 
184
  # Close the connection
 
283
  print(f"in questionaire retrieval:")
284
  try:
285
  questionaire_doc.pop("_id", None)
286
+ questionaire_doc.pop("userId", None)
287
  user_doc['career_questions'] = questionaire_doc
288
  except Exception as e:
289
  # If questionnaire fails, continue with what we have