site_tps/users/views.py

from django.contrib.auth.models import User, Group
from django.views.generic import CreateView, UpdateView
from django.urls import reverse, reverse_lazy
from django.shortcuts import get_object_or_404

from .models import UserProfile, SchoolProfile


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


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(CreateView):
    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 EditSchool(UpdateView):
    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'] = "Édition de l'école"
        context['validate'] = "Modifier"
        return context