From 7759cdf64d4777e1158a327c695ff7cde70ebe68 Mon Sep 17 00:00:00 2001 From: urec56 Date: Mon, 3 Jun 2024 18:20:44 +0500 Subject: [PATCH] =?UTF-8?q?=D0=98=D0=B7=D0=BC=D0=B5=D0=BD=D0=B8=D0=BB=20?= =?UTF-8?q?=D0=B2=D0=B5=D0=B1=D1=81=D0=BE=D0=BA=D0=B5=D1=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/users/dependencies.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/app/users/dependencies.py b/app/users/dependencies.py index d36f27f..e7338c9 100644 --- a/app/users/dependencies.py +++ b/app/users/dependencies.py @@ -1,4 +1,6 @@ -from fastapi import Depends, WebSocket +from typing import Annotated + +from fastapi import Depends, Header from fastapi.security import HTTPBearer from jose import JWTError, jwt, ExpiredSignatureError @@ -49,11 +51,10 @@ async def check_verificated_user_with_exc(user: SUser = Depends(get_current_user return user -def get_token_ws(websocket: WebSocket) -> str: - token = websocket.cookies.get("black_phoenix_access_token") - if not token: +def get_token_ws(authorization: Annotated[str | None, Header()] = None) -> str: + if authorization in None: raise TokenAbsentException - return token + return authorization.split()[1] async def get_current_user_ws(token: str = Depends(get_token_ws), uow=Depends(UnitOfWork)):