Merge pull request #756 from OpenUpSA/hack-795-adding-subtotal-rows

Hack 795 adding subtotal rows
OpenUpSA · Sep 27, 2024 · 566454b · 566454b
2 parents 06adefa + 524d2ff
commit 566454b
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/municipal_finance/management/commands/make_sub_totals.py b/municipal_finance/management/commands/make_sub_totals.py
@@ -25,6 +25,19 @@ def handle(self, *args, **options):
                 "0170",
                 "0180",
             ],
+            "0310": [
+                "0210",
+                "0220",
+                "0230",
+                "0240",
+                "0250",
+                "0260",
+                "0270",
+                "0280",
+                "0290",
+                "0300",
+            ],
+            "0320": ["0190", "0310"],
             "0430": ["0350", "0360", "0370", "0380", "0390", "0400", "0410", "0420"],
             "0490": ["0450", "0460", "0470", "0480"],
             "0500": ["0430", "0490"],
@@ -72,5 +85,7 @@ def handle(self, *args, **options):
                                 demarcation_code=muni_code,
                                 amount_type=amount_type_model,
                                 period_length="year",
-                                amount=sub_total,
+                                defaults={
+                                    'amount': sub_total,
+                                }
                             )