ereuse/IdHub
8
0
Fork 0
Code Issues 1 Pull requests Packages Projects Releases Wiki Activity

fix login admin

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

16
idhub/views.py
View file

@ -1,6 +1,8 @@
from django.urls import reverse_lazy from django.urls import reverse_lazy
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
from django.contrib.auth import views as auth_views 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): class LoginView(auth_views.LoginView):
@ -11,10 +13,16 @@ class LoginView(auth_views.LoginView):
} }
def get(self, request, *args, **kwargs): 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'): if request.GET.get('next'):
self.extra_context['success_url'] = request.GET.get('next') self.extra_context['success_url'] = request.GET.get('next')
return super().get(request, *args, **kwargs) 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'])