49 lines
No EOL
1.7 KiB
Python
49 lines
No EOL
1.7 KiB
Python
from django.contrib.auth.views import LoginView as Login, LogoutView as Logout
|
|
from django.http.response import HttpResponseRedirect
|
|
from django.contrib.auth import login
|
|
from django.shortcuts import resolve_url
|
|
from django.conf import settings
|
|
from django.views.generic import FormView
|
|
from django.contrib import messages
|
|
|
|
from .forms import RegistrationForm, VerificationForm
|
|
from .models import User
|
|
from .mixins import SuperUserRequiredMixin
|
|
|
|
from clients.models import ClientProfile
|
|
from partners.models import PartnerProfile
|
|
|
|
class LoginView(Login):
|
|
template_name = "localauth/register.html"
|
|
|
|
class LogoutView(Logout):
|
|
next_page = "/"
|
|
|
|
class RegistrationView(Login):
|
|
form_class = RegistrationForm
|
|
template_name = "localauth/register.html"
|
|
|
|
def form_valid(self, form):
|
|
user = User.objects.create_user(form.cleaned_data["email"])
|
|
user.set_password(form.cleaned_data["password1"])
|
|
login(self.request, user)
|
|
|
|
messages.success(self.request, "Erfolgreich registriert!")
|
|
return HttpResponseRedirect(self.get_success_url())
|
|
|
|
def get_default_redirect_url(self):
|
|
return resolve_url(self.next_page or settings.REGISTER_REDIRECT_URL)
|
|
|
|
class VerificationView(SuperUserRequiredMixin, FormView):
|
|
form_class = VerificationForm
|
|
template_name = "localauth/verify.html"
|
|
|
|
def form_valid(self, form):
|
|
pid = form.cleaned_data["profile"]
|
|
ptype = ClientProfile if profile.startswith("C") else PartnerProfile
|
|
pobj = ptype.objects.get(id=profile[1:])
|
|
pobj.update(verified=True)
|
|
|
|
messages.success(self.request, "Benutzer %s bestätigt!" % pobj.full_name)
|
|
|
|
return HttpResponseRedirect(resolve_url("localauth:verify")) |