Added pin serialize & fixed friend serializer if status is not null (was not displaying)

master
Alix JEUDI--LEMOINE 5 months ago
parent 32ffb3109b
commit e432fc984c

@ -1,3 +1,6 @@
from app.models.friend import Friend
def friends_serialize(friends: list, external_friends: list) -> dict: def friends_serialize(friends: list, external_friends: list) -> dict:
serialized_friends: dict = { serialized_friends: dict = {
'friends': [], 'friends': [],
@ -5,23 +8,23 @@ def friends_serialize(friends: list, external_friends: list) -> dict:
} }
for friend in friends: for friend in friends:
status = friend.status if hasattr(friend, 'status') else 'null' status = friend['status'] if 'status' in friend else 'null'
serialized_friends["friends"].append({ serialized_friends["friends"].append(Friend(**{
"id": str(friend["_id"]), "id": str(friend["_id"]),
"user_id": friend["user_id"], "user_id": friend["user_id"],
"friend_user_id": friend["friend_user_id"], "friend_user_id": friend["friend_user_id"],
"status": status "status": status
}) }))
for external_friend in external_friends: for external_friend in external_friends:
status = external_friend.status if hasattr(external_friend, 'status') else 'null' status = external_friend['status'] if 'status' in external_friend else 'null'
serialized_friends["external_friends"].append({ serialized_friends["external_friends"].append(Friend(**{
"id": str(external_friend["_id"]), "id": str(external_friend["_id"]),
"user_id": external_friend["user_id"], "user_id": external_friend["user_id"],
"friend_user_id": external_friend["friend_user_id"], "friend_user_id": external_friend["friend_user_id"],
"status": status "status": status
}) }))
return serialized_friends return serialized_friends

@ -1,10 +1,16 @@
from app.models.pin import Pin
def pin_serialize(pin: list) -> Pin:
return Pin(**{
"id": str(pin["_id"]),
"title": pin["title"],
"description": pin["description"]
})
def pins_serialize(pins: list) -> list: def pins_serialize(pins: list) -> list:
serialized_pins: list = [] serialized_pins: list = []
for pin in pins: for pin in pins:
serialized_pins.append({ serialized_pins.append(pin_serialize(pin))
"id": str(pin["_id"]),
"title": pin["title"],
"description": pin["description"]
})
return serialized_pins return serialized_pins
Loading…
Cancel
Save