flo/financeplanner
1
1
Fork 0
Code Releases Activity

seperate subjects/transaction on a user basis

Browse Source
This commit is contained in:
Flo Ha
2021-01-18 23:35:10 +01:00
parent 49ef36fbce
commit 0447f9f6ef
7 changed files with 107 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
core/views.py
View File

@@ -11,6 +11,7 @@ class BalanceView(TemplateView):
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
subjects = Subject.objects.filter(user=self.request.user).order_by("name")
amount, amount_error = Decimal(), False
if amount_query := self.request.GET.get("amount"):
try:
@@ -19,7 +20,7 @@ class BalanceView(TemplateView):
amount_error = True
context["amount"] = round(amount, 2)
context["amount_error"] = amount_error
context["future_transactions"] = predict_balance(amount)
context["future_transactions"] = predict_balance(subjects, amount)
return context
@@ -28,5 +29,6 @@ class SubjectsView(TemplateView):
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context["prediction_list"] = predict_all(Subject.objects.order_by("name"))
subjects = Subject.objects.filter(user=self.request.user).order_by("name")
context["prediction_list"] = predict_all(subjects)
return context