zcdsj
/
factory
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/utils/pagination.py
View File

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