|
|
|
@ -110,68 +110,51 @@ class TransactionListView(NummiListView):
|
|
|
|
|
context_object_name = "transactions"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TransactionMonthView(UserMixin, MonthArchiveView):
|
|
|
|
|
class TransactionACMixin:
|
|
|
|
|
model = Transaction
|
|
|
|
|
|
|
|
|
|
def get_queryset(self):
|
|
|
|
|
if "account" in self.kwargs:
|
|
|
|
|
self.account = get_object_or_404(
|
|
|
|
|
Account.objects.filter(user=self.request.user),
|
|
|
|
|
pk=self.kwargs["account"],
|
|
|
|
|
)
|
|
|
|
|
return super().get_queryset().filter(account=self.account)
|
|
|
|
|
if "category" in self.kwargs:
|
|
|
|
|
self.category = get_object_or_404(
|
|
|
|
|
Category.objects.filter(user=self.request.user),
|
|
|
|
|
pk=self.kwargs["category"],
|
|
|
|
|
)
|
|
|
|
|
return super().get_queryset().filter(category=self.category)
|
|
|
|
|
|
|
|
|
|
return super().get_queryset()
|
|
|
|
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
|
|
|
context_data = super().get_context_data(**kwargs)
|
|
|
|
|
if "category" in self.kwargs:
|
|
|
|
|
return context_data | {"category": self.category}
|
|
|
|
|
if "account" in self.kwargs:
|
|
|
|
|
return context_data | {"account": self.account}
|
|
|
|
|
return context_data
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TransactionMonthView(UserMixin, TransactionACMixin, MonthArchiveView):
|
|
|
|
|
model = Transaction
|
|
|
|
|
date_field = "date"
|
|
|
|
|
context_object_name = "transactions"
|
|
|
|
|
month_format = "%m"
|
|
|
|
|
|
|
|
|
|
def get_queryset(self):
|
|
|
|
|
if "account" in self.kwargs:
|
|
|
|
|
self.account = get_object_or_404(
|
|
|
|
|
Account.objects.filter(user=self.request.user),
|
|
|
|
|
pk=self.kwargs["account"],
|
|
|
|
|
)
|
|
|
|
|
return super().get_queryset().filter(account=self.account)
|
|
|
|
|
if "category" in self.kwargs:
|
|
|
|
|
self.category = get_object_or_404(
|
|
|
|
|
Category.objects.filter(user=self.request.user),
|
|
|
|
|
pk=self.kwargs["category"],
|
|
|
|
|
)
|
|
|
|
|
return super().get_queryset().filter(category=self.category)
|
|
|
|
|
|
|
|
|
|
return super().get_queryset()
|
|
|
|
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
|
|
|
context_data = super().get_context_data(**kwargs)
|
|
|
|
|
if "category" in self.kwargs:
|
|
|
|
|
return context_data | {"category": self.category}
|
|
|
|
|
if "account" in self.kwargs:
|
|
|
|
|
return context_data | {"account": self.account}
|
|
|
|
|
return context_data
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TransactionYearView(UserMixin, YearArchiveView):
|
|
|
|
|
class TransactionYearView(UserMixin, TransactionACMixin, YearArchiveView):
|
|
|
|
|
model = Transaction
|
|
|
|
|
date_field = "date"
|
|
|
|
|
context_object_name = "transactions"
|
|
|
|
|
make_object_list = True
|
|
|
|
|
|
|
|
|
|
def get_queryset(self):
|
|
|
|
|
if "account" in self.kwargs:
|
|
|
|
|
self.account = get_object_or_404(
|
|
|
|
|
Account.objects.filter(user=self.request.user),
|
|
|
|
|
pk=self.kwargs["account"],
|
|
|
|
|
)
|
|
|
|
|
return super().get_queryset().filter(account=self.account)
|
|
|
|
|
if "category" in self.kwargs:
|
|
|
|
|
self.category = get_object_or_404(
|
|
|
|
|
Category.objects.filter(user=self.request.user),
|
|
|
|
|
pk=self.kwargs["category"],
|
|
|
|
|
)
|
|
|
|
|
return super().get_queryset().filter(category=self.category)
|
|
|
|
|
|
|
|
|
|
return super().get_queryset()
|
|
|
|
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
|
|
|
context_data = super().get_context_data(**kwargs)
|
|
|
|
|
context_data |= {
|
|
|
|
|
return context_data | {
|
|
|
|
|
"history": history(
|
|
|
|
|
context_data["transactions"].exclude(category__budget=False)
|
|
|
|
|
),
|
|
|
|
|
}
|
|
|
|
|
if "category" in self.kwargs:
|
|
|
|
|
return context_data | {"category": self.category}
|
|
|
|
|
if "account" in self.kwargs:
|
|
|
|
|
return context_data | {"account": self.account}
|
|
|
|
|
return context_data
|
|
|
|
|