fix
All checks were successful
Deploy Dev / Build (pull_request) Successful in 5s
Deploy Dev / Push (pull_request) Successful in 7s
Deploy Dev / Deploy dev (pull_request) Successful in 7s
Deploy Prod / Build (pull_request) Successful in 5s
Deploy Prod / Push (pull_request) Successful in 7s
Deploy Prod / Deploy prod (pull_request) Successful in 9s

This commit is contained in:
2024-11-22 01:00:40 +03:00
parent 6d6622b825
commit 4fb1043a63
3 changed files with 35 additions and 16 deletions

16
main.py
View File

@@ -8,23 +8,9 @@ from app.routers import finish
from app.storage import mongo
from fastapi import responses
QUEUES_TOKEN = os.getenv('QUEUES_TOKEN')
app = fastapi.FastAPI()
@app.middleware('http')
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 responses.JSONResponse(status_code=403, content={'message': 'token is not provided or incorrect'})
return await call_next(request)
app.include_router(take.router)
app.include_router(put.router)
app.include_router(finish.router)
@@ -33,4 +19,6 @@ mongo.create_indexes()
if __name__ == '__main__':
if os.getenv('NEW') == 'true':
uvicorn.run(app, host="0.0.0.0", port=1239)
uvicorn.run(app, host="0.0.0.0", port=1238)