feat: support optionally passing login query string to oauth authoriz…

…e redirect
dpgaspar · Mar 21, 2024 · ebd0aa0 · ebd0aa0
1 parent c65e067
commit ebd0aa0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/flask_appbuilder/security/views.py b/flask_appbuilder/security/views.py
@@ -643,6 +643,11 @@ def login(self, provider: Optional[str] = None) -> WerkzeugResponse:
             request.args.to_dict(flat=False), random_state, algorithm="HS256"
         )
         session["oauth_state"] = random_state
+
+        params = {}
+        if current_app.config.get("AUTH_OAUTH_PRESERVE_LOGIN_QUERY", False):
+            params = request.args
+
         try:
             if provider == "twitter":
                 return self.appbuilder.sm.oauth_remotes[provider].authorize_redirect(
@@ -659,6 +664,7 @@ def login(self, provider: Optional[str] = None) -> WerkzeugResponse:
                         ".oauth_authorized", provider=provider, _external=True
                     ),
                     state=state.decode("ascii") if isinstance(state, bytes) else state,
+                    **params,
                 )
         except Exception as e:
             log.error("Error on OAuth authorize: %s", e)