self/queues
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request 'up' (#5) from master into dev

Browse Source 
Reviewed-on: #5
This commit is contained in:
emmatveev 2024-11-17 02:37:03 +03:00
commit e183c184ac
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -8,6 +8,8 @@ from app.routers import finish
from app.storage import mongo from app.storage import mongo
from fastapi import responses
QUEUES_TOKEN = os.getenv('QUEUES_TOKEN') QUEUES_TOKEN = os.getenv('QUEUES_TOKEN')
@ -19,7 +21,7 @@ async def check_token(request: fastapi.Request, call_next):
if QUEUES_TOKEN: if QUEUES_TOKEN:
token = request.headers.get('X-Queues-Token') token = request.headers.get('X-Queues-Token')
if not token or token != 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) return await call_next(request)