add company code to organization serialzier

apps/product/web/api/v1/product_api.py

@@ -424,6 +424,10 @@ class QuotaViewSet(viewsets.ModelViewSet):  # noqa
             quota.save(calculate_final_price=True)
             return Response(data, status=status.HTTP_201_CREATED)
 
+    @transaction.atomic
+    def update(self, request, *args, **kwargs):
+        pass
+
     @action(
         methods=['get'],
         detail=False,
@@ -489,7 +493,7 @@ class QuotaViewSet(viewsets.ModelViewSet):  # noqa
             return Response(e, status=status.HTTP_204_NO_CONTENT)
 
 
-class QuotaIncentiveAssignmentViewSet(viewsets.ModelViewSet):
+class QuotaIncentiveAssignmentViewSet(viewsets.ModelViewSet):  # noqa
     """ apis for incentive assignment """
 
     queryset = product_models.QuotaIncentiveAssignment.objects.all()

apps/product/web/api/v1/quota_distribution_api.py

@@ -50,6 +50,23 @@ class QuotaDistributionViewSet(viewsets.ModelViewSet):
             return Response(serializer.data, status=status.HTTP_201_CREATED)
         return Response(serializer.errors, status=status.HTTP_403_FORBIDDEN)
 
+    @transaction.atomic
+    def update(self, request, *args, **kwargs):
+        """ Custom update of quota distribution """
+
+        partial = kwargs.pop('partial', False)
+        instance = self.get_object()
+        serializer = self.get_serializer(instance, data=request.data, partial=partial)
+        serializer.is_valid(raise_exception=True)
+        self.perform_update(serializer)
+
+        if getattr(instance, '_prefetched_objects_cache', None):
+            # If 'prefetch_related' has been applied to a queryset, we need to
+            # forcibly invalidate the prefetch cache on the instance.
+            instance._prefetched_objects_cache = {}
+
+        return Response(serializer.data)
+
     @action(
         methods=['put'],
         detail=True,