implemented pagination for my_products and my_company

This commit is contained in:
Sam
2021-03-04 11:50:26 +00:00
parent 8068921bb7
commit c99b81b751
4 changed files with 106 additions and 8 deletions

View File

@@ -54,9 +54,20 @@ class ProductViewSet(viewsets.ModelViewSet):
@api_view(['GET',])
@permission_classes([IsAuthenticated,])
def my_products(request):
qs = Product.objects.filter(creator=request.user)
product_serializer = ProductSerializer(qs, many=True)
return Response(data=product_serializer.data)
limit = request.GET.get('limit')
offset = request.GET.get('offset')
qs = Product.objects.filter(creator=request.user)
product_serializer = ProductSerializer(qs, many=True)
data = product_serializer.data
# RESULTS PAGINATION
if limit is not None and offset is not None:
limit = int(limit)
offset = int(offset)
data = data[offset:(limit+offset)]
elif limit is not None:
limit = int(limit)
data = data[:limit]
return Response(data=data)
@api_view(['POST',])