81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
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
|