github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-06-26 06:52:53 +00:00
Code Issues Projects Releases Wiki Activity

perf: update login success redirect

Browse Source
This commit is contained in:
ibuler 2024-12-19 11:38:54 +08:00 committed by feng626
parent f6f897317e
commit fab156dc5f
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/authentication/views/login.py
View File

@ -227,6 +227,10 @@ class UserLoginView(mixins.AuthMixin, UserLoginContextMixin, FormView):
return redirect_url
def get(self, request, *args, **kwargs):
next_page = request.GET.get(self.redirect_field_name)
if next_page:
request.session[self.redirect_field_name] = next_page
if request.user.is_staff:
first_login_url = redirect_user_first_login_or_index(
request, self.redirect_field_name

10
apps/users/utils.py
View File

@ -68,9 +68,13 @@ def get_redirect_client_url(request):
def redirect_user_first_login_or_index(request, redirect_field_name):
url = request.POST.get(redirect_field_name)
if not url:
url = request.GET.get(redirect_field_name)
sources = [request.session, request.POST, request.GET]
url = ''
for source in sources:
url = source.get(redirect_field_name)
if url:
break
if url == 'client':
url = get_redirect_client_url(request)