from django.views.generic import TemplateView, UpdateView from django.urls import reverse_lazy, reverse from django.contrib.auth.mixins import PermissionRequiredMixin, LoginRequiredMixin from django.contrib.auth.decorators import permission_required from django.contrib.auth.models import Group, User from django.shortcuts import get_object_or_404, redirect, render from django.contrib import messages from content.models import Category from users.models import School from .models import SiteSettings from .forms import SelectUserForm class SettingsView(LoginRequiredMixin, PermissionRequiredMixin, TemplateView): template_name = "settings/settings.html" permission_required = 'settings.change_sitesettings' def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['categories'] = Category.objects.all() context['site_settings'], _ = SiteSettings.objects.get_or_create() context['schools'] = School.objects.all() context['settings'] = True context['administrators'] = Group.objects.get(name='admins').user_set.all() return context class EditSiteSettingsView(LoginRequiredMixin, PermissionRequiredMixin, UpdateView): template_name = "edit.html" model = SiteSettings fields = '__all__' success_url = reverse_lazy('settings:index') permission_required = 'settings.change_sitesettings' def get_object(self, queryset=None): obj,_ = self.model.objects.get_or_create() return obj def get_context_data(self, **kwargs): context = super(UpdateView, self).get_context_data(**kwargs) context['title'] = "Édition des " + self.object.PRETTY_NAME return context @permission_required('auth.change_user') def degrade_user(request, pk): user = get_object_or_404(User, pk=pk) admins,_ = Group.objects.get_or_create(name='admins') user.groups.remove(admins) user.save() messages.success(request, user.username + ' a été enlevé des administrateurs du site') return redirect(reverse('settings:index')) @permission_required('auth.change_user') def promote_user(request): user_form = SelectUserForm(request.POST or None) user_form.populate() if user_form.is_valid(): user=user_form.get_user() admins,_ = Group.objects.get_or_create(name='admins') user.groups.add(admins) user.save() messages.success(request, user.username + ' a été ajouté des administrateurs du site') return redirect(reverse('settings:index')) return render(request, 'edit.html', { 'form' : user_form, 'title' : "Ajout d'un administrateur", 'validate' : "Ajouter" })