Upload 140 files

src/apis/controllers/auth_controller.py

@@ -9,6 +9,7 @@ jwt_provider = JWTProvider()
 
 
 async def login_control(token):
+    first_login = False
     if not token:
         raise HTTPException(
             status_code=status.HTTP_401_UNAUTHORIZED,
@@ -25,6 +26,7 @@ async def login_control(token):
     existing_user = await UserCRUD.read_one({"email": user.email})
     if not existing_user:
         user_id = await UserCRUD.create(user.model_dump())
+        first_login = True
         logger.info(f"User {user.email} created.")
     else:
         user_id = existing_user["_id"]
@@ -35,4 +37,4 @@ async def login_control(token):
     user_data.pop("created_at", None)
     user_data.pop("updated_at", None)
     user_data.pop("expire_at", None)
-    return token, user_data
+    return token, user_data, first_login

src/apis/controllers/chat_controller.py

@@ -52,20 +52,20 @@ async def save_history(user_id, human_message, ai_message, intent):
             {"session_id": user_id}, {"intent": intent}
         )
         logger.info("History updated")
-    history_cache = await get_key_redis(f"chat_history_{user_id}")
-    if history_cache is not None:
-        history_cache = eval(history_cache)
-        history_cache["message"] = (
-            history_cache["message"] + messages_add_to_history_dict
-        )
-        history_cache["intent"] = intent
-        await set_key_redis(
-            f"chat_history_{user_id}",
-            str(history_cache),
-        )
-        return {"message": "History updated"}
-    await set_key_redis(f"chat_history_{user_id}", str(messages_add_to_history_cache))
-    return {"message": "History created"}
+    # history_cache = await get_key_redis(f"chat_history_{user_id}")
+    # if history_cache is not None:
+    #     history_cache = eval(history_cache)
+    #     history_cache["message"] = (
+    #         history_cache["message"] + messages_add_to_history_dict
+    #     )
+    #     history_cache["intent"] = intent
+    #     await set_key_redis(
+    #         f"chat_history_{user_id}",
+    #         str(history_cache),
+    #     )
+    #     return {"message": "History updated"}
+    # await set_key_redis(f"chat_history_{user_id}", str(messages_add_to_history_cache))
+    # return {"message": "History created"}
 
 
 async def chat_streaming_function(user, data: Chat, background_tasks: BackgroundTasks):
@@ -223,6 +223,7 @@ async def get_history_function(session_id):
                 i.model_dump(include=["type", "content"]) for i in history_messages
             ]
             intent = await get_intent_function(session_id)
+            logger.info(F"INTENT {intent}")
             return {"message": history_messages, "intent": intent}
         history_messages = eval(history_messages)
         return history_messages

src/apis/routes/auth_route.py

@@ -17,9 +17,14 @@ user_dependency = Annotated[User, Depends(get_current_user)]
 )
 async def login(credential: Credential):
     try:
-        token, user_data = await login_control(credential.credential)
+        token, user_data, first_login = await login_control(credential.credential)
         return JSONResponse(
-            content={"token": token, "user_data": user_data}, status_code=200
+            content={
+                "token": token,
+                "user_data": user_data,
+                "first_login": first_login,
+            },
+            status_code=200,
         )
     except Exception as e:
         return JSONResponse(content={"message": str(e)}, status_code=500)

src/apis/routes/post_router.py

@@ -47,7 +47,7 @@ async def create_post(body: BodyPost, user: user_dependency):
         return JSONResponse(content=result, status_code=201)
 
 
-@router.post("/get/{post_id}", status_code=status.HTTP_200_OK)
+@router.get("/get/{post_id}", status_code=status.HTTP_200_OK)
 async def get_post(post_id: str, user: user_dependency):
     if user is None:
         return JSONResponse(content={"message": "User not found"}, status_code=404)

src/apis/routes/scheduling_router.py

@@ -40,6 +40,7 @@ class BodyActivity(BaseDocument):
 
 @router.post("/create", status_code=status.HTTP_201_CREATED)
 async def create_activity(body: BodyActivity, user: user_dependency):
+    print(body)
     if user is None:
         return JSONResponse(content={"message": "User not found"}, status_code=404)
     user_id = user["id"]
@@ -53,7 +54,6 @@ async def create_activity(body: BodyActivity, user: user_dependency):
         end_time,
         user_id,
     )
-    logger.info(f"RESULT: {result}")
     if result["status"] == "error":
         return JSONResponse(content=result, status_code=404)
     else:
@@ -61,6 +61,30 @@ async def create_activity(body: BodyActivity, user: user_dependency):
 
 
 @router.get("/search", status_code=status.HTTP_200_OK)
+async def search_activities(start_time: str, end_time: str, user: user_dependency):
+    if user is None:
+        return JSONResponse(content={"message": "User not found"}, status_code=404)
+    user_id = user["id"]
+
+    if start_time and end_time:
+        start_time = convert_string_date_to_iso(start_time)
+        end_time = convert_string_date_to_iso(end_time)
+
+    result = await search_activities_controller(start_time, end_time, user_id)
+    if result["status"] == "error":
+        return JSONResponse(content={"message": result["message"]}, status_code=404)
+    else:
+        for activity in result["message"]:
+            activity["start_time"] = datetime_to_iso_string(activity["start_time"])
+            activity["end_time"] = datetime_to_iso_string(activity["end_time"])
+            activity["created_at"] = datetime_to_iso_string(activity["created_at"])
+            activity["updated_at"] = datetime_to_iso_string(activity["updated_at"])
+            activity["id"] = activity.pop("_id")
+            del activity["user_id"]
+        return JSONResponse(content=result, status_code=200)
+
+
+@router.get("/search_all", status_code=status.HTTP_200_OK)
 async def search_activities(user: user_dependency):
     if user is None:
         return JSONResponse(content={"message": "User not found"}, status_code=404)