Merge pull request #30 from segment-oj/fix-status-list-ztl

fix status list count bug

Author: szdytom

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):

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):