update 8-8-2022

app/src/routers/post_save.py

@@ -54,16 +54,16 @@ async def create_post(post_save: SavePost):
         data = data_output.get("user_name", None)
         if data == None:
             return JSONResponse(status_code=status.HTTP_401_UNAUTHORIZED, content={"message": "UNAUTHORIZED"})
-        count = await db["posts"].count_documents({"post_id": post_save.post_id, "username": data})
+        count = await db["posts"].count_documents({"_id": post_save.post_id})
         if count == 0:
-            return f"Post {post_save.post_id} not found"
+            return JSONResponse(status_code=status.HTTP_409_CONFLICT, content={"message": "Bài viết không tồn tại"})
         post_save.created_at = datetime.datetime.now(tz=tz)
         post_save = jsonable_encoder(post_save)
         del post_save["token"]
         post_save["username"] = data
         created_post = await db["post_save"].count_documents({"post_id": post_save["post_id"], "username": data})
         if created_post == 1:
-            return {"message": f'Post {post_save["post_id"]} saved'}
+            return JSONResponse(status_code=status.HTTP_409_CONFLICT, content={"message": "Bài viết đã được lưu trước đó"})
         new_post = await db["post_save"].insert_one(post_save)
         created_post = await db["post_save"].find_one({"_id": new_post.inserted_id})
         return JSONResponse(status_code=status.HTTP_201_CREATED, content=created_post)
@@ -87,17 +87,15 @@ async def list_post(post_save: SavePost):
     if data == None:
         return JSONResponse(status_code=status.HTTP_401_UNAUTHORIZED, content={"message": "UNAUTHORIZED"})
     posts = await db["post_save"].find({"username": data}).to_list(1000)
-    print(posts)
-    print(len(posts))
     return JSONResponse(status_code=status.HTTP_200_OK, content=posts)
 
 
 @post_save.delete("/delete_save_post", response_description="Delete save post")
-async def delete_save_post(post_save: SavePost):
+async def delete_save_post(post_id: str, token: str):
     # delete_result = await db["post_save"].delete_one({"_id": user_id})
     try:
         url = "https://sandboxapi.ebacsi.com.vn/auth/oauth/check_token"
-        payload = {'token': post_save.token}
+        payload = {'token': token}
         headers = {
             'Authorization': 'Basic RGljdGlvbmFyeU1lZGlob21lOlJ4aXR6ZnZvaWFmZmNtb2l0ZW0='
         }
@@ -107,13 +105,10 @@ async def delete_save_post(post_save: SavePost):
         data = data_output.get("user_name", None)
         if data == None:
             return JSONResponse(status_code=status.HTTP_401_UNAUTHORIZED, content={"message": "UNAUTHORIZED"})
-
-        post_save = jsonable_encoder(post_save)
-        delete_result = await db["post_save"].delete_one({"post_id": post_save["post_id"], "username": data})
+        delete_result = await db["post_save"].delete_one({"post_id": post_id, "username": data})
         if delete_result.deleted_count == 1:
             return JSONResponse(status_code=status.HTTP_200_OK, content={"message": "Delete post save successful"})
-
         raise HTTPException(
-            status_code=404, detail=f'Post save {post_save["post_id"]} not found')
+            status_code=404, detail=f'Post save {post_id} not found')
     except NameError:
         return NameError