From b50da4683860cc8fee0479397d37dc91a78e1342 Mon Sep 17 00:00:00 2001 From: Alix JEUDI--LEMOINE Date: Tue, 7 Jan 2025 15:14:34 +0100 Subject: [PATCH] :sparkles: Adding a friend now set status to pending + fixed duplication in friend serializer --- app/serializers/friend_serializer.py | 20 +++----------------- 1 file changed, 3 insertions(+), 17 deletions(-) diff --git a/app/serializers/friend_serializer.py b/app/serializers/friend_serializer.py index e280c39..0c2b582 100644 --- a/app/serializers/friend_serializer.py +++ b/app/serializers/friend_serializer.py @@ -1,7 +1,7 @@ from app.models.friend import Friend def friend_serialize(friend: list) -> Friend: - status = friend['status'] if 'status' in friend else 'null' + status = friend['status'] if 'status' in friend else 'pending' return Friend(**{ "id": str(friend["_id"]), @@ -17,23 +17,9 @@ def friends_serialize(friends: list, external_friends: list) -> dict: } for friend in friends: - status = friend['status'] if 'status' in friend else 'null' - - serialized_friends["friends"].append(Friend(**{ - "id": str(friend["_id"]), - "user_id": friend["user_id"], - "friend_user_id": friend["friend_user_id"], - "status": status - })) + serialized_friends["friends"].append(friend_serialize(friend)) for external_friend in external_friends: - status = external_friend['status'] if 'status' in external_friend else 'null' - - serialized_friends["external_friends"].append(Friend(**{ - "id": str(external_friend["_id"]), - "user_id": external_friend["user_id"], - "friend_user_id": external_friend["friend_user_id"], - "status": status - })) + serialized_friends["external_friends"].append(friend_serialize(external_friend)) return serialized_friends \ No newline at end of file