This commit is contained in:
2024-11-17 01:53:10 +03:00
parent f6b300fc2c
commit ef3c8c8006
7 changed files with 22 additions and 1 deletions

View File

View File

@@ -0,0 +1,13 @@
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)