add - IR to all sheba in devices

apps/product/pos/api/v1/serializers/product_serializers.py

@@ -40,13 +40,13 @@ class POSFreeProductSerializer(serializers.ModelSerializer):
         }
 
         representation['total_price'] = instance.price + instance.company_fee
-        representation['main_shaba'] = instance.organization.bank_information.all().first().sheba  # noqa
+        representation['main_shaba'] = "IR" + instance.organization.bank_information.all().first().sheba  # noqa
 
         main_company = product_models.Organization.objects.get(
             national_unique_id='1111111111'
         )
         main_company_bank_account = main_company.bank_information.first()
-        representation['company_sheba'] = main_company_bank_account.sheba
+        representation['company_sheba'] = "IR" + main_company_bank_account.sheba
 
         return representation
 

apps/product/pos/api/v1/serializers/quota_distribution_serializers.py

@@ -119,7 +119,7 @@ class QuotaDistributionSerializer(serializers.ModelSerializer):
         )
 
         representation['pricing'] = {  # noqa
-            'main_account_sheba': organization.bank_information.first().sheba,
+            'main_account_sheba': "IR" + organization.bank_information.first().sheba,
             'pricing_attributes': quota_attribute_value(instance.quota),
             'sharing': sharing_list,
             'base_prices': quota_pricing_items_by_type(instance.quota, sharing=sharing_list)

apps/product/services/quota_distribution_service.py

@@ -1,4 +1,6 @@
+from django.db import transaction
 from django.db.models import Sum
+from rest_framework.exceptions import APIException
 
 from apps.product import models as product_models
 
@@ -15,14 +17,57 @@ class QuotaDistributionService:
         self.distribution = product_models.QuotaDistribution.objects.filter(
             quota=quota,
             assigned_organization=organization,
-            quota__is_closed=False,
             warehouse_entry__gt=0
         ).first()
 
         @property
-        def total_weight(self):
+        def total_weight():
             return self.distribution.aggregate(
                 total=Sum('weight')
             )['total'] or 0
 
-        pass
+        @property
+        def remaining_weight():
+            return self.distribution.aggregate(
+                total=Sum('remaining_weight')
+            )['total'] or 0
+
+        def to_representation(self):  # noqa
+            return {
+                'quota_id': self.quota.id,
+                'quota_identity': self.quota.quota_id,
+                'product': self.quota.product.name,
+                'total_weight': self.total_weight,
+                'remaining_weight': self.remaining_weight,
+                'distribution_ids': self.distribution_ids,
+            }
+
+        @transaction.atomic
+        def consume(self, weight):  # noqa
+            """
+            Consume 'amount' of weight across all distributions in order.
+            Automatically splits usage if needed.
+            """
+
+            if weight > self.remaining_weight:
+                raise APIException('Not enough weight to consume')
+
+            remaining_to_consume = weight
+
+            for dist in self.distribution.select_for_update():
+                if remaining_to_consume <= 0:
+                    break
+
+                available = dist.remaining_weight
+                if available <= 0:
+                    continue
+
+                consume_amount = min(available, remaining_to_consume)
+                dist.remaining_weight -= consume_amount
+                dist.save(update_fields=["remaining_weight"])
+                remaining_to_consume -= consume_amount
+
+            if remaining_to_consume > 0:
+                raise APIException('Not enough weight to consume')
+
+            return True