django-vue-admin/dvadmin-backend/apps/op_drf/pagination.py

from collections import OrderedDict
from typing import Any

from rest_framework.pagination import PageNumberPagination, _positive_int
from rest_framework.utils.urls import replace_query_param

from .response import SuccessResponse


class Pagination(PageNumberPagination):
    """
    标准分页器
    """
    page_size_query_param = 'pageSize'
    other_page_size_query_param = []
    # other_page_size_query_param = ['pageSize', ]

    page_query_param = "pageNum"
    other_page_query_param = []
    # other_page_query_param = ['currentPage', 'pageNum']
    max_page_size = 1000
    page_size = 10

    def paginate_queryset(self, queryset, request, view=None):
        page_num = request.query_params.get(self.page_query_param)
        # 判断，如果 pageNum 为all 则取消分页返回所有
        if page_num == 'all':
            return None
        return super().paginate_queryset(queryset, request, view)

    def get_next_link(self):
        return super().get_next_link()

    def get_previous_link(self):
        return super().get_previous_link()

    def get_page_size(self, request):
        """
        获取页大小
        :param request:
        :return:
        """
        if self.other_page_size_query_param:
            for param_name in self.other_page_size_query_param:
                if param_name in request.query_params:
                    return _positive_int(
                        request.query_params[param_name],
                        strict=True,
                        cutoff=self.max_page_size
                    )
        return super().get_page_size(request)

    def get_page_num(self, request):
        """
        获取页码
        :param request:
        :return:
        """
        if self.other_page_query_param:
            for param_name in self.other_page_query_param:
                if param_name in request.query_params:
                    return _positive_int(request.query_params[param_name], strict=True)
        page_num = request.query_params.get(self.page_query_param, 1)
        return _positive_int(page_num, strict=True)

    def get_paginated_response(self, data, search_fields=''):
        return SuccessResponse(OrderedDict([
            ('count', self.page.paginator.count),
            ('next', self.get_next_link()),
            ('previous', self.get_previous_link()),
            ('results', data)
        ]))


class JsonPagination(Pagination):
    """
    Json数据内存分页器
    """

    def __init__(self, count=0) -> None:
        super().__init__()
        self._page_size = 0
        self._page_num = 0
        self._count = count
        self.data_count = 0

    def get_page_size(self, request):
        _page_size = super().get_page_size(request)
        self._page_size = _page_size
        return _page_size

    def get_page_num(self, request):
        _page_num = super().get_page_num(request)
        self._page_num = _page_num
        return _page_num

    def get_next_link(self):
        if self._page_size * self._page_num >= self.data_count:
            return None
        url = self.request.build_absolute_uri()
        url = replace_query_param(url, self.page_query_param, self._page_num + 1)
        url = replace_query_param(url, self.page_size_query_param, self._page_size)
        return url

    def get_previous_link(self, count=0):
        if self._page_num <= 1:
            return None
        url = self.request.build_absolute_uri()
        url = replace_query_param(url, self.page_query_param, self._page_num - 1)
        url = replace_query_param(url, self.page_size_query_param, self._page_size)
        return url

    def paginate_queryset(self, queryset, request, view=None):
        self.get_page_size(request)
        self.get_page_num(request)
        return super().paginate_queryset(queryset, request, view)

    def get_paginated_response(self, data, search_fields=''):
        return SuccessResponse(OrderedDict([
            ('count', self.data_count),
            ('next', self.get_next_link()),
            ('previous', self.get_previous_link()),
            ('results', data)
        ]))