seperate subjects/transaction on a user basis

2021-01-18 23:35:10 +01:00
parent 49ef36fbce
commit 0447f9f6ef
7 changed files with 107 additions and 97 deletions
--- a/core/prediction.py
+++ b/core/prediction.py
@@ -150,8 +150,8 @@ def get_color_for_amount(amount: Decimal, minimum: Decimal, maximum: Decimal):
    )


-def predict_balance(start_balance=Decimal("0")):
-    prediction_list = predict_all(Subject.objects.order_by("name"))
+def predict_balance(subjects: Iterable[Subject], start_balance=Decimal("0")):
+    prediction_list = predict_all(subjects)
    today = timezone.now().date()
    future_transactions = []
    minimum, maximum = Decimal(), Decimal()