commit
e183c184ac
4
main.py
4
main.py
@ -8,6 +8,8 @@ from app.routers import finish
|
||||
|
||||
from app.storage import mongo
|
||||
|
||||
from fastapi import responses
|
||||
|
||||
|
||||
QUEUES_TOKEN = os.getenv('QUEUES_TOKEN')
|
||||
|
||||
@ -19,7 +21,7 @@ async def check_token(request: fastapi.Request, call_next):
|
||||
if QUEUES_TOKEN:
|
||||
token = request.headers.get('X-Queues-Token')
|
||||
if not token or token != QUEUES_TOKEN:
|
||||
return fastapi.JSONResponse(status_code=403, content={'message': 'token is not provided or incorrect'})
|
||||
return responses.JSONResponse(status_code=403, content={'message': 'token is not provided or incorrect'})
|
||||
return await call_next(request)
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user