From 2bd7f7ea81592c1783f0b5f6ab2f022138615a59 Mon Sep 17 00:00:00 2001 From: emmatveev Date: Sun, 17 Nov 2024 02:36:46 +0300 Subject: [PATCH] up --- main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 05768b2..8c17be6 100644 --- a/main.py +++ b/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)