You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.6 KiB
47 lines
1.6 KiB
3 years ago
|
"""
|
||
|
authlib.openid.connect.core.tokens
|
||
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
||
|
This module contains methods for adding JWT tokens to requests.
|
||
|
"""
|
||
|
from oauthlib.oauth2.rfc6749.tokens import TokenBase, random_token_generator, get_token_from_header
|
||
|
|
||
|
|
||
|
class JWTToken(TokenBase):
|
||
|
__slots__ = (
|
||
|
'request_validator', 'token_generator',
|
||
|
'refresh_token_generator', 'expires_in'
|
||
|
)
|
||
|
|
||
|
def __init__(self, request_validator=None, token_generator=None,
|
||
|
expires_in=None, refresh_token_generator=None):
|
||
|
self.request_validator = request_validator
|
||
|
self.token_generator = token_generator or random_token_generator
|
||
|
self.refresh_token_generator = (
|
||
|
refresh_token_generator or self.token_generator
|
||
|
)
|
||
|
self.expires_in = expires_in or 3600
|
||
|
|
||
|
def create_token(self, request, refresh_token=False):
|
||
|
"""Create a JWT Token, using requestvalidator method."""
|
||
|
|
||
|
if callable(self.expires_in):
|
||
|
expires_in = self.expires_in(request)
|
||
|
else:
|
||
|
expires_in = self.expires_in
|
||
|
|
||
|
request.expires_in = expires_in
|
||
|
|
||
|
return self.request_validator.get_jwt_bearer_token(None, None, request)
|
||
|
|
||
|
def validate_request(self, request):
|
||
|
token = get_token_from_header(request)
|
||
|
return self.request_validator.validate_jwt_bearer_token(
|
||
|
token, request.scopes, request)
|
||
|
|
||
|
def estimate_type(self, request):
|
||
|
token = get_token_from_header(request)
|
||
|
if token and token.startswith('ey') and token.count('.') in (2, 4):
|
||
|
return 10
|
||
|
return 0
|