remove prefix

This commit is contained in:
2025-03-26 12:54:10 -03:00
parent 98a945efbc
commit 3564119581
8 changed files with 62 additions and 63 deletions

View File

@@ -37,7 +37,7 @@ from botocore.endpoint_provider import Enum
from cognito import get_user
APIKEY_PREFIX = 'edxg'
APIKEY_PREFIX = 'sk-'
tracer = Tracer()
logger = Logger(__name__)
@@ -76,19 +76,15 @@ class BearerToken:
token: str
def _parse_bearer_token(
s: str,
*,
apikey_prefix: str = APIKEY_PREFIX,
) -> BearerToken | None:
def _parse_bearer_token(s: str) -> BearerToken | None:
"""Parses and identifies a bearer token as either an API key or a user token."""
try:
_, bearer_token = s.split(' ')
if bearer_token.startswith(f'{apikey_prefix}-'):
if bearer_token.startswith(APIKEY_PREFIX):
return BearerToken(
TokenType.API_KEY,
bearer_token.removeprefix(f'{apikey_prefix}-'),
bearer_token.removeprefix(APIKEY_PREFIX),
)
except ValueError:
return None