Add method to clear redirect url from session.

1 files changed, 7 insertions, 0 deletions

diff --git a/gn2/wqflask/oauth2/session.py b/gn2/wqflask/oauth2/session.py
index 78c766a8..883a3228 100644
--- a/gn2/wqflask/oauth2/session.py
+++ b/gn2/wqflask/oauth2/session.py
@@ -98,6 +98,7 @@ def set_masquerading(masq_info):
         "masquerading": orig_user
     })
 
+
 def unset_masquerading():
     """Restore the original session."""
     the_session = session_info()
@@ -120,9 +121,15 @@ def is_token_refreshing():
     """Returns whether the token is being refreshed or not."""
     return session_info().get("token_refreshing", False)
 
+
 def set_redirect_url(url):
     """Save the current endpoint object"""
     return save_session_info({
         **session_info(),
         "redirect_url": url
     })
+
+
+def clear_redirect_url():
+    """Clear the redirect url from Session"""
+    session_info().pop("redirect_url", None)