From 91dcd0962ec988dd4e5cdeb064d950a2ad0f1f42 Mon Sep 17 00:00:00 2001
From: "Edgar P. Burkhart" <git@edgarpierre.fr>
Date: Sun, 16 Apr 2023 17:33:51 +0200
Subject: [PATCH] Fix category budget toggle

---
 nummi/api/views.py  | 1 +
 nummi/main/forms.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/nummi/api/views.py b/nummi/api/views.py
index 2d46911..31024eb 100644
--- a/nummi/api/views.py
+++ b/nummi/api/views.py
@@ -44,6 +44,7 @@ class HistoryView(UserMixin, MultipleObjectMixin, View):
             {
                 "data": list(
                     self.get_queryset()
+                    .filter(category__budget=True)
                     .values(month=TruncMonth("date"))
                     .annotate(
                         sum_p=Sum(
diff --git a/nummi/main/forms.py b/nummi/main/forms.py
index 146432a..4ae8339 100644
--- a/nummi/main/forms.py
+++ b/nummi/main/forms.py
@@ -28,6 +28,7 @@ class CategoryForm(NummiForm):
         fields = [
             "name",
             "icon",
+            "budget",
         ]