(HP-1699): revert settings, add header to fetch_token

uc-cdis · Sep 18, 2024 · 0e76a2d · 0e76a2d
1 parent bfd534b
commit 0e76a2d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/wts/api.py b/wts/api.py
@@ -109,7 +109,6 @@ def load_settings(app):
                 "redirect_uri": redirect_uri,
                 "scope": scope,
                 "state_prefix": state_prefix,
-                "headers": {"User-Agent": "Gen3WTS / 0.4.3"},
             }
 
     app.config["COMMONS_HOSTNAMES"] = list(

diff --git a/wts/resources/oauth2.py b/wts/resources/oauth2.py
@@ -28,7 +28,11 @@ def client_do_authorize():
     if mismatched_state:
         raise AuthError("could not authorize; state did not match across auth requests")
     try:
-        tokens = client.fetch_token(token_url, **flask.request.args.to_dict())
+        tokens = client.fetch_token(
+            token_url,
+            **flask.request.args.to_dict(),
+            headers={"User-Agent": "Gen3WTS / 0.4.3"}
+        )
         refresh_refresh_token(tokens, requested_idp, username_field)
     except KeyError as e:
         raise AuthError("error in token response: {}".format(tokens))