feat(networth): add projected view

2026-07-15 09:12:39 +02:00 · 2024-12-15 01:22:33 -03:00
parent 77cc043b90
commit 188e737121
4 changed files with 122 additions and 10 deletions
@@ -14,7 +14,7 @@ from apps.transactions.utils.calculations import (
 )


-def net_worth_main(request):
+def net_worth_current(request):
    transactions_currency_queryset = Transaction.objects.filter(
        is_paid=True, account__is_archived=False
    ).order_by(
@@ -110,3 +110,103 @@ def net_worth_main(request):
            "accounts": accounts,
        },
    )
+
+
+def net_worth_projected(request):
+    transactions_currency_queryset = Transaction.objects.filter(
+        account__is_archived=False
+    ).order_by(
+        "account__currency__name",
+    )
+    transactions_account_queryset = Transaction.objects.filter(
+        account__is_archived=False
+    ).order_by(
+        "account__group__name",
+        "account__name",
+    )
+
+    currency_net_worth = calculate_currency_totals(
+        transactions_queryset=transactions_currency_queryset
+    )
+    account_net_worth = calculate_account_totals(
+        transactions_queryset=transactions_account_queryset
+    )
+
+    historical_currency_net_worth = calculate_historical_currency_net_worth(
+        is_paid=False
+    )
+
+    labels = (
+        list(historical_currency_net_worth.keys())
+        if historical_currency_net_worth
+        else []
+    )
+    currencies = (
+        list(historical_currency_net_worth[labels[0]].keys())
+        if historical_currency_net_worth
+        else []
+    )
+
+    datasets = []
+    for i, currency in enumerate(currencies):
+        data = [
+            float(month_data[currency])
+            for month_data in historical_currency_net_worth.values()
+        ]
+        datasets.append(
+            {
+                "label": currency,
+                "data": data,
+                "yAxisID": f"y{i}",
+                "fill": False,
+                "tension": 0.1,
+            }
+        )
+
+    chart_data_currency = {"labels": labels, "datasets": datasets}
+
+    chart_data_currency_json = json.dumps(chart_data_currency, cls=DjangoJSONEncoder)
+
+    historical_account_balance = calculate_historical_account_balance(is_paid=False)
+
+    labels = (
+        list(historical_account_balance.keys()) if historical_account_balance else []
+    )
+    accounts = (
+        list(historical_account_balance[labels[0]].keys())
+        if historical_account_balance
+        else []
+    )
+
+    datasets = []
+    for i, account in enumerate(accounts):
+        data = [
+            float(month_data[account])
+            for month_data in historical_account_balance.values()
+        ]
+        datasets.append(
+            {
+                "label": account,
+                "data": data,
+                "fill": False,
+                "tension": 0.1,
+                "yAxisID": f"y-axis-{i}",  # Assign each dataset to its own Y-axis
+            }
+        )
+
+    chart_data_accounts = {"labels": labels, "datasets": datasets}
+
+    chart_data_accounts_json = json.dumps(chart_data_accounts, cls=DjangoJSONEncoder)
+
+    return render(
+        request,
+        "net_worth/net_worth.html",
+        {
+            "currency_net_worth": currency_net_worth,
+            "account_net_worth": account_net_worth,
+            "chart_data_currency_json": chart_data_currency_json,
+            "currencies": currencies,
+            "chart_data_accounts_json": chart_data_accounts_json,
+            "accounts": accounts,
+        },
+    )