Upload 164 files

src/apis/controllers/post_controller.py

@@ -62,13 +62,18 @@ async def get_a_post_controller(post_id: str) -> Dict:
         return {"status": "error", "message": str(e)}
 
 
+import time
+
+
 async def list_all_posts_controller_priority(
     user_id: str,
     page: int = 1,
     top_5_destinations: List[str] = [],
     view_post_ids: List[str] = [],
 ):
+    print("view_post_ids", view_post_ids)
     try:
+        start_time = time.time()
         PAGE_SIZE = 10
         TOP_LIMIT = int(PAGE_SIZE * 0.7)
         OTHER_LIMIT = PAGE_SIZE - TOP_LIMIT
@@ -87,13 +92,11 @@ async def list_all_posts_controller_priority(
             "_id": {"$nin": excluded_ids},
         }
 
-        # Đếm tổng số bài viết mỗi nhóm để tính total_items/pages
         total_top = await PostCRUD.count(top_filter)
         total_other = await PostCRUD.count(other_filter)
         total_items = total_top + total_other
         total_pages = math.ceil(total_items / PAGE_SIZE)
 
-        # Lấy bài viết đã tính điểm
         top_posts = await PostCRUD.find_many_with_score(
             filter=top_filter,
             top_destinations=top_5_destinations,
@@ -108,10 +111,12 @@ async def list_all_posts_controller_priority(
             skip=skip_other,
         )
 
-        # Xen kẽ
+            # Xen kẽ
         combined_posts = []
         i = j = 0
-        while len(combined_posts) < PAGE_SIZE and (i < len(top_posts) or j < len(other_posts)):
+        while len(combined_posts) < PAGE_SIZE and (
+            i < len(top_posts) or j < len(other_posts)
+        ):
             if i < len(top_posts):
                 combined_posts.append(top_posts[i])
                 i += 1
@@ -119,26 +124,25 @@ async def list_all_posts_controller_priority(
                 combined_posts.append(other_posts[j])
                 j += 1
 
-        # Enrich thông tin
         user_ids = list({post.get("user_id") for post in combined_posts})
         destination_ids = list({post.get("destination_id") for post in combined_posts})
-        post_ids = [post["_id"] for post in combined_posts]
-
-        user_infos, destination_infos, reactions = await gather(
-            *[
-                gather(*[UserCRUD.find_by_id(uid) for uid in user_ids]),
-                gather(*[DestinationCRUD.find_by_id(did) for did in destination_ids]),
-                gather(
-                    *[
-                        ReactionCRUD.read_one(
-                            {"user_id": user_id, "post_id": serialize_datetime(pid)}
-                        )
-                        for pid in post_ids
-                    ]
-                ),
-            ]
+        post_ids = [serialize_datetime(post["_id"]) for post in combined_posts]
+
+        user_infos_result = await UserCRUD.read({"_id": {"$in": user_ids}})
+        destination_infos_result = await DestinationCRUD.read(
+            {"_id": {"$in": destination_ids}}
         )
 
+        reactions_result = (
+            await ReactionCRUD.read({"user_id": user_id, "post_id": {"$in": post_ids}})
+            if user_id
+            else []
+        )
+
+        user_infos = user_infos_result
+        destination_infos = destination_infos_result
+        reactions = reactions_result
+
         user_info_map = {
             info["_id"]: {
                 "user_id": info["_id"],
@@ -188,7 +192,8 @@ async def list_all_posts_controller_priority(
                     "user_info": user_info_map.get(uid),
                 }
             )
-
+        end_time = time.time()
+        print(f"Time taken: {end_time - start_time} seconds")
         return {
             "status": "success",
             "message": {
@@ -204,8 +209,10 @@ async def list_all_posts_controller_priority(
         logger.error(f"Error listing personalized posts: {str(e)}")
         return {"status": "error", "message": str(e)}
 
+
 async def list_all_posts_controller(user_id: str, page: int = 1):
     try:
+        start_time = time.time()
         PAGE_SIZE = 5
         # Calculate skip value for pagination
         skip = (page - 1) * PAGE_SIZE
@@ -222,8 +229,15 @@ async def list_all_posts_controller(user_id: str, page: int = 1):
             sort=[("created_at", -1)],  # Sort by created_at descending
         )
 
+        # Get unique user and destination IDs
         user_ids = list({post.get("user_id") for post in posts})
-        user_infos = await gather(*[UserCRUD.find_by_id(uid) for uid in user_ids])
+        destination_ids = list({post.get("destination_id") for post in posts})
+        
+        # Use $in operator for batch lookups
+        user_infos = await UserCRUD.read({"_id": {"$in": user_ids}})
+        destination_infos = await DestinationCRUD.read({"_id": {"$in": destination_ids}})
+        
+        # Create user info map
         user_info_map = {
             info.get("_id"): {
                 "user_id": info.get("_id"),
@@ -233,22 +247,20 @@ async def list_all_posts_controller(user_id: str, page: int = 1):
             for info in user_infos
             if info
         }
-        destination_ids = list({post.get("destination_id") for post in posts})
-        destination_infos = await gather(
-            *[DestinationCRUD.find_by_id(did) for did in destination_ids]
-        )
+        
+        # Create destination info map
         destination_info_map = {
             info.get("_id"): info.get("name") for info in destination_infos if info
         }
+        
         formatted_user_reactions_map = {}
         if user_id:
             all_post_ids = [serialize_datetime(post.get("_id")) for post in posts]
-            reactions = await gather(
-                *[
-                    ReactionCRUD.read_one({"user_id": user_id, "post_id": post_id})
-                    for post_id in all_post_ids
-                ]
-            )
+            # Use $in operator for batch lookup of reactions
+            reactions = await ReactionCRUD.read({
+                "user_id": user_id,
+                "post_id": {"$in": all_post_ids}
+            })
             for reaction in reactions:
                 if reaction:
                     post_id = reaction.get("post_id")
@@ -298,6 +310,7 @@ async def list_posts_by_destination_controller(
     destination_id: str, user_id: str, page: int = 1
 ):
     try:
+        start_time = time.time()
         PAGE_SIZE = 5  # Changed from 1 to 5 posts per page
         # Calculate skip value for pagination
         skip = (page - 1) * PAGE_SIZE
@@ -328,7 +341,9 @@ async def list_posts_by_destination_controller(
 
         # Get unique user_ids
         user_ids = list({post.get("user_id") for post in posts})
-        user_infos = await gather(*[UserCRUD.find_by_id(uid) for uid in user_ids])
+        
+        # Use $in operator for batch user lookup
+        user_infos = await UserCRUD.read({"_id": {"$in": user_ids}})
         user_info_map = {
             info.get("_id"): {
                 "user_id": info.get("_id"),
@@ -347,12 +362,11 @@ async def list_posts_by_destination_controller(
         formatted_user_reactions_map = {}
         if user_id:
             all_post_ids = [serialize_datetime(post.get("_id")) for post in posts]
-            reactions = await gather(
-                *[
-                    ReactionCRUD.read_one({"user_id": user_id, "post_id": post_id})
-                    for post_id in all_post_ids
-                ]
+            # Use $in operator for batch lookup of reactions
+            reactions = await ReactionCRUD.read(
+                {"user_id": user_id, "post_id": {"$in": all_post_ids}}
             )
+
             for reaction in reactions:
                 if reaction:
                     post_id = reaction.get("post_id")

src/apis/models/destination_models.py

@@ -7,7 +7,12 @@ from bson import ObjectId
 class Destination(BaseDocument):
     name: str = Field("", description="Destination's name")
     description: str = Field("", description="Destination's description")
+    type: str = Field("", description="Destination's type")
+    tags: str = Field("", description="Destination's tags")
     image: str = Field("", description="Destination's picture")
+    lat: float = Field(0, description="Destination's latitude")
+    long: float = Field(0, description="Destination's longitude")
+    exclude_weather: list[str] = Field([], description="Destination's exclude weather")
     created_user_id: str = Field("", description="Destination's created user")
     updated_user_id: str = Field("", description="Destination's updated user")
     model_config = {
@@ -15,7 +20,12 @@ class Destination(BaseDocument):
             "example": {
                 "name": "Hanoi",
                 "description": "Hanoi is the capital of Vietnam",
+                "type": "city",
+                "tags": "Vietnam, Hanoi, Capital",
                 "image": "https://www.google.com.vn",
+                "lat": 0,
+                "long": 0,
+                "exclude_weather": ["rainy", "snowy"],
                 "created_user_id": "1234567890",
                 "updated_user_id": "1234567890",
             }