fix error 500 on fetching pictures

HE-Arc · May 3, 2024 · 288226a · 288226a
1 parent dbf72f5
commit 288226a
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)