from django.shortcuts import render, get_object_or_404, redirect from django.urls import reverse from django.contrib.auth.models import User, Group, Permission from django.contrib.auth import authenticate, login, logout from django.contrib import messages from django.db.models import Q from django.http import HttpResponse, HttpResponseRedirect from django.core.paginator import EmptyPage, PageNotAnInteger, Paginator from django.contrib.auth.decorators import login_required, permission_required from django.forms.models import model_to_dict import simplejson as json from datetime import datetime, timedelta from dal import autocomplete import csv from coopeV3.acl import admin_required, superuser_required, self_or_has_perm, active_required from .models import CotisationHistory, WhiteListHistory, School from .forms import CreateUserForm, LoginForm, CreateGroupForm, EditGroupForm, SelectUserForm, GroupsEditForm, EditPasswordForm, addCotisationHistoryForm, addCotisationHistoryForm, addWhiteListHistoryForm, SelectNonAdminUserForm, SelectNonSuperUserForm, SchoolForm, ExportForm from gestion.models import Reload, Consumption, ConsumptionHistory, MenuHistory @active_required def loginView(request): """ Display the login form for :model:`User`. **Context** ``form_entete`` Title of the form. ``form`` The login form. ``form_button`` Content of the form button. **Template** :template:`form.html` """ form = LoginForm(request.POST or None) if(form.is_valid()): user = authenticate(username=form.cleaned_data['username'], password=form.cleaned_data['password']) if user is not None: login(request, user) messages.success(request, "Vous êtes à présent connecté sous le compte " + str(user)) if(request.user.has_perm('gestion.can_manage')): return redirect(reverse('gestion:manage')) else: return redirect(reverse('users:profile', kwargs={'pk':request.user.pk})) else: messages.error(request, "Nom d'utilisateur et/ou mot de passe invalide") return render(request, "form.html", {"form_entete": "Connexion", "form": form, "form_title": "Connexion", "form_button": "Se connecter"}) @active_required @login_required def logoutView(request): """ Logout the logged user """ logout(request) messages.success(request, "Vous êtes à présent déconnecté") return redirect(reverse('home')) @active_required @login_required @permission_required('auth.view_user') def index(request): """ Display the index for user related actions **Template** :template:`users/index.html` """ export_form = ExportForm(request.POST or None) return render(request, "users/index.html", {"export_form": export_form}) def export_csv(request): export_form = ExportForm(request.POST or None) if export_form.is_valid(): users = User.objects qt = export_form.cleaned_data['query_type'] if qt == 'all': users = users.all() filename="Utilisateurs-coope" elif qt == 'all_active': users = users.filter(is_active=True) filename="Utilisateurs-actifs-coope" elif qt == 'adherent': pks = [x.pk for x in User.objects.all() if x.profile.is_adherent] users = users.filter(pk__in=pks) filename="Adherents-coope" elif qt == 'adherent_active': pks = [x.pk for x in User.objects.filter(is_active=True) if x.profile.is_adherent] users = users.filter(pk__in=pks) filename="Adherents-actifs-coope" response = HttpResponse(content_type='text/csv') response['Content-Disposition'] = 'attachment; filename="'+ filename + '.csv"' writer = csv.writer(response) fields = export_form.cleaned_data['fields'] top = ["#"] for field in fields: top.append(dict(ExportForm.FIELDS_CHOICES)[field]) writer.writerow(top) for user in users: userD = model_to_dict(user) profileD = model_to_dict(user.profile) row = [user.pk] for field in fields: if "profile" in field: if "balance" in field: row.append(user.profile.balance) elif "school" in field: row.append(str(user.profile.school)) else: row.append(profileD[field[8:]]) else: row.append(userD[field]) writer.writerow(row) return response else: return redirect(reverse('users:index')) ########## users ########## @active_required @login_required @self_or_has_perm('pk', 'auth.view_user') def profile(request, pk): """ Display the profile for the requested user ``pk`` The primary key for user **Context** ``user`` The instance of User ``self`` Boolean value wich indicates if the current logged user and the request user are the same ``cotisations`` List of the user's cotisations ``whitelists`` List of the user's whitelists ``reloads`` List of the last 5 reloads of the user **Template** :template:`users/profile.html` """ user = get_object_or_404(User, pk=pk) self = request.user == user cotisations = CotisationHistory.objects.filter(user=user) whitelists = WhiteListHistory.objects.filter(user=user) reloads = Reload.objects.filter(customer=user).order_by('-date')[:5] consumptionsChart = Consumption.objects.filter(customer=user) products = [] quantities = [] for ch in consumptionsChart: if ch.product in products: i = products.index(ch.product) quantities[i] += ch.quantity else: products.append(ch.product) quantities.append(ch.quantity) lastConsumptions = ConsumptionHistory.objects.filter(customer=user).order_by('-date')[:10] lastMenus = MenuHistory.objects.filter(customer=user).order_by('-date')[:10] return render(request, "users/profile.html", { "user":user, "self":self, "cotisations":cotisations, "whitelists": whitelists, "reloads": reloads, "products": products, "quantities": quantities, "lastConsumptions": lastConsumptions, "lastMenus": lastMenus, }) @active_required @login_required @permission_required('auth.add_user') def createUser(request): """ Display a CreateUserForm instance. **Context** ``form_entete`` The form title. ``form`` The CreateUserForm instance. ``form_button`` The content of the form button. **Template** :template:`form.html` """ form = CreateUserForm(request.POST or None) if(form.is_valid()): user = form.save(commit=False) user.set_password(user.username) user.save() user.profile.school = form.cleaned_data['school'] user.save() messages.success(request, "L'utilisateur a bien été créé") return redirect(reverse('users:profile', kwargs={'pk':user.pk})) return render(request, "form.html", {"form_entete": "Gestion des utilisateurs", "form":form, "form_title":"Création d'un nouvel utilisateur", "form_button":"Créer l'utilisateur"}) @active_required @login_required @permission_required('auth.view_user') def searchUser(request): """ Display a simple searchForm for User. **Context** ``form_entete`` The form title. ``form`` The searchForm instance. ``form_button`` The content of the form button. **Template** :template:`form.html` """ form = SelectUserForm(request.POST or None) if(form.is_valid()): return redirect(reverse('users:profile', kwargs={"pk":form.cleaned_data['user'].pk})) return render(request, "form.html", {"form_entete": "Gestion des utilisateurs", "form": form, "form_title": "Rechercher un utilisateur", "form_button": "Afficher le profil"}) @active_required @login_required @permission_required('auth.view_user') def usersIndex(request): """ Display the list of all users. **Context** ``users`` The list of all users **Template** :template:`users/users_index.html` """ users = User.objects.all() return render(request, "users/users_index.html", {"users":users}) @active_required @login_required @permission_required('auth.change_user') def editGroups(request, pk): """ Edit the groups of a user. ``pk`` The pk of the user. **Context** ``form_entete`` The form title. ``form`` The GroupsEditForm instance. ``form_button`` The content of the form button. **Template** :template:`form.html` """ user = get_object_or_404(User, pk=pk) form = GroupsEditForm(request.POST or None, instance=user) if(form.is_valid()): form.save() messages.success(request, "Les groupes de l'utilisateur " + user.username + " ont bien été enregistrés.") return redirect(reverse('users:profile', kwargs={'pk':pk})) extra_css = "#id_groups{height:200px;}" return render(request, "form.html", {"form_entete": "Gestion de l'utilisateur " + user.username, "form": form, "form_title": "Modification des groupes", "form_button": "Enregistrer", "extra_css": extra_css}) @active_required @login_required @permission_required('auth.change_user') def editPassword(request, pk): """ Change the password of a user. ``pk`` The pk of the user. **Context** ``form_entete`` The form title. ``form`` The EditPasswordForm instance. ``form_button`` The content of the form button. **Template** :template:`form.html` """ user = get_object_or_404(User, pk=pk) if user != request.user: messages.error(request, "Vous ne pouvez modifier le mot de passe d'un autre utilisateur") return redirect(reverse('home')) else: form = EditPasswordForm(request.POST or None) if(form.is_valid()): if authenticate(username=user.username, password = form.cleaned_data['password']) is not None: user.set_password(form.cleaned_data['password2']) user.save() messages.success(request, "Votre mot de passe a bien été mis à jour") return redirect(reverse('users:profile', kwargs={'pk':pk})) else: messages.error(request, "Le mot de passe actuel est incorrect") return render(request, "form.html", {"form_entete": "Modification de mon compte", "form": form, "form_title": "Modification de mon mot de passe", "form_button": "Modifier mon mot de passe"}) @active_required @login_required @permission_required('auth.change_user') def editUser(request, pk): """ Edit a user and user profile ``pk`` The pk of the user. **Context** ``form_entete`` The form title. ``form`` The CreateUserForm instance. ``form_button`` The content of the form button. **Template** :template:`form.html` """ user = get_object_or_404(User, pk=pk) form = CreateUserForm(request.POST or None, instance=user, initial = {'school': user.profile.school}) if(form.is_valid()): user.profile.school = form.cleaned_data['school'] user.save() messages.success(request, "Les modifications ont bien été enregistrées") return redirect(reverse('users:profile', kwargs={'pk': pk})) return render(request, "form.html", {"form_entete":"Modification du compte " + user.username, "form": form, "form_title": "Modification des informations", "form_button": "Modifier"}) @active_required @login_required @permission_required('auth.change_user') def resetPassword(request, pk): """ Reset the password of a user. ``pk`` The pk of the user """ user = get_object_or_404(User, pk=pk) if user.is_superuser: messages.error(request, "Impossible de réinitialiser le mot de passe de " + user.username + " : il est superuser.") return redirect(reverse('users:profile', kwargs={'pk': pk})) else: user.set_password(user.username) user.save() messages.success(request, "Le mot de passe de " + user.username + " a bien été réinitialisé.") return redirect(reverse('users:profile', kwargs={'pk': pk})) @active_required @login_required @permission_required('auth.view_user') def getUser(request, pk): """ Return username and balance of the requested user (pk) ``pk`` The pk of the user """ user = get_object_or_404(User, pk=pk) data = json.dumps({"username": user.username, "balance": user.profile.balance}) return HttpResponse(data, content_type='application/json') @active_required @login_required @self_or_has_perm('pk', 'auth.view_user') def allReloads(request, pk, page): """ Display all the reloads of the requested user. ``pk`` The pk of the user. ``page`` The page number. **Context** ``reloads`` The reloads of the page. ``user`` The requested user **Template** :template:`users/allReloads.html` """ user = get_object_or_404(User, pk=pk) allReloads = Reload.objects.filter(customer=user).order_by('-date') paginator = Paginator(allReloads, 10) reloads = paginator.get_page(page) return render(request, "users/allReloads.html", {"reloads": reloads, "user":user}) @active_required @login_required @self_or_has_perm('pk', 'auth.view_user') def all_consumptions(request, pk, page): """ Display all the consumptions of the requested user. ``pk`` The pk of the user. ``page`` The page number. **Context** ``reloads`` The reloads of the page. ``user`` The requested user **Template** :template:`users/all_consumptions.html` """ user = get_object_or_404(User, pk=pk) all_consumptions = ConsumptionHistory.objects.filter(customer=user).order_by('-date') paginator = Paginator(all_consumptions, 10) consumptions = paginator.get_page(page) return render(request, "users/all_consumptions.html", {"consumptions": consumptions, "user":user}) @active_required @login_required @self_or_has_perm('pk', 'auth.view_user') def all_menus(request, pk, page): """ Display all the menus of the requested user. ``pk`` The pk of the user. ``page`` The page number. **Context** ``reloads`` The reloads of the page. ``user`` The requested user **Template** :template:`users/all_menus.html` """ user = get_object_or_404(User, pk=pk) all_menus = MenuHistory.objects.filter(customer=user).order_by('-date') paginator = Paginator(all_menus, 10) menus = paginator.get_page(page) return render(request, "users/all_menus.html", {"menus": menus, "user":user}) ########## Groups ########## @active_required @login_required @permission_required('auth.view_group') def groupsIndex(request): """ Display all the groups. **Context** ``groups`` List of all groups. **Template** :template:`users/groups_index.html` """ groups = Group.objects.all() return render(request, "users/groups_index.html", {"groups": groups}) @active_required @login_required @permission_required('auth.view_group') def groupProfile(request, pk): """ Display the profile of a group. ``pk`` The pk of the group. **Context** ``group`` The requested group. **Template** :template:`users/group_profile.html` """ group = get_object_or_404(Group, pk=pk) return render(request, "users/group_profile.html", {"group": group}) @active_required @login_required @permission_required('auth.add_group') def createGroup(request): """ Create a group with a CreateGroupForm instance. **Context** ``form_entete`` The form title. ``form`` The CreateGroupForm instance. ``form_button`` The content of the form button. **Template** :template:`form.html` """ form = CreateGroupForm(request.POST or None) if(form.is_valid()): group = form.save() messages.success(request, "Le groupe " + form.cleaned_data['name'] + " a bien été crée.") return redirect(reverse('users:groupProfile', kwargs={'pk': group.pk})) return render(request, "form.html", {"form_entete": "Gestion des utilisateurs", "form":form, "form_title": "Création d'un groupe de droit", "form_button": "Créer le groupe de droit"}) @active_required @login_required @permission_required('auth.change_group') def editGroup(request, pk): """ Edit a group with a EditGroupForm instance. ``pk`` The pk of the group. **Context** ``form_entete`` The form title. ``form`` The EditGroupForm instance. ``form_button`` The content of the form button. **Template** :template:`form.html` """ group = get_object_or_404(Group, pk=pk) form = EditGroupForm(request.POST or None, instance=group) extra_css = "#id_permissions{height:200px;}" if(form.is_valid()): form.save() messages.success(request, "Le groupe " + group.name + " a bien été modifié.") return redirect(reverse('users:groupProfile', kwargs={'pk': group.pk})) return render(request, "form.html", {"form_entete": "Gestion des utilisateurs", "form": form, "form_title": "Modification du groupe de droit " + group.name, "form_button": "Modifier le groupe de droit", "extra_css":extra_css}) @active_required @login_required @permission_required('auth.delete_group') def deleteGroup(request, pk): """ Delete the requested group. ``pk`` The pk of the group """ group = get_object_or_404(Group, pk=pk) if group.user_set.count() == 0: name = group.name group.delete() messages.success(request, "Le groupe " + name + " a bien été supprimé") return redirect(reverse('users:index') + '#second') else: messages.error(request, "Impossible de supprimer le groupe " + group.name + " : il y a encore des utilisateurs") return redirect(reverse('users:groupProfile', kwargs={'pk': group.pk})) @active_required @login_required @permission_required('auth.change_group') def removeRight(request, groupPk, permissionPk): """ Remove a right from a given group. ``groupPk`` The pk of the group. ``permissionPk`` The pk of the right. """ group = get_object_or_404(Group, pk=groupPk) perm = get_object_or_404(Permission, pk=permissionPk) if perm in group.permissions.all(): group.permissions.remove(perm) messages.success(request, "La permission " + perm.codename + " a bien été retirée du groupe " + group.name) else: messages.error(request, "Impossible de retirer la permission " + perm.codename + " du groupe " + group.name) return redirect(reverse('users:groupProfile', kwargs={'pk': groupPk}) + "#second") @active_required @login_required @permission_required('auth.change_user') def removeUser(request, groupPk, userPk): """ Remove a user from a given group. ``groupPk`` The pk of the group. ``userPk`` The pk of the user. """ group = get_object_or_404(Group, pk=groupPk) user = get_object_or_404(User, pk=userPk) if(group in user.groups.all()): user.groups.remove(group) messages.success(request, "L'utilisateur " + user.username + " a bien été retiré du groupe " + group.name) else: messages.error(request, "Impossible de retirer l'utilisateur " + user.username + " du groupe " + group.name) return redirect(reverse('users:groupProfile', kwargs={'pk': groupPk}) + "#second") ########## admins ########## @active_required @login_required @admin_required def adminsIndex(request): """ Lists the staff **Context** ``admins`` List of staff **Template** :template:`users/admins_index.html` """ admins = User.objects.filter(is_staff=True) return render(request, "users/admins_index.html", {"admins": admins}) @active_required @login_required @admin_required def addAdmin(request): """ Form to add a member to staff **Context** ``form`` The SelectNonAdminUserForm form instance ``form_title`` The title of the form ``form_button`` The text of the button **Template** :template:`form.html` """ form = SelectNonAdminUserForm(request.POST or None) if(form.is_valid()): user = form.cleaned_data['user'] user.is_staff = True user.save() messages.success(request, "L'utilisateur " + user.username + " a bien été rajouté aux admins") return redirect(reverse('users:adminsIndex')) return render(request, "form.html", {"form": form, "form_title": "Ajout d'un admin", "form_button":"Ajouter l'utilisateur aux admins"}) @active_required @login_required @admin_required def removeAdmin(request, pk): """ Remove an user form staff ``pk`` The primary key of the user """ user = get_object_or_404(User, pk=pk) if user.is_staff: if user.is_superuser: messages.error(request, "Impossible de retirer l'utilisateur " + user.username + " des admins : il est superuser") else: if User.objects.filter(is_staff=True).count() > 1: user.is_staff = False user.save() messages.success(request, "L'utilisateur " + user.username + " a bien été retiré des admins.") else: messages.error(request, "Impossible de retirer l'utilisateur " + user.username + " des admins : il doit en rester au moins un.") else: messages.error(request, "Impossible de retirer l'utilisateur " + user.username + " des admins : il n'en fait pas partie.") return redirect(reverse('users:adminsIndex')) ########## superusers ########## @active_required @login_required @superuser_required def superusersIndex(request): """ Lists the superusers **Context** ``superusers`` List of superusers **Template** :template:`users/superusers_index.html` """ superusers = User.objects.filter(is_superuser=True) return render(request, "users/superusers_index.html", {"superusers": superusers}) @active_required @login_required @superuser_required def addSuperuser(request): """ Displays a form to add a superuser **Context** ``form`` The SelectNonSuperUserForm form instance ``form_title`` The title of the form ``form_button`` The text of the button **Template** :template:`form.html` """ form = SelectNonSuperUserForm(request.POST or None) if form.is_valid(): user = form.cleaned_data['user'] user.is_admin = True user.is_superuser = True user.save() messages.success(request, "L'utilisateur " + user.username + " a bien été rajouté aux superusers") return redirect(reverse('users:superusersIndex')) return render(request, "form.html", {"form_entete": "Gestion des superusers", "form": form, "form_title": "Ajout d'un superuser", "form_button":"Ajouter l'utilisateur aux superusers"}) @active_required @login_required @superuser_required def removeSuperuser(request, pk): """ Removes a user from superusers ``pk`` The primary key of the user """ user = get_object_or_404(User, pk=pk) if user.is_superuser: if User.objects.filter(is_superuser=True).count() > 1: user.is_superuser = False user.save() messages.success(request, "L'utilisateur " + user.username + " a bien été retiré des superusers.") else: messages.error(request, "Impossible de retirer l'utilisateur " + user.username + " des superusers : il doit en rester au moins un.") else: messages.error(request, "Impossible de retirer l'utilisateur " + user.username + " des superusers : il n'en fait pas partie.") return redirect(reverse('users:superusersIndex')) ########## Cotisations ########## @active_required @login_required @permission_required('users.add_cotisationhistory') def addCotisationHistory(request, pk): """ Add a cotisation to the requested user ``pk`` The primary key of the user **Context** ``form`` The addCotisationHistoryForm form instance ``form_title`` The title of the form ``form_button`` The text of the button **Template** :template:`form.html` """ user = get_object_or_404(User, pk=pk) form = addCotisationHistoryForm(request.POST or None) if(form.is_valid()): cotisation = form.save(commit=False) if(cotisation.paymentMethod.affect_balance): if(user.profile.balance >= cotisation.cotisation.amount): user.profile.debit += cotisation.cotisation.amount else: messages.error(request, "Solde insuffisant") return redirect(reverse('users:profile',kwargs={'pk':user.pk})) cotisation.user = user cotisation.coopeman = request.user cotisation.amount = cotisation.cotisation.amount cotisation.duration = cotisation.cotisation.duration if(user.profile.cotisationEnd): cotisation.endDate = user.profile.cotisationEnd + timedelta(days=cotisation.cotisation.duration) else: cotisation.endDate = datetime.now() + timedelta(days=cotisation.cotisation.duration) user.profile.cotisationEnd = cotisation.endDate user.save() cotisation.save() messages.success(request, "La cotisation a bien été ajoutée") return redirect(reverse('users:profile',kwargs={'pk':user.pk})) return render(request, "form.html",{"form": form, "form_title": "Ajout d'une cotisation pour l'utilisateur " + str(user), "form_button": "Ajouter"}) @active_required @login_required @permission_required('users.validate_cotisationhistory') def validateCotisationHistory(request, pk): """ Validate the requested :model:`users.CotisationHistory` ``pk`` The primary key of the :model:`users.CotisationHistory` """ cotisationHistory = get_object_or_404(CotisationHistory, pk=pk) cotisationHistory.valid = CotisationHistory.VALID cotisationHistory.save() messages.success(request, "La cotisation a bien été validée") return HttpResponseRedirect(request.META.get('HTTP_REFERER')) @active_required @login_required @permission_required('users.validate_cotisationhistory') def invalidateCotisationHistory(request, pk): """ Invalidate the requested :model:`users.CotisationHistory` ``pk`` The primary key of the :model:`users.CotisationHistory` """ cotisationHistory = get_object_or_404(CotisationHistory, pk=pk) cotisationHistory.valid = CotisationHistory.INVALID cotisationHistory.save() user = cotisationHistory.user user.profile.cotisationEnd = user.profile.cotisationEnd - timedelta(days=cotisationHistory.duration) if(cotisationHistory.paymentMethod.affect_balance): user.profile.balance += cotisation.amount user.save() messages.success(request, "La cotisation a bien été invalidée") return HttpResponseRedirect(request.META.get('HTTP_REFERER')) ########## Whitelist ########## @active_required @login_required @permission_required('users.add_whitelisthistory') def addWhiteListHistory(request, pk): """ Add a :model:`users.WhitelistHistory` to the requested user ``pk`` The primary key of the user **Context** ``form`` The addWhiteListHistoryForm form instance ``form_title`` The title of the form ``form_button`` The text of the button **Template** :template:`form.html` """ user = get_object_or_404(User, pk=pk) form = addWhiteListHistoryForm(request.POST or None) if(form.is_valid()): whiteList = form.save(commit=False) whiteList.user = user whiteList.coopeman = request.user if(user.profile.cotisationEnd): whiteList.endDate = user.profile.cotisationEnd + timedelta(days=whiteList.duration) else: whiteList.endDate = datetime.now() + timedelta(days=whiteList.duration) user.profile.cotisationEnd = whiteList.endDate user.save() whiteList.save() messages.success(request, "L'accès gracieux a bien été ajouté") return redirect(reverse('users:profile', kwargs={'pk':user.pk})) return render(request, "form.html", {"form": form, "form_title": "Ajout d'un accès gracieux pour " + user.username, "form_button": "Ajouter"}) ########## Schools ########## @active_required @login_required @permission_required('users.view_school') def schoolsIndex(request): """ Lists the :model:`users.School` **Context** ``schools`` List of the :model:`users.School` **Template** :template:`users/schools_index.html` """ schools = School.objects.all() return render(request, "users/schools_index.html", {"schools": schools}) @active_required @login_required @permission_required('users.add_school') def createSchool(request): """ Displays form to create :model:`users.School` **Context** ``form`` The SchoolForm form instance ``form_title`` The title of the form ``form_button`` The text of the button **Template** :template:`form.html` """ form = SchoolForm(request.POST or None) if form.is_valid(): form.save() messages.success(request, "L'école a bien été créée") return redirect(reverse('users:schoolsIndex')) return render(request, "form.html", {"form": form, "form_title": "Création d'une école", "form_button": "Créer"}) @active_required @login_required @permission_required('users.change_school') def editSchool(request, pk): """ Displays form to create :model:`users.School` ``pk`` The primary key of :model:`users.School` **Context** ``form`` The SchoolForm form instance ``form_title`` The title of the form ``form_button`` The text of the button **Template** :template:`form.html` """ school = get_object_or_404(School, pk=pk) form = SchoolForm(request.POST or None, instance=school) if(form.is_valid()): form.save() messages.success(request, "L'école a bien été modifiée") return redirect(reverse('users:schoolsIndex')) return render(request, "form.html", {"form": form, "form_title": "Modification de l'école " + str(school), "form_button": "Modifier"}) @active_required @login_required @permission_required('users.delete_school') def deleteSchool(request, pk): """ Delete a :model:`users.School` ``pk`` The primary key of the school to delete """ school = get_object_or_404(School, pk=pk) message = "L'école " + str(school) + " a bien été supprimée" school.delete() messages.success(request, message) return redirect(reverse('users:schoolsIndex')) ########## Autocomplete searchs ########## class AllUsersAutocomplete(autocomplete.Select2QuerySetView): """ Autcomplete for all users """ def get_queryset(self): qs = User.objects.all() if self.q: qs = qs.filter(Q(username__istartswith=self.q) | Q(first_name__istartswith=self.q) | Q(last_name__istartswith=self.q)) return qs class ActiveUsersAutocomplete(autocomplete.Select2QuerySetView): """ Autocomplete for active users """ def get_queryset(self): qs = User.objects.filter(is_active=True) if self.q: qs = qs.filter(Q(username__istartswith=self.q) | Q(first_name__istartswith=self.q) | Q(last_name__istartswith=self.q)) return qs class AdherentAutocomplete(autocomplete.Select2QuerySetView): """ Autocomplete for adherents """ def get_queryset(self): qs = User.objects.all() return qs class NonSuperUserAutocomplete(autocomplete.Select2QuerySetView): """ Autocomplete for non-superuser users """ def get_queryset(self): qs = User.objects.filter(is_superuser=False) if self.q: qs = qs.filter(Q(username__istartswith=self.q) | Q(first_name__istartswith=self.q) | Q(last_name__istartswith=self.q)) return qs class NonAdminUserAutocomplete(autocomplete.Select2QuerySetView): """ Autocomplete for non-admin users """ def get_queryset(self): qs = User.objects.filter(is_staff=False) if self.q: qs = qs.filter(Q(username__istartswith=self.q) | Q(first_name__istartswith=self.q) | Q(last_name__istartswith=self.q)) return qs