This commit is contained in:
2025-11-10 11:01:45 +03:30
parent 66f8818679
commit be29ab8cfa
2 changed files with 4 additions and 3 deletions

View File

@@ -257,6 +257,7 @@ class Broker(BaseModel):
broker_type = models.CharField(choices=BROKER_TYPES, max_length=20, null=True) broker_type = models.CharField(choices=BROKER_TYPES, max_length=20, null=True)
fix_broker_price = models.PositiveBigIntegerField(default=0) fix_broker_price = models.PositiveBigIntegerField(default=0)
fix_broker_price_state = models.BooleanField(default=False) fix_broker_price_state = models.BooleanField(default=False)
suggested_broker_price = models.PositiveBigIntegerField(default=0)
required = models.BooleanField(default=False) required = models.BooleanField(default=False)
def __str__(self): def __str__(self):

View File

@@ -43,16 +43,16 @@ class QuotaDistributionService:
} }
@transaction.atomic @transaction.atomic
def consume(self, weight): # noqa def consume(self, amount): # noqa
""" """
Consume 'amount' of weight across all distributions in order. Consume 'amount' of weight across all distributions in order.
Automatically splits usage if needed. Automatically splits usage if needed.
""" """
if weight > self.remaining_weight: if amount > self.remaining_weight:
raise APIException('Not enough weight to consume') raise APIException('Not enough weight to consume')
remaining_to_consume = weight remaining_to_consume = amount
for dist in self.distribution.select_for_update(): for dist in self.distribution.select_for_update():
if remaining_to_consume <= 0: if remaining_to_consume <= 0: