fix(typing): validate OAuth code before processing access token (#18288)

2025-04-17 16:58:29 +08:00 · 2025-04-17 16:58:29 +08:00 · 8f547e6340
parent defd5520ea
commit 8f547e6340
1 changed files with 3 additions and 1 deletions
--- a/api/controllers/console/auth/data_source_oauth.py
+++ b/api/controllers/console/auth/data_source_oauth.py
@ -74,7 +74,9 @@ class OAuthDataSourceBinding(Resource):
        if not oauth_provider:
            return {"error": "Invalid provider"}, 400
        if "code" in request.args:
-            code = request.args.get("code")
+            code = request.args.get("code", "")
+            if not code:
+                return {"error": "Invalid code"}, 400
            try:
                oauth_provider.get_access_token(code)
            except requests.exceptions.HTTPError as e: