Merge branch 'main' of github.com:HE-Arc/kodecupid

HE-Arc · May 3, 2024 · fa6a019 · fa6a019
2 parents 65f6091 + 4c954eb
commit fa6a019
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/backend/kodecupidapp/views/user.py b/backend/kodecupidapp/views/user.py
@@ -115,7 +115,9 @@ def add_picture(self, request):
     @action(detail=True, methods=['get'])
     def pictures(self, request, pk=None):
         user = self.get_object()
-        queryset = Picture.objects.filter(user=user.id).exclude(id=user.pfp.id)
-        serializer = PictureSerializer(queryset, many=True)
-
+        if user.pfp:
+            queryset = Picture.objects.filter(user=user.id).exclude(id=user.pfp.id)
+            serializer = PictureSerializer(queryset, many=True)
+        else:
+            return Response({"message": "User does not have a profile picture."}, status=status.HTTP_404_NOT_FOUND)
         return Response(serializer.data)