From 1536e7f674ba3581bec24c9c8e5e55c244ff2061 Mon Sep 17 00:00:00 2001 From: "Edgar P. Burkhart" Date: Sat, 31 Dec 2022 11:38:39 +0100 Subject: [PATCH] Code cleanup Use mixins for listviews --- nummi/main/views.py | 28 ++++++++++++++++++---------- 1 file changed, 18 insertions(+), 10 deletions(-) diff --git a/nummi/main/views.py b/nummi/main/views.py index aec332b..73a3c74 100644 --- a/nummi/main/views.py +++ b/nummi/main/views.py @@ -241,15 +241,7 @@ class AccountMixin: } -class AccountTListView(AccountMixin, TransactionListView): - pass - - -class AccountSListView(AccountMixin, SnapshotListView): - pass - - -class SnapshotTListView(TransactionListView): +class SnapshotMixin: def get_queryset(self): return super().get_queryset().filter(snapshot=self.kwargs.get("pk")) @@ -259,7 +251,7 @@ class SnapshotTListView(TransactionListView): } -class CategoryTListView(TransactionListView): +class CategoryMixin: def get_queryset(self): return super().get_queryset().filter(category=self.kwargs.get("pk")) @@ -269,6 +261,22 @@ class CategoryTListView(TransactionListView): } +class AccountTListView(AccountMixin, TransactionListView): + pass + + +class AccountSListView(AccountMixin, SnapshotListView): + pass + + +class SnapshotTListView(SnapshotMixin, TransactionListView): + pass + + +class CategoryTListView(CategoryMixin, TransactionListView): + pass + + class SearchView(TransactionListView): def post(self, *args, **kwargs): return redirect("search", search=self.request.POST.get("search"))