From 46b0b49c928884fac858bfb5163b305aff491f0e Mon Sep 17 00:00:00 2001 From: emmatveev Date: Sun, 17 Nov 2024 02:32:25 +0300 Subject: [PATCH 1/2] up --- main.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index d704da5..f3b6552 100644 --- a/main.py +++ b/main.py @@ -1,7 +1,6 @@ import fastapi import uvicorn - -from app.middlewares import check_token +import os from app.routers import take from app.routers import put @@ -10,9 +9,19 @@ from app.routers import finish from app.storage import mongo +QUEUES_TOKEN = os.getenv('QUEUES_TOKEN') + app = fastapi.FastAPI() -app.add_middleware(check_token.CheckToken()) + +@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: + raise fastapi.HTTPException(403) + return await call_next(request) + app.include_router(take.router) app.include_router(put.router) From dc20588b1823c704c05ffdf383d3bce7d9779e4e Mon Sep 17 00:00:00 2001 From: emmatveev Date: Sun, 17 Nov 2024 02:32:45 +0300 Subject: [PATCH 2/2] up --- app/middlewares/__init__.py | 0 app/middlewares/check_token.py | 13 ------------- 2 files changed, 13 deletions(-) delete mode 100644 app/middlewares/__init__.py delete mode 100644 app/middlewares/check_token.py diff --git a/app/middlewares/__init__.py b/app/middlewares/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/app/middlewares/check_token.py b/app/middlewares/check_token.py deleted file mode 100644 index 3c017b6..0000000 --- a/app/middlewares/check_token.py +++ /dev/null @@ -1,13 +0,0 @@ -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)