# HG changeset patch # User Yader Velasquez # Date 1614730961 21600 # Tue Mar 02 18:22:41 2021 -0600 # Node ID ec8a1e3db30e44ac025e6c8bcd81cb7235ad7e0e # Parent 71d987199d77bfda3e47ae3bdd972f326a4a1af2 Support next GET value diff --git a/nlotp/middleware.py b/nlotp/middleware.py --- a/nlotp/middleware.py +++ b/nlotp/middleware.py @@ -1,5 +1,6 @@ import functools +from django.contrib.auth import REDIRECT_FIELD_NAME from django.http import HttpResponseRedirect from django.utils.deprecation import MiddlewareMixin from django.utils.functional import SimpleLazyObject @@ -29,5 +30,9 @@ else: return None if request.path not in settings.NLOTP_VERIFY_EXCLUDED_URLS: - return HttpResponseRedirect(settings.NLOTP_VERIFY_URL) + next_url = request.GET.get(REDIRECT_FIELD_NAME) + return HttpResponseRedirect( + '{0}?{1}={2}'.format(settings.NLOTP_VERIFY_URL, + REDIRECT_FIELD_NAME, next_url) + ) return None