分页器-DRF-自定义分页器给APIView视图类使用

1、自定义分页器

class APIViewPageNumberPaginations():
    from rest_framework.pagination import PageNumberPagination
    class APIViewBasePaginations(PageNumberPagination):
        page_size = 2  # 每页数据量
        page_query_param = 'page'  # ?page=1跳转页
        max_page_size = 10  # 每页最大数据量
        page_size_query_param = 'size'

    def __init__(self, request, serializer, queryset):
        '''
        :param request: 当前的请求
        :param serializer:  模型类的序列化器
        :param queryset:  要操作的模型对象
        '''
        self.request = request
        self.serializer = serializer
        self.queryset = queryset

    def start(self):
        # 1、实例分页对象
        paginat = self.APIViewBasePaginations()
        # 2、拿到分页后的数据
        page_list = paginat.paginate_queryset(queryset=self.queryset, request=self.request, view=self)
        # 3、对分页后的数据将进行序列化
        ser = self.serializer(instance=page_list, many=True)
        count = paginat.page.paginator.count  # 总数据量
        next_page = 1 if paginat.get_next_link() else 0  # 有下一页，返回1，没有返回0
        previous_page = 1 if paginat.get_previous_link() else 0  # 有上一页时，返回1，没有返回0
        page_size = paginat.page_size  # 每页大小
        pages = count // page_size  # 总页数
        current_page = self.request.query_params.get('page')
        try:
            current_page = int(current_page)
        except Exception as e:
            current_page = 'error'
        if count % page_size:
            pages += 1
        return {
    
    
            'data': ser.data,  # 当前页的数据
            'next': next_page,  # 是否有下一页
            'previous': previous_page,  # 是否有上一页
            'count': count,  # 总数据量
            'pages': pages,  # 总页数
            'current_page': current_page  # 当前的页数
        }

2、在视图中使用：

class UserInfoAPIView(APIView):
    #只有管理员才能使用，获取用户信息
    permission_classes = [permission.AdminPermission]
    def get(self, request):
        '''
        功能：返回用户信息（除了role=1的用户，管理员用户信息不展示）
        :return:
        '''
        #搜索的参数
        name = request.query_params.get('name')
        role = request.query_params.get('role')
        # 1、拿到数据集
        instance = models.UserModel.objects.filter(role__gt=1).order_by('id')
        if name and name!='全部':
            instance = instance.filter(name__contains=name)
        if role and role!='全部':
            instance = instance.filter(role=role)
        # 2、拿到序列化器
        ser = serializer.UserInfoModelSerializer
        # 3、实例化化分页对象，
        paginat = APIViewPageNumberPaginations(request, ser, instance)
        # 4、执行分页功能，拿到分页后的数据
        data = paginat.start()
        return Response(data=data)