57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
# -*- 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),
|
||
]))
|