add field of activity organization - handle quota weight

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

@@ -2,7 +2,7 @@ from rest_framework import serializers
 from apps.product import models as product_models
 from apps.product.web.api.v1.product_serializers import QuotaSerializer
 from django.db import models
-from rest_framework.exceptions import APIException
+from apps.product.exceptions import QuotaWeightException
 from rest_framework import status
 
 
@@ -20,22 +20,18 @@ class QuotaDistributionSerializer(serializers.ModelSerializer):
         """ to validate if distribution weight
         more than quota weight raise exception """
 
-        quota = data['quota']
+        quota = product_models.Quota.objects.get(id=data['quota'])
         amount = data['weight']
         instance_id = self.instance.id if self.instance else None
 
         total = product_models.QuotaDistribution.objects.filter(
-            quota_id=quota
+            quota=quota
         ).exclude(id=instance_id).aggregate(
             total=models.Sum('weight')
         )['total'] or 0
         print(total)
         if total + amount > self.instance.weight:
-            raise APIException(
-                "مقدار وارد شده باعث می‌شود مجموع سهمیه‌ها از مقدار کل سهمیه بیشتر شود.",  # noqa
-                status.HTTP_400_BAD_REQUEST,
-
-            )
+            raise QuotaWeightException()
 
         return data