(experimental) add running totals functionality

Author: PetrDlouhy

hordak/admin.py

@@ -27,7 +27,7 @@ class AccountAdmin(MPTTModelAdmin):
         "balance_sum",
         "income",
     )
-    readonly_fields = ("balance",)
+    readonly_fields = ("balance", "balance_sum", "income")
     raw_id_fields = ("parent",)
     search_fields = (
         "code",

hordak/management/commands/recalculate_running_totals.py

@@ -0,0 +1,19 @@
+from django.core.management.base import BaseCommand
+
+from hordak.models import Account, RunningTotal
+
+
+class Command(BaseCommand):
+    help = "Recalculate running totals for all accounts"
+
+    def handle(self, *args, **options):
+        print("Recalculating running totals for all accounts")
+        queryset = Account.objects.all()
+        for account in queryset[:1000]:
+            total = account.balance()
+            for currency in total.currencies():
+                RunningTotal.objects.update_or_create(
+                    account=account,
+                    currency=currency,
+                    defaults={"balance": total[currency]},
+                )