fix: base 支持从request.data中获取page和page_size

This commit is contained in:
caoqianming 2024-03-08 18:15:58 +08:00
parent 985eb2ac22
commit 47a66cc765
1 changed files with 2 additions and 2 deletions

View File

@ -27,8 +27,8 @@ class MyPagination(PageNumberPagination):
return super().get_page_size(request) return super().get_page_size(request)
def paginate_queryset(self, queryset, request, view=None): def paginate_queryset(self, queryset, request, view=None):
if request.query_params.get('pageoff', None) or request.query_params.get('page', None) == '0': if request.query_params.get('pageoff', None) or request.query_params.get('page', None) == '0' or request.data.get('page', None) in (0, '0'):
if queryset.count() < 800: if queryset.count() < 3000:
return None return None
raise ParseError('单次请求数据量大,请分页获取') raise ParseError('单次请求数据量大,请分页获取')
return super().paginate_queryset(queryset, request, view=view) return super().paginate_queryset(queryset, request, view=view)