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:
serialized_friends: dict = {
'friends': [],
@ -5,23 +8,23 @@ def friends_serialize(friends: list, external_friends: list) -> dict:
}
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"]),
"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 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"]),
"user_id": external_friend["user_id"],
"friend_user_id": external_friend["friend_user_id"],
"status": status
})
}))
return serialized_friends

@ -1,10 +1,16 @@
def pins_serialize(pins: list) -> list:
serialized_pins: list = []
for pin in pins:
serialized_pins.append({
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:
serialized_pins: list = []
for pin in pins:
serialized_pins.append(pin_serialize(pin))
return serialized_pins
Loading…
Cancel
Save