74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
|
import logging
|
||
|
|
||
|
from django.urls import reverse_lazy
|
||
|
from django.contrib.auth import views as auth_views
|
||
|
from django.contrib.auth import login as auth_login
|
||
|
from django.contrib.auth import logout as auth_logout
|
||
|
from django.utils.translation import gettext_lazy as _
|
||
|
from django.shortcuts import redirect
|
||
|
from django.http import HttpResponseRedirect
|
||
|
|
||
|
|
||
|
logger = logging.getLogger(__name__)
|
||
|
|
||
|
|
||
|
class LoginView(auth_views.LoginView):
|
||
|
template_name = 'login.html'
|
||
|
extra_context = {
|
||
|
'title': _('Login'),
|
||
|
'success_url': reverse_lazy('dashboard:dashboard'),
|
||
|
# 'commit_id': settings.COMMIT,
|
||
|
}
|
||
|
|
||
|
def get(self, request, *args, **kwargs):
|
||
|
self.extra_context['success_url'] = request.GET.get(
|
||
|
'next',
|
||
|
reverse_lazy('dashboard:dashboard')
|
||
|
)
|
||
|
if not self.request.user.is_anonymous:
|
||
|
return redirect(reverse_lazy('dashboard:dashboard'))
|
||
|
|
||
|
return super().get(request, *args, **kwargs)
|
||
|
|
||
|
def form_valid(self, form):
|
||
|
user = form.get_user()
|
||
|
auth_login(self.request, user)
|
||
|
|
||
|
if user.is_anonymous:
|
||
|
return redirect(reverse_lazy("login:login"))
|
||
|
|
||
|
return redirect(self.extra_context['success_url'])
|
||
|
|
||
|
|
||
|
def LogoutView(request):
|
||
|
auth_logout(request)
|
||
|
return redirect(reverse_lazy("login:login"))
|
||
|
|
||
|
|
||
|
class PasswordResetConfirmView(auth_views.PasswordResetConfirmView):
|
||
|
template_name = 'password_reset_confirm.html'
|
||
|
success_url = reverse_lazy('login:password_reset_complete')
|
||
|
|
||
|
def form_valid(self, form):
|
||
|
password = form.cleaned_data.get("new_password1")
|
||
|
user = form.user
|
||
|
user.set_password(password)
|
||
|
user.save()
|
||
|
return HttpResponseRedirect(self.success_url)
|
||
|
|
||
|
|
||
|
class PasswordResetView(auth_views.PasswordResetView):
|
||
|
template_name = 'password_reset.html'
|
||
|
email_template_name = 'password_reset_email.txt'
|
||
|
html_email_template_name = 'password_reset_email.html'
|
||
|
subject_template_name = 'password_reset_subject.txt'
|
||
|
success_url = reverse_lazy('login:password_reset_done')
|
||
|
|
||
|
def form_valid(self, form):
|
||
|
try:
|
||
|
return super().form_valid(form)
|
||
|
except Exception as err:
|
||
|
logger.error(err)
|
||
|
return HttpResponseRedirect(self.success_url)
|
||
|
|