def paginate(db_model: Type['models.Model'],
serializer: Type['serializers.Serializer'],
request: Request,
query: Query = None,
query_params: dict = None):
if not query:
query = db_model.objects
if query_params is None:
query_params = {k: v[0] for k, v in dict(request.query_params).items()}
for key, val in query_params.items():
if key.endswith('__in') or key == 'order_by':
query_params[key] = [s.strip() for s in val.split(',')]
page_number = int(query_params.pop('page_number', 1))
page_size = int(query_params.pop('page_size', 10))
order_by = query_params.pop('order_by', [])
query = query.filter(**query_params)
query = query.order_by(*order_by)
count = query.count()
result_models = query[(page_number - 1) * page_size:page_number *
page_size]
items = [
serializer(obj, context={
'request': request
}).data for obj in result_models
]
return Response(
data={
'total_size': count,
'page_number': page_number,
'page_size': len(items),
'items': items
})
