From c6cd525c4958d162bb8b37ffe48d9a2f2530cd83 Mon Sep 17 00:00:00 2001 From: Herculino Trotta Date: Sat, 9 Aug 2025 03:42:21 -0300 Subject: [PATCH] fix(insights): display untracked accounts on sankey by account --- app/apps/insights/utils/transactions.py | 9 +++++++-- app/apps/insights/views.py | 2 +- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/app/apps/insights/utils/transactions.py b/app/apps/insights/utils/transactions.py index 4fd5100..fbfbc0a 100644 --- a/app/apps/insights/utils/transactions.py +++ b/app/apps/insights/utils/transactions.py @@ -13,7 +13,9 @@ from apps.insights.forms import ( ) -def get_transactions(request, include_unpaid=True, include_silent=False): +def get_transactions( + request, include_unpaid=True, include_silent=False, include_untracked_accounts=False +): transactions = Transaction.objects.all() filter_type = request.GET.get("type", None) @@ -95,6 +97,9 @@ def get_transactions(request, include_unpaid=True, include_silent=False): Q(Q(category__mute=True) & ~Q(category=None)) | Q(mute=True) ) - transactions = transactions.exclude(account__in=request.user.untracked_accounts.all()) + if not include_untracked_accounts: + transactions = transactions.exclude( + account__in=request.user.untracked_accounts.all() + ) return transactions diff --git a/app/apps/insights/views.py b/app/apps/insights/views.py index 5ebbcfe..098c43d 100644 --- a/app/apps/insights/views.py +++ b/app/apps/insights/views.py @@ -74,7 +74,7 @@ def index(request): def sankey_by_account(request): # Get filtered transactions - transactions = get_transactions(request) + transactions = get_transactions(request, include_untracked_accounts=True) # Generate Sankey data sankey_data = generate_sankey_data_by_account(transactions)