1
0
mirror of https://github.com/ijaric/voice_assistant.git synced 2025-05-24 14:33:26 +00:00
voice_assistant/src/fastapi_app/lib/api/v1/services/token.py
2023-09-20 10:43:29 +01:00

27 lines
844 B
Python

import fastapi
from jose import JWTError, jwt
from pydantic import ValidationError
from lib.api.v1 import schemas as app_schemas
from lib.app import settings as app_settings
app = fastapi.FastAPI()
settings = app_settings.get_settings()
security = fastapi.security.HTTPBearer()
def get_token_data(
authorization: fastapi.security.HTTPAuthorizationCredentials = fastapi.Security(security),
) -> app_schemas.entity.Token:
token = authorization.credentials
try:
secret_key = settings.jwt_secret_key
payload = jwt.decode(token, secret_key, algorithms=["HS256"])
return app_schemas.entity.Token(**payload)
except (JWTError, ValidationError):
raise fastapi.HTTPException(
status_code=fastapi.status.HTTP_401_UNAUTHORIZED,
detail="Could not validate credentials",
)