From aba08494a077a1f6edbe47bbe720b8ee4f29c2be Mon Sep 17 00:00:00 2001 From: Abhiuday Gupta <77210185+aeswibon@users.noreply.github.com> Date: Fri, 1 Mar 2024 17:13:33 +0530 Subject: [PATCH] fix: Adds pagination to Facility linked list API (#1911) * fix(user): adds pagination to linked facilities list API * fix(user): removed pagination class --- care/users/api/viewsets/users.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/care/users/api/viewsets/users.py b/care/users/api/viewsets/users.py index 9fa5651f53..af2d91e4c5 100644 --- a/care/users/api/viewsets/users.py +++ b/care/users/api/viewsets/users.py @@ -205,11 +205,12 @@ def check_facility_user_exists(self, user, facility): @action(detail=True, methods=["GET"], permission_classes=[IsAuthenticated]) def get_facilities(self, request, *args, **kwargs): user = self.get_object() - facilities = Facility.objects.filter(users=user).select_related( + queryset = Facility.objects.filter(users=user).select_related( "local_body", "district", "state", "ward" ) + facilities = self.paginate_queryset(queryset) facilities = FacilityBasicInfoSerializer(facilities, many=True) - return Response(facilities.data) + return self.get_paginated_response(facilities.data) @extend_schema(tags=["users"]) @action(detail=True, methods=["PUT"], permission_classes=[IsAuthenticated])