Merge pull request #57 from wuzhihui1123/master

next vaule: Priority use request.full_path
alexferl · Jul 14, 2019 · 6f5e15e · 6f5e15e
2 parents d69968d + 4337d94
commit 6f5e15e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flask_simpleldap/__init__.py b/flask_simpleldap/__init__.py
@@ -304,7 +304,7 @@ def login_required(func):
         def wrapped(*args, **kwargs):
             if g.user is None:
                 return redirect(url_for(current_app.config['LDAP_LOGIN_VIEW'],
-                                        next=request.path))
+                                        next=request.full_path or request.path))
             return func(*args, **kwargs)
 
         return wrapped
@@ -330,7 +330,7 @@ def wrapped(*args, **kwargs):
                 if g.user is None:
                     return redirect(
                         url_for(current_app.config['LDAP_LOGIN_VIEW'],
-                                next=request.path))
+                                next=request.full_path or request.path))
                 match = [group for group in groups if group in g.ldap_groups]
                 if not match:
                     abort(401)