Upload 157 files

src/apis/controllers/post_controller.py

@@ -1,9 +1,11 @@
 from typing import Dict
 from src.utils.mongo import PostCRUD
 from src.utils.logger import logger
-from src.utils.mongo import UserCRUD, ReactionCRUD
+from src.utils.mongo import UserCRUD, ReactionCRUD, DestinationCRUD
 from asyncio import gather
 from src.utils.helper import call_external_api, serialize_datetime
+from datetime import datetime
+from bson import ObjectId
 
 
 async def create_a_post_controller(
@@ -57,64 +59,63 @@ async def get_a_post_controller(post_id: str) -> Dict:
 async def list_all_posts_controller(user_id: str):
     try:
         posts = await PostCRUD.find_all()
-        user_ids = [post.get("user_id") for post in posts]
-        user_infos = await gather(
-            *[UserCRUD.find_by_id(user_id) for user_id in 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])
         user_info_map = {
-            user_info.get("_id"): user_info for user_info in user_infos if user_info
+            info.get("_id"): {
+                "name": info.get("name"),
+                "picture": info.get("picture"),
+            }
+            for info in user_infos
+            if info
         }
-        all_post_ids = {serialize_datetime(post.get("_id")) for post in posts}
-
-        formatted_user_reactions = []
+        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]
+        )
+        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
                 ]
             )
-            formatted_user_reactions = [
-                {
-                    "id": serialize_datetime(reaction.get("_id", None)),
-                    "post_id": reaction.get("post_id", None),
-                    "user_id": reaction.get("user_id", None),
-                    "reaction_type": reaction.get("type", None),
-                }
-                for reaction in reactions
-                if reaction is not None  # Ensure only non-None reactions are processed
-            ]
-
+            for reaction in reactions:
+                if reaction:
+                    post_id = reaction.get("post_id")
+                    formatted_user_reactions_map[post_id] = {
+                        "id": serialize_datetime(reaction.get("_id")),
+                        "post_id": post_id,
+                        "user_id": reaction.get("user_id"),
+                        "reaction_type": reaction.get("type"),
+                    }
         serialized_posts = []
-        for index, post in enumerate(posts):
-            user_id = post.get("user_id")
-            user_info = user_info_map.get(user_id)
+        for post in posts:
+            post_id = serialize_datetime(post.get("_id"))
+            uid = post.get("user_id")
+            dest_id = post.get("destination_id")
             serialized_post = {
-                "id": serialize_datetime(post.get("_id")),
+                "id": post_id,
                 "content": post.get("content"),
-                "destination_id": post.get("destination_id"),
+                "destination_id": dest_id,
+                "destination_name": destination_info_map.get(dest_id),
                 "comment_count": post.get("comment_count", []),
                 "reaction_count": post.get("reaction_count", []),
-                "current_user_reaction": (
-                    formatted_user_reactions[index]
-                    if formatted_user_reactions
-                    else None
-                ),
+                "current_user_reaction": formatted_user_reactions_map.get(post_id),
                 "picture": post.get("picture", []),
                 "created_at": serialize_datetime(post.get("created_at")),
                 "updated_at": serialize_datetime(post.get("updated_at")),
-                "user_info": (
-                    {
-                        "name": user_info.get("name"),
-                        "picture": user_info.get("picture"),
-                    }
-                    if user_info
-                    else None
-                ),
+                "user_info": user_info_map.get(uid),
             }
             serialized_posts.append(serialized_post)
+
         return {"status": "success", "message": serialized_posts}
+
     except Exception as e:
         logger.error(f"Error listing posts: {str(e)}")
         return {"status": "error", "message": str(e)}