from django.contrib.auth.models import User, Group from django.contrib.auth.mixins import PermissionRequiredMixin, LoginRequiredMixin from django.views.generic import CreateView, UpdateView, DeleteView, DetailView from django.contrib.auth.views import LoginView, LogoutView, PasswordChangeView, login_required from django.contrib.auth.hashers import make_password from django.contrib.messages.views import SuccessMessageMixin from django.contrib import messages from django.urls import reverse, reverse_lazy from django.shortcuts import get_object_or_404, redirect from .models import UserProfile, SchoolProfile from content.models import Content class CreateUser(CreateView): model = User fields = [ 'first_name', 'last_name', 'email', 'username', 'password', ] template_name = 'edit.html' def get_success_url(self): return reverse( 'users:create-userprofile', kwargs={'pk': self.object.pk} ) def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['title'] = "Inscription" context['validate'] = "S'inscrire" return context def form_valid(self, form): r = super().form_valid(form) self.object.set_password(form.cleaned_data['password']) self.object.save() return r class Profile(LoginRequiredMixin, UpdateView): model = User template_name = 'users/profile.html' fields = [ 'username', 'first_name', 'last_name', 'email' ] def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['title'] = self.object.username context['validate'] = "Modifier" return context def get_success_url(self): return reverse( 'users:profile', kwargs={'pk': self.object.pk} ) class CreateUserProfile(CreateView): model = UserProfile fields = ['school'] template_name = 'edit.html' success_url = reverse_lazy('home') def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['title'] = "Choix de l'école" context['validate'] = "Choisir" return context def form_valid(self, form): form.instance.user = get_object_or_404(User, pk=self.kwargs['pk']) return super(CreateUserProfile, self).form_valid(form) class CreateSchool(LoginRequiredMixin, PermissionRequiredMixin, CreateView): permission_required = 'users.add_schoolprofile' model = Group fields = ['name'] template_name = 'edit.html' success_url = reverse_lazy('settings:index') def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['title'] = "Création de l'école" context['validate'] = "Créer" return context def form_valid(self, form): response = super(CreateSchool, self).form_valid(form) profile = SchoolProfile() profile.group = form.instance profile.save() return response class EditSchoolName(LoginRequiredMixin, PermissionRequiredMixin, UpdateView): model = Group fields = ['name'] template_name = 'edit.html' queryset = Group.objects.filter(school__isnull=False) def get_success_url(self): return reverse('users:edit-school-phone', kwargs={'pk':self.object.school.pk}) def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['title'] = "Édition de l'école" context['validate'] = "Modifier" return context def has_permission(self): return self.request.user.has_perm('users.manage_'+str(self.kwargs['pk'])) def form_valid(self, *args, **kwargs): r = super().form_valid(*args, **kwargs) self.object.school.save() return r class EditSchoolPhone(LoginRequiredMixin, PermissionRequiredMixin, UpdateView): model = SchoolProfile fields = ['phone'] template_name = 'edit.html' def get_success_url(self): return reverse('users:school', kwargs={'pk':self.object.group.pk}) def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['title'] = "Édition de l'école" context['validate'] = "Modifier" return context def has_permission(self): return self.request.user.has_perm('users.manage_'+str(self.kwargs['pk'])) class DeleteSchool(LoginRequiredMixin, PermissionRequiredMixin, DeleteView): model = Group permission_required = 'users.delete_schoolprofile' queryset = Group.objects.filter(school__isnull=False) class School(LoginRequiredMixin, PermissionRequiredMixin, DetailView): model = Group template_name = "users/school.html" queryset = Group.objects.filter(school__isnull=False) def get_context_data(self, **kwargs): context = super().get_context_data() context['contents'] = Content.objects.filter(school_owner=self.object.school) context['school'] = True context['members'] = User.objects.filter(userprofile__school=self.object.school) context['manager_right'] = 'auth.manage_' + str(self.object.pk) context['manager_group'],_ = Group.objects.get_or_create(name=str(self.object.pk)+'_admins') return context def has_permission(self): return self.request.user.has_perm('users.view_'+str(self.kwargs['pk'])) class Logout(SuccessMessageMixin, LogoutView): success_message = "Vous vous êtes bien déconnecté." class Login(SuccessMessageMixin, LoginView): template_name = "edit.html" success_message = "Bienvenue !" extra_context = { 'title' : "Connexion", 'validate' : "Se connecter", } class PasswordChange(SuccessMessageMixin, PasswordChangeView): template_name = "edit.html" success_url = reverse_lazy("home") success_message = "Le mot de passe a été changé." extra_context = { 'title' : "Changer le mot de passe", 'validate' : "Changer", } @login_required def promote_user(request, school_pk, user_pk): school = get_object_or_404(Group, pk=school_pk) user = get_object_or_404(User, pk=user_pk) if request.user.has_perm('manage_'+str(school.pk)): admins,_ = Group.objects.get_or_create(name=str(school.pk)+'_admins') user.groups.add(admins) user.save() messages.success(request, user.username + ' a été ajouté aux administrateurs de ' + school.name) return redirect(reverse('users:school', kwargs={'pk':school.pk})) messages.error(request, "Vous n'aves pas ce droit.") return redirect('home') @login_required def degrade_user(request, school_pk, user_pk): school = get_object_or_404(Group, pk=school_pk) user = get_object_or_404(User, pk=user_pk) if request.user.has_perm('manage_'+str(school.pk)): admins,_ = Group.objects.get_or_create(name=str(school.pk)+'_admins') user.groups.remove(admins) user.save() messages.success(request, user.username + ' a été enlevé des administrateurs de ' + school.name) return redirect(reverse('users:school', kwargs={'pk':school.pk})) messages.error(request, "Vous n'aves pas ce droit.") return redirect('home')