37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
|
from rest_framework.pagination import PageNumberPagination
|
|||
|
|
from rest_framework.response import Response
|
|||
|
|
|
|||
|
|
class StandardResultsSetPagination(PageNumberPagination):
|
|||
|
|
"""标准分页器，支持标准响应格式"""
|
|||
|
|
page_size = 10
|
|||
|
|
page_size_query_param = 'page_size'
|
|||
|
|
max_page_size = 100
|
|||
|
|
|
|||
|
|
def get_paginated_response(self, data):
|
|||
|
|
"""
|
|||
|
|
返回标准格式的分页响应
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
data: 已经封装为标准格式的响应数据
|
|||
|
|
"""
|
|||
|
|
# 如果data已经是标准格式{code, message, data}，则用data['data']取出实际数据
|
|||
|
|
actual_data = data.get('data') if isinstance(data, dict) and 'data' in data else data
|
|||
|
|
|
|||
|
|
# 准备分页元数据
|
|||
|
|
pagination_data = {
|
|||
|
|
'count': self.page.paginator.count,
|
|||
|
|
'next': self.get_next_link(),
|
|||
|
|
'previous': self.get_previous_link(),
|
|||
|
|
'results': actual_data,
|
|||
|
|
'total_pages': self.page.paginator.num_pages,
|
|||
|
|
'current_page': self.page.number,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 如果data是标准格式，则保持原有message和code，否则使用默认值
|
|||
|
|
response_data = {
|
|||
|
|
'code': data.get('code', 200) if isinstance(data, dict) and 'code' in data else 200,
|
|||
|
|
'message': data.get('message', '获取数据成功') if isinstance(data, dict) and 'message' in data else '获取数据成功',
|
|||
|
|
'data': pagination_data
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return Response(response_data)
|