83 lines
2.0 KiB
Python
83 lines
2.0 KiB
Python
import secrets
|
|
from collections.abc import Collection
|
|
|
|
from authlib.oauth2.rfc6749 import (
|
|
ClientMixin,
|
|
list_to_scope,
|
|
scope_to_list,
|
|
)
|
|
|
|
|
|
class OAuth2Client(ClientMixin):
|
|
def __init__(
|
|
self,
|
|
client_id: str,
|
|
client_secret: str,
|
|
scope: str,
|
|
redirect_uris: list,
|
|
response_types: list,
|
|
grant_types: list,
|
|
token_endpoint_auth_method: str = 'client_secret_basic',
|
|
) -> None:
|
|
self.client_id = client_id
|
|
self.client_secret = client_secret
|
|
self.scope = scope
|
|
self.redirect_uris = redirect_uris
|
|
self.response_types = response_types
|
|
self.grant_types = grant_types
|
|
self.token_endpoint_auth_method = token_endpoint_auth_method
|
|
|
|
def get_client_id(self):
|
|
return self.client_id
|
|
|
|
def get_allowed_scope(
|
|
self,
|
|
scope: Collection[str],
|
|
) -> str:
|
|
if not scope:
|
|
return ''
|
|
|
|
allowed = set(self.scope.split())
|
|
scopes = scope_to_list(scope)
|
|
return list_to_scope([s for s in scopes if s in allowed])
|
|
|
|
def get_default_redirect_uri(self) -> str:
|
|
if self.redirect_uris:
|
|
return self.redirect_uris[0]
|
|
|
|
raise ValueError('Missing redirect_uris')
|
|
|
|
def check_response_type(
|
|
self,
|
|
response_type: str,
|
|
) -> bool:
|
|
return response_type in self.response_types
|
|
|
|
def check_redirect_uri(
|
|
self,
|
|
redirect_uri: str,
|
|
) -> bool:
|
|
return redirect_uri in self.redirect_uris
|
|
|
|
def check_endpoint_auth_method(
|
|
self,
|
|
method: str,
|
|
endpoint: str,
|
|
) -> bool:
|
|
if endpoint == 'token':
|
|
return self.token_endpoint_auth_method == method
|
|
|
|
return True
|
|
|
|
def check_grant_type(
|
|
self,
|
|
grant_type: str,
|
|
) -> bool:
|
|
return grant_type in self.grant_types
|
|
|
|
def check_client_secret(
|
|
self,
|
|
client_secret: str,
|
|
) -> bool:
|
|
return secrets.compare_digest(self.client_secret, client_secret)
|