fix - warehouse allocation service

apps/product/web/api/v1/serializers/quota_serializers.py

@@ -37,6 +37,7 @@ class QuotaSerializer(serializers.ModelSerializer):
                 representation['remaining_weight'] = quota_weight_by_org['remaining_weight']
                 representation['been_sold'] = quota_weight_by_org['been_sold']
                 representation['inventory_received'] = quota_weight_by_org['inventory_received']
+                representation['inventory_entry_balance'] = quota_weight_by_org['inventory_entry_balance']
                 representation['distributions_number_by_me'] = instance.distributions_assigned.filter(
                     assigner_organization=org
                 ).count()

apps/warehouse/services/warehouse_allocation_service.py

@@ -56,10 +56,12 @@ class WarehouseAllocationService:
 
             org = entry.organization.parent_organization
             while org:
-                stat = OrganizationQuotaStats.objects.get(
+                stat = OrganizationQuotaStats.objects.filter(
                     quota=entry.quota,
                     organization=org
                 )
-                stat.inventory_received += entry.weight
-                stat.save()
-                org = org.parent_organization
+                if stat.exists():
+                    stat = stat.first()
+                    stat.inventory_received += entry.weight
+                    stat.save()
+                    org = org.parent_organization