⚗️ Modifying data serializers to return the right information

app/serializers/friend_serializer.py

@@ -1,8 +1,27 @@
-def friends_serialize(friends: list) -> list:
+def friends_serialize(friends: list, external_friends: list) -> dict:
-    serialized_friends: list = []
+    serialized_friends: dict = {
+        'friends': [],
+        'external_friends': []
+    }
+
     for friend in friends:
-        serialized_friends.append({
+        status = friend.status if hasattr(friend, 'status') else 'null'
+
+        serialized_friends["friends"].append({
             "id": str(friend["_id"]),
-            "user_id": friend["user_id"]
+            "user_id": friend["user_id"],
+            "friend_user_id": friend["friend_user_id"],
+            "status": status
         })
+
+    for external_friend in external_friends:
+        status = external_friend.status if external_friend.status else 'null'
+
+        serialized_friends["external_friends"].append({
+            "id": str(external_friend["_id"]),
+            "user_id": external_friend["user_id"],
+            "friend_user_id": external_friend["friend_user_id"],
+            "status": status
+        })
+
     return serialized_friends

app/serializers/user_serializer.py

@@ -1,3 +1,6 @@
+from app.models.user import User
+
+
 def users_serialize(users: list) -> list:
     serialized_users: list = []
     for user in users:
@@ -6,3 +9,11 @@ def users_serialize(users: list) -> list:
             "username": user["username"]
         })
     return serialized_users
+
+def user_serialize(user) -> User:
+    print(user)
+    return User(
+        _id=str(user['_id']),
+        username=user['username'],
+        password=user['password']
+    )