site_tps/settings/views.py

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 SchoolProfile
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'] = SchoolProfile.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"
    })