fix login admin

This commit is contained in:
Cayo Puigdefabregas 2023-11-21 15:20:15 +01:00
parent b3cd5e656d
commit 5ac3790588
1 changed files with 12 additions and 4 deletions

View File

@ -1,6 +1,8 @@
from django.urls import reverse_lazy
from django.utils.translation import gettext_lazy as _
from django.contrib.auth import views as auth_views
from django.contrib.auth import login as auth_login
from django.http import HttpResponseRedirect
class LoginView(auth_views.LoginView):
@ -11,10 +13,16 @@ class LoginView(auth_views.LoginView):
}
def get(self, request, *args, **kwargs):
if request.user.is_admin:
self.extra_context['success_url'] = reverse_lazy(
'idhub:admin_dashboard'
)
if request.GET.get('next'):
self.extra_context['success_url'] = request.GET.get('next')
return super().get(request, *args, **kwargs)
def form_valid(self, form):
user = form.get_user()
if not user.is_anonymous and user.is_admin:
user_dashboard = reverse_lazy('idhub:user_dashboard')
admin_dashboard = reverse_lazy('idhub:admin_dashboard')
if self.extra_context['success_url'] == user_dashboard:
self.extra_context['success_url'] = admin_dashboard
auth_login(self.request, user)
return HttpResponseRedirect(self.extra_context['success_url'])