Skip to content

Commit

Permalink
Precommit errors fixed.
Browse files Browse the repository at this point in the history
Signed-off-by: Lalith Kota <[email protected]>
  • Loading branch information
lalithkota committed Sep 17, 2024
1 parent 7129993 commit 6e0bef9
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions g2p_auth_oidc/models/auth_oauth_provider.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,7 @@ def _oidc_get_tokens_auth_code_flow(self, params, oidc_redirect_uri=None):
redirect_uri=oidc_redirect_uri,
)
if self.enable_pkce:
token_request_data['code_verifier'] = self.code_verifier
token_request_data["code_verifier"] = self.code_verifier
response = requests.post(self.token_endpoint, data=token_request_data, timeout=10)
response.raise_for_status()
response_json = response.json()
Expand All @@ -160,7 +160,7 @@ def _oidc_get_tokens_auth_code_flow(self, params, oidc_redirect_uri=None):
redirect_uri=oidc_redirect_uri,
)
if self.enable_pkce:
token_request_data['code_verifier'] = self.code_verifier
token_request_data["code_verifier"] = self.code_verifier
response = requests.post(
self.token_endpoint,
auth=token_request_auth,
Expand All @@ -179,7 +179,7 @@ def _oidc_get_tokens_auth_code_flow(self, params, oidc_redirect_uri=None):
redirect_uri=oidc_redirect_uri,
)
if self.enable_pkce:
token_request_data['code_verifier'] = self.code_verifier
token_request_data["code_verifier"] = self.code_verifier
response = requests.post(self.token_endpoint, data=token_request_data, timeout=10)
response.raise_for_status()
response_json = response.json()
Expand All @@ -195,7 +195,7 @@ def _oidc_get_tokens_auth_code_flow(self, params, oidc_redirect_uri=None):
redirect_uri=oidc_redirect_uri,
)
if self.enable_pkce:
token_request_data['code_verifier'] = self.code_verifier
token_request_data["code_verifier"] = self.code_verifier
response = requests.post(self.token_endpoint, data=token_request_data, timeout=10)
response.raise_for_status()
response_json = response.json()
Expand Down Expand Up @@ -493,7 +493,9 @@ def list_providers(
if flow and flow.startswith("oidc"):
params["nonce"] = secrets.token_urlsafe()
if provider.get("enable_pkce"):
params["code_challenge"] = base64.urlsafe_b64encode(hashlib.sha256(provider["code_verifier"].encode("ascii")).digest()).rstrip(b"=")
params["code_challenge"] = base64.urlsafe_b64encode(
hashlib.sha256(provider["code_verifier"].encode("ascii")).digest()
).rstrip(b"=")
params["code_challenge_method"] = "S256"
extra_auth_params = json.loads(provider.get("extra_authorize_params") or "{}")
params.update(extra_auth_params)
Expand Down

0 comments on commit 6e0bef9

Please sign in to comment.