Basic search structure

matmat/views.py

@@ -0,0 +1,146 @@
+# -*- coding:utf-8 -*
+#
+# Copyright 2017
+# - Sli <skia@libskia.so>
+#
+# Ce fichier fait partie du site de l'Association des Étudiants de l'UTBM,
+# http://ae.utbm.fr.
+#
+# This program is free software; you can redistribute it and/or modify it under
+# the terms of the GNU General Public License a published by the Free Software
+# Foundation; either version 3 of the License, or (at your option) any later
+# version.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+# details.
+#
+# You should have received a copy of the GNU General Public License along with
+# this program; if not, write to the Free Sofware Foundation, Inc., 59 Temple
+# Place - Suite 330, Boston, MA 02111-1307, USA.
+#
+#
+
+from django.shortcuts import render
+
+# Create your views here.
+
+from django.shortcuts import get_object_or_404
+from django.views.generic import ListView
+from django.views.generic.edit import FormView
+from django.core.urlresolvers import reverse_lazy, reverse
+from django.utils.translation import ugettext_lazy as _
+from django.views.generic.detail import SingleObjectMixin
+from django.core.exceptions import PermissionDenied
+from django.db import transaction
+from django.forms import CheckboxSelectMultiple
+from django import forms
+
+from core.models import User
+from core.views import WasSuscribed
+from core.views.forms import SelectDate
+from phonenumber_field.widgets import PhoneNumberInternationalFallbackWidget
+
+# Custom form
+
+
+class SearchForm(forms.ModelForm):
+    class Meta:
+        model = User
+        fields = [
+            'first_name',
+            'last_name',
+            'nick_name',
+            'sex',
+            'role',
+            'department',
+            'semester',
+            'promo',
+            'date_of_birth',
+            'phone',
+        ]
+        widgets = {
+            'date_of_birth': SelectDate,
+            'phone': PhoneNumberInternationalFallbackWidget,
+            # 'sex': CheckboxSelectMultiple,
+        }
+
+    def __init__(self, *args, **kwargs):
+        super(SearchForm, self).__init__(*args, **kwargs)
+        for key in self.fields.keys():
+            self.fields[key].required = False
+        self.fields['sex'].choices.append(("INDIFFERENT", _("Indifferent")))
+
+
+# Views
+
+class SearchFormListView(WasSuscribed, SingleObjectMixin, ListView):
+    model = User
+    template_name = 'matmat/search_form.jinja'
+
+    def dispatch(self, request, *args, **kwargs):
+        self.form_class = kwargs['form']
+        self.reverse = kwargs['reverse']
+        if 'valid_form' in kwargs.keys():
+            self.valid_form = kwargs['valid_form']
+        else:
+            self.valid_form = None
+
+        self.init_query = self.model.objects
+        if not (request.user.is_board_member or request.user.is_root):
+            self.init_query = self.init_query.exclude(is_subscriber_viewable=False)
+
+        return super(SearchFormListView, self).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(SearchFormListView, self).get_context_data(**kwargs)
+        kwargs['form'] = self.form_class()
+        return kwargs
+
+    def get_queryset(self):
+        if self.valid_form is not None:
+            if self.reverse:
+                return self.init_query.filter(phone=self.valid_form['phone']).all()
+            else:
+                q = self.init_query
+                # f = self.valid_form
+                return q.all()
+        else:
+            return self.model.objects.none()
+
+
+class SearchFormView(WasSuscribed, FormView):
+    """
+    Allows users to search inside the user list
+    """
+    form_class = SearchForm
+    reverse = False
+
+    def dispatch(self, request, *args, **kwargs):
+        self.init_query = User.objects
+        kwargs['form'] = self.get_form_class()
+        kwargs['reverse'] = self.reverse
+        return super(SearchFormView, self).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()
+        return view(request, *args, **kwargs)
+
+    def get_initial(self):
+        return super(SearchFormView, self).get_initial()
+
+
+class SearchReverseFormView(SearchFormView):
+    reverse = True