starred/WYGIWYH
1
0
Fork 0
mirror of https://github.com/eitchtee/WYGIWYH.git synced 2026-03-17 23:13:57 +01:00
Code Issues 14 Packages Projects Releases 10 Wiki Activity

refactor: fetch currency balance from currency instead of account

Browse Source
This commit is contained in:
Herculino Trotta
2024-10-17 13:18:00 -03:00
parent f04a4e5bc5
commit 25385d6d09
1 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
app/apps/net_worth/utils/calculate_net_worth.py
View File

@@ -89,10 +89,10 @@ def calculate_account_net_worth():
def calculate_currency_net_worth():
# Subquery to calculate balance for each account
# Subquery to calculate balance for each currency
balance_subquery = (
Transaction.objects.filter(account=OuterRef("pk"), is_paid=True)
.values("account")
Transaction.objects.filter(account__currency=OuterRef("pk"), is_paid=True)
.values("account__currency")
.annotate(
balance=Sum(
Case(
@@ -106,22 +106,24 @@ def calculate_currency_net_worth():
.values("balance")
)
# Main query to fetch all account data
accounts_data = Account.objects.annotate(
# Main query to fetch all currency data
currencies_data = Currency.objects.annotate(
balance=Coalesce(Subquery(balance_subquery), Decimal("0"))
).select_related("currency")
)
print(currencies_data[0].balance)
net_worth = {}
for item in accounts_data:
currency_name = item.currency.name
for item in currencies_data:
currency_name = item.name
net_worth[currency_name] = {
"amount": net_worth.get(currency_name, {}).get("amount", Decimal("0"))
+ item.balance,
"code": item.currency.code,
"code": item.code,
"name": currency_name,
"prefix": item.currency.prefix,
"suffix": item.currency.suffix,
"decimal_places": item.currency.decimal_places,
"prefix": item.prefix,
"suffix": item.suffix,
"decimal_places": item.decimal_places,
}
return net_worth