diff --git a/core/apps/counterparty/models/conterparty.py b/core/apps/counterparty/models/conterparty.py index bab6bbe..0e90e4e 100644 --- a/core/apps/counterparty/models/conterparty.py +++ b/core/apps/counterparty/models/conterparty.py @@ -134,25 +134,9 @@ class CounterpartyBalance(BaseModel): @property def total_balance(self): - PartyAmount = apps.get_model('orders', 'PartyAmount') - Income = apps.get_model('finance', 'Income') - Expense = apps.get_model('finance', 'Expence') + usd_course = Decimal(UsdCourse.objects.first().value) if UsdCourse.objects.exists() else Decimal(12000) + return self.total_balance_usd + (self.total_balance_usd * usd_course) - total = Decimal(0) - - for item in PartyAmount.objects.filter(party__orders__counterparty=self.counterparty): - rate = item.exchange_rate or Decimal(1) - total += Decimal(item.calculated_amount) * rate - - for inc in Income.objects.filter(counterparty=self.counterparty): - rate = inc.exchange_rate or Decimal(1) - total += Decimal(inc.price) * rate - - for exp in Expense.objects.filter(counterparty=self.counterparty): - rate = exp.exchange_rate or Decimal(1) - total -= Decimal(exp.price) * rate - - return total def __str__(self): return f"{self.counterparty.name}"