Browse Source

feat: wip chat route

Timothy J. Baek 1 year ago
parent
commit
540b50e176
2 changed files with 13 additions and 3 deletions
  1. 4 1
      backend/apps/web/models/chats.py
  2. 9 2
      backend/apps/web/routers/chats.py

+ 4 - 1
backend/apps/web/models/chats.py

@@ -88,7 +88,10 @@ class ChatTable:
     ) -> List[ChatModel]:
         return [
             ChatModel(**model_to_dict(chat))
-            for chat in Chat.select(Chat.user_id == user_id).limit(limit).offset(skip)
+            for chat in Chat.select(Chat.id, Chat.title)
+            .where(Chat.user_id == user_id)
+            .limit(limit)
+            .offset(skip)
         ]
 
     def get_chat_by_id_and_user_id(self, id: str, user_id: str) -> Optional[ChatModel]:

+ 9 - 2
backend/apps/web/routers/chats.py

@@ -33,7 +33,7 @@ async def get_user_chats(skip: int = 0, limit: int = 50, cred=Depends(bearer_sch
     user = Users.get_user_by_token(token)
 
     if user:
-        return Chats.get_chat_titles_and_ids_by_user_id(user.id, skip, limit)
+        return Chats.get_chat_lists_by_user_id(user.id, skip, limit)
     else:
         raise HTTPException(
             status_code=status.HTTP_401_UNAUTHORIZED,
@@ -92,7 +92,14 @@ async def update_chat_by_id(
     user = Users.get_user_by_token(token)
 
     if user:
-        return Chats.update_chat_by_id_and_user_id(id, user.id, form_data.chat)
+        chat = Chats.get_chat_by_id_and_user_id(id, user.id)
+        if chat:
+            return Chats.update_chat_by_id(id, form_data.chat)
+        else:
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
+            )
     else:
         raise HTTPException(
             status_code=status.HTTP_401_UNAUTHORIZED,