14 lines
377 B
Python
14 lines
377 B
Python
import fastapi
|
|
import os
|
|
|
|
|
|
QUEUES_TOKEN = os.getenv('QUEUES_TOKEN')
|
|
|
|
class CheckToken:
|
|
async def __call__(self, request: fastapi.Request, call_next):
|
|
if QUEUES_TOKEN:
|
|
token = request.headers.get('X-Queues-Token')
|
|
if not token or token != QUEUES_TOKEN:
|
|
raise fastapi.HTTPException(403)
|
|
return await call_next(request)
|