diff --git a/app/serializers/friend_serializer.py b/app/serializers/friend_serializer.py index 838689b..e280c39 100644 --- a/app/serializers/friend_serializer.py +++ b/app/serializers/friend_serializer.py @@ -1,5 +1,14 @@ from app.models.friend import Friend +def friend_serialize(friend: list) -> Friend: + status = friend['status'] if 'status' in friend else 'null' + + return Friend(**{ + "id": str(friend["_id"]), + "user_id": friend["user_id"], + "friend_user_id": friend["friend_user_id"], + "status": status + }) def friends_serialize(friends: list, external_friends: list) -> dict: serialized_friends: dict = {