20 lines
638 B
Python
20 lines
638 B
Python
from fastapi import Request, status
|
|
from fastapi.responses import JSONResponse
|
|
from config import api_config
|
|
|
|
from endpoints.routes import get_safe_endpoint_urls
|
|
|
|
|
|
async def token_middleware(request: Request, call_next):
|
|
base_url = request.url.path
|
|
safe_endpoints = [_[0] for _ in get_safe_endpoint_urls()]
|
|
if base_url in safe_endpoints:
|
|
return await call_next(request)
|
|
|
|
token = request.headers.get(api_config.ACCESS_TOKEN_TAG, None)
|
|
if not token:
|
|
return JSONResponse(content={"error": "EYS_0002"}, status_code=status.HTTP_401_UNAUTHORIZED)
|
|
|
|
response = await call_next(request)
|
|
return response
|