diff --git a/problem/views.py b/problem/views.py index f9ce4ff..ed7bc5f 100644 --- a/problem/views.py +++ b/problem/views.py @@ -135,10 +135,10 @@ def process_data(x): problems = pg.paginate_queryset(queryset=queryset, request=request, view=self) ps = ProblemListSerializer(problems, many=True) - return Response( - {"count": queryset.count(), "res": [process_data(x) for x in ps.data]}, - status=status.HTTP_200_OK, - ) + return Response({ + "count": queryset.count(), + "res": [process_data(x) for x in ps.data] + }, status=status.HTTP_200_OK) class ProblemListCountView(APIView): def get(self, request): diff --git a/status/views.py b/status/views.py index 4fdf01b..1135f16 100644 --- a/status/views.py +++ b/status/views.py @@ -74,8 +74,10 @@ def process(x): statuses = pg.paginate_queryset(queryset=queryset, request=request, view=self) ss = StatusListSerializer(statuses, many=True) - return Response({"res": [process(x) for x in ss.data]}, status=status.HTTP_200_OK) - + return Response({ + "count": queryset.count(), + "res": [process(x) for x in ss.data] + }, status=status.HTTP_200_OK) class StatusListCountView(APIView): def get(self, request):