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

@@ -158,6 +158,17 @@ class CompanyViewSet(viewsets.ModelViewSet):
@api_view(['GET',])
@permission_classes([IsAuthenticated,])
def my_company(request):
qs = Company.objects.filter(creator=request.user)
company_serializer = CompanySerializer(qs, many=True)
return Response(data=company_serializer.data)
limit = request.GET.get('limit')
offset = request.GET.get('offset')
qs = Company.objects.filter(creator=request.user)
company_serializer = CompanySerializer(qs, many=True)
data = company_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)