init

backend/utils/pagination.py

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+from collections import OrderedDict
+
+from django.core import paginator
+from django.core.paginator import Paginator as DjangoPaginator
+from rest_framework.pagination import PageNumberPagination
+from rest_framework.response import Response
+from django.core.paginator import InvalidPage
+
+
+class CustomPagination(PageNumberPagination):
+    page_size = 20
+    page_size_query_param = "pageSize"
+    max_page_size = 999
+    django_paginator_class = DjangoPaginator
+
+    def paginate_queryset(self, queryset, request, view=None):
+        """
+        重写paginate_queryset让分页超过正常分页:有原来的4000错误无效页面。改写为返回2000成功，data=[]提示
+        """
+        page_size = self.get_page_size(request)
+        if not page_size:
+            return None
+        paginator = self.django_paginator_class(queryset, page_size)
+        page_number = self.get_page_number(request, paginator)
+        try:
+            self.page = paginator.page(page_number)
+        except InvalidPage as exc:
+            self.page = []
+
+        if paginator.num_pages > 1 and self.template is not None:
+            # The browsable API should display pagination controls.
+            self.display_page_controls = True
+
+        self.request = request
+        return list(self.page)
+
+    def get_paginated_response(self, data):
+        code = 0
+        msg = 'ok'
+        total = self.page.paginator.count if self.page else 0
+        res = {
+            "total": total,
+            "items": data
+        }
+        if not data:
+            code = 0
+            msg = "暂无数据"
+            res['data'] = []
+
+        return Response(OrderedDict([
+            ('code', code),
+            ('message', msg),
+            ('data', res),
+            ('error', None),
+        ]))