diff --git a/app/main.py b/app/main.py index 7d71f5e..78aa79d 100644 --- a/app/main.py +++ b/app/main.py @@ -199,7 +199,7 @@ async def add_pin(pin: PinDTO, current_user: User = Depends(get_current_user)): responses={401: {"model": HTTPError}} ) async def list_pins(current_user: User = Depends(get_current_user)): - pins = serializers.pins_serialize(pins_collection.find().to_list()) + pins = serializers.pins_serialize(pins_collection.find().to_list(), current_user.uid) return pins @app.get( diff --git a/app/serializers/pin_serializer.py b/app/serializers/pin_serializer.py index bd6473d..f07077f 100644 --- a/app/serializers/pin_serializer.py +++ b/app/serializers/pin_serializer.py @@ -11,9 +11,10 @@ def pin_serialize(pin: list) -> Pin: "user_id": pin["user_id"] }) -def pins_serialize(pins: list) -> list: +def pins_serialize(pins: list, uid: str) -> list: serialized_pins: list = [] for pin in pins: - serialized_pins.append(pin_serialize(pin)) + if pin["user_id"] == uid: + serialized_pins.append(pin_serialize(pin)) return serialized_pins \ No newline at end of file