switched my_company and my_products from method to class views

products/views.py

@@ -52,27 +52,16 @@ class ProductViewSet(viewsets.ModelViewSet):
             return Response(data=[])
 
 
-@api_view(['GET',])
-@permission_classes([IsAuthenticated,])
-def my_products(request):
-    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]
-    # prepare response payload
-    payload = {}
-    payload['results'] = data
-    payload['count'] = len(payload['results'])
-    return Response(data=payload)
+class MyProductsViewSet(viewsets.ModelViewSet):
+    model = Product
+    serializer_class = ProductSerializer
+    permission_classes = [IsAuthenticated]
+
+    def get_queryset(self):
+        return self.model.objects.filter(creator=self.request.user)
+
+    def perform_create(self, serializer):
+        serializer.save(creator=self.request.user)
 
 
 @api_view(['POST',])