From be29ab8cfa7227f1e8cd2f0c7ebcc26c1db51bc4 Mon Sep 17 00:00:00 2001 From: Mojtaba-z Date: Mon, 10 Nov 2025 11:01:45 +0330 Subject: [PATCH] test --- apps/product/models.py | 1 + apps/product/services/quota_distribution_service.py | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/apps/product/models.py b/apps/product/models.py index b2d1154..c034b4d 100644 --- a/apps/product/models.py +++ b/apps/product/models.py @@ -257,6 +257,7 @@ class Broker(BaseModel): broker_type = models.CharField(choices=BROKER_TYPES, max_length=20, null=True) fix_broker_price = models.PositiveBigIntegerField(default=0) fix_broker_price_state = models.BooleanField(default=False) + suggested_broker_price = models.PositiveBigIntegerField(default=0) required = models.BooleanField(default=False) def __str__(self): diff --git a/apps/product/services/quota_distribution_service.py b/apps/product/services/quota_distribution_service.py index 5253212..7aba870 100644 --- a/apps/product/services/quota_distribution_service.py +++ b/apps/product/services/quota_distribution_service.py @@ -43,16 +43,16 @@ class QuotaDistributionService: } @transaction.atomic - def consume(self, weight): # noqa + def consume(self, amount): # noqa """ Consume 'amount' of weight across all distributions in order. Automatically splits usage if needed. """ - if weight > self.remaining_weight: + if amount > self.remaining_weight: raise APIException('Not enough weight to consume') - remaining_to_consume = weight + remaining_to_consume = amount for dist in self.distribution.select_for_update(): if remaining_to_consume <= 0: