refactor: Matmatronch

matmat/views.py

@@ -20,150 +20,44 @@
 # Place - Suite 330, Boston, MA 02111-1307, USA.
 #
 #
-from ast import literal_eval
-from enum import Enum
 
-from django import forms
 from django.db.models import F
-from django.http.response import HttpResponseRedirect
-from django.urls import reverse
-from django.utils.translation import gettext_lazy as _
-from django.views.generic import ListView, View
-from django.views.generic.detail import SingleObjectMixin
-from django.views.generic.edit import FormMixin, FormView
+from django.views.generic import ListView
+from django.views.generic.edit import FormMixin
 
 from core.auth.mixins import FormerSubscriberMixin
-from core.models import User
+from core.models import User, UserQuerySet
 from core.schemas import UserFilterSchema
 from matmat.forms import SearchForm
 
 
-class SearchType(Enum):
-    NORMAL = 1
-    REVERSE = 2
-    QUICK = 3
-
-
-# Views
-
-
-class SearchFormListView(FormerSubscriberMixin, SingleObjectMixin, ListView):
+class MatmatronchView(FormerSubscriberMixin, FormMixin, ListView):
     model = User
-    ordering = ["-id"]
-    paginate_by = 12
+    paginate_by = 20
     template_name = "matmat/search_form.jinja"
-
-    def dispatch(self, request, *args, **kwargs):
-        self.form_class = kwargs["form"]
-        self.search_type = kwargs["search_type"]
-        self.session = request.session
-        self.last_search = self.session.get("matmat_search_result", str([]))
-        self.last_search = literal_eval(self.last_search)
-        self.valid_form = kwargs.get("valid_form")
-
-        self.init_query = self.model.objects
-        self.can_see_hidden = True
-        if not (request.user.is_board_member or request.user.is_root):
-            self.can_see_hidden = False
-            self.init_query = self.init_query.filter(is_viewable=True)
-
-        return super().dispatch(request, *args, **kwargs)
-
-    def post(self, request, *args, **kwargs):
-        return self.get(request, *args, **kwargs)
-
-    def get_context_data(self, **kwargs):
-        self.object = None
-        kwargs = super().get_context_data(**kwargs)
-        kwargs["form"] = self.form_class
-        kwargs["result_exists"] = self.result_exists
-        return kwargs
-
-    def get_queryset(self):
-        q = self.init_query
-        if self.valid_form is not None:
-            if self.search_type == SearchType.REVERSE:
-                q = q.filter(phone=self.valid_form["phone"]).all()
-            elif self.search_type == SearchType.QUICK:
-                if self.valid_form["quick"].strip():
-                    q = list(
-                        UserFilterSchema(search=self.valid_form["quick"])
-                        .filter(User.objects.viewable_by(self.request.user))
-                        .order_by(F("last_login").desc(nulls_last=True))
-                    )
-                else:
-                    q = []
-            else:
-                search_dict = {}
-                for key, value in self.valid_form.items():
-                    if key not in ("phone", "quick") and not (
-                        value == "" or value is None
-                    ):
-                        search_dict[key + "__icontains"] = value
-                q = q.filter(**search_dict).all()
-        else:
-            q = q.filter(pk__in=self.last_search).all()
-        if isinstance(q, list):
-            self.result_exists = len(q) > 0
-        else:
-            self.result_exists = q.exists()
-        self.last_search = []
-        for user in q:
-            self.last_search.append(user.id)
-        self.session["matmat_search_result"] = str(self.last_search)
-        return q
-
-
-class SearchFormView(FormerSubscriberMixin, FormView):
-    """Allows users to search inside the user list."""
-
     form_class = SearchForm
 
-    def dispatch(self, request, *args, **kwargs):
-        self.session = request.session
-        self.init_query = User.objects
-        kwargs["form"] = self.get_form()
-        kwargs["search_type"] = self.search_type
-        return super().dispatch(request, *args, **kwargs)
-
     def get(self, request, *args, **kwargs):
-        view = SearchFormListView.as_view()
-        return view(request, *args, **kwargs)
-
-    def post(self, request, *args, **kwargs):
-        form = self.get_form()
-        view = SearchFormListView.as_view()
-        if form.is_valid():
-            kwargs["valid_form"] = form.clean()
-            request.session["matmat_search_form"] = form.cleaned_data_json
-        return view(request, *args, **kwargs)
+        self.form = self.get_form()
+        return super().get(request, *args, **kwargs)
 
     def get_initial(self):
-        init = self.session.get("matmat_search_form", {})
-        if not init:
-            init["department"] = ""
-        return init
+        return self.request.GET
 
+    def get_form_kwargs(self):
+        res = super().get_form_kwargs()
+        if self.request.GET:
+            res["data"] = self.request.GET
+        return res
 
-class SearchNormalFormView(SearchFormView):
-    search_type = SearchType.NORMAL
+    def get_queryset(self) -> UserQuerySet:
+        if not self.form.is_valid():
+            return User.objects.none()
+        data = self.form.cleaned_data
+        data["search"] = data.get("name")
+        filters = UserFilterSchema(**{key: val for key, val in data.items() if val})
+        qs = User.objects.viewable_by(self.request.user).select_related("profile_pict")
+        return filters.filter(qs).order_by(F("last_login").desc(nulls_last=True))
 
-
-class SearchReverseFormView(SearchFormView):
-    search_type = SearchType.REVERSE
-
-
-class SearchQuickFormView(SearchFormView):
-    search_type = SearchType.QUICK
-
-
-class SearchClearFormView(FormerSubscriberMixin, View):
-    """Clear SearchFormView and redirect to it."""
-
-    def dispatch(self, request, *args, **kwargs):
-        super().dispatch(request, *args, **kwargs)
-        if "matmat_search_form" in request.session:
-            request.session.pop("matmat_search_form")
-        if "matmat_search_result" in request.session:
-            request.session.pop("matmat_search_result")
-        return HttpResponseRedirect(reverse("matmat:search"))
+    def get_context_data(self, **kwargs):
+        return super().get_context_data(form=self.form, **kwargs)