Spaces:
Paused
Paused
| from fastapi import Body, Request, HTTPException, status | |
| from fastapi.encoders import jsonable_encoder | |
| import sys | |
| from ..models.calls import UpdateCall, UserCall, UserCaptions | |
| # Used within calls to create call record in main.py | |
| def create_calls(collection, user: UserCall = Body(...)): | |
| calls = jsonable_encoder(user) | |
| new_calls = collection.insert_one(calls) | |
| created_calls = collection.find_one({"_id": new_calls.inserted_id}) | |
| return created_calls | |
| def list_calls(collection, limit: int): | |
| try: | |
| calls = collection.find(limit = limit) | |
| return list(calls) | |
| except: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"No existing call records yet.") | |
| '''Finding calls based on call id''' | |
| def find_calls(collection, call_id: str): | |
| user_calls = collection.find_one({"call_id": call_id}) | |
| if user_calls is not None: | |
| return user_calls | |
| else: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Call with ID: '{call_id}' not found.") | |
| '''Finding calls based on user id''' | |
| def find_user_calls(collection, user_id: str): | |
| user_calls = list(collection.find({"$or": [{"caller_id": user_id}, {"callee_id": user_id}]})) # match on caller or callee ID | |
| if len(user_calls): | |
| return user_calls | |
| else: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"User with ID: '{user_id}' has no calls yet.") | |
| '''Finding calls based on key terms list''' | |
| def list_transcripts_by_key_terms(collection, key_terms_list: list[str] = Body(...)): | |
| key_terms_list = jsonable_encoder(key_terms_list) | |
| call_records = list(collection.find({"key_terms": {"$in": key_terms_list}}, {'_id': 0})) # exclude returning ObjectID in find() | |
| # Check if any call records were returned | |
| if len(call_records): | |
| return call_records | |
| else: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Call with key terms: '{key_terms_list}' not found!") | |
| '''Finding calls based on date ranges''' | |
| def list_transcripts_by_dates(collection, start_date: str, end_date: str): | |
| # print(start_date, end_date) | |
| # Convert strings to date string in YYYY-MM-ddT00:00:00 format | |
| start_date = f'{start_date}T00:00:00' | |
| end_date = f'{end_date}T00:00:00' | |
| call_records = list(collection.find({"date":{"$gte": start_date, "$lte": end_date}})) | |
| if len(call_records): | |
| return call_records | |
| else: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Call with creation date between: '{start_date} - {end_date}' not found!") | |
| '''Finding calls based on call lengths''' | |
| def list_transcripts_by_duration(collection, min_len: int, max_len: int): | |
| call_records = list(collection.find({"duration":{"$gte": min_len, "$lte": max_len}})) | |
| if len(call_records): | |
| return call_records | |
| else: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Call with duration between: '{min_len} - {max_len}' milliseconds not found!") | |
| def update_calls(collection, call_id: str, calls: UpdateCall = Body(...)): | |
| # calls = {k: v for k, v in calls.model_dump().items() if v is not None} #loop in the dict | |
| calls = {k: v for k, v in calls.items() if v is not None} #loop in the dict | |
| if len(calls) >= 1: | |
| update_result = collection.update_one({"call_id": call_id}, {"$set": calls}) | |
| if update_result.modified_count == 0: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Call not updated!") | |
| if (existing_item := collection.find_one({"call_id": call_id})) is not None: | |
| return existing_item | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Call not found!") | |
| def update_captions(collection, call_id: str, calls: UserCaptions = Body(...)): | |
| # calls = {k: v for k, v in calls.model_dump().items() if v is not None} | |
| calls = {k: v for k, v in calls.items() if v is not None} | |
| if len(calls) >= 1: | |
| update_result = collection.update_one({"call_id": call_id}, | |
| {"$push": {"captions": calls}}) | |
| if update_result.modified_count == 0: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Captions not updated!") | |
| if (existing_item := collection.find_one({"call_id": call_id})) is not None: | |
| return existing_item | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Captions not found!") | |
| def delete_calls(collection, call_id: str): | |
| deleted_calls = collection.delete_one({"call_id": call_id}) | |
| if deleted_calls.deleted_count == 1: | |
| return f"Call deleted sucessfully!" | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Call not found!") | |