Изменение вебсокета

app/chat/websocket.py

@@ -1,5 +1,7 @@
+from typing import Annotated
+
 import websockets
-from fastapi import WebSocket, WebSocketDisconnect, Depends
+from fastapi import WebSocket, WebSocketDisconnect, Depends, Header
 
 from app.exceptions import IncorrectDataException, UserDontHavePermissionException
 from app.services.message_service import MessageService
@@ -133,13 +135,14 @@ manager = ConnectionManager()
 async def websocket_endpoint(
 	chat_id: int,
 	websocket: WebSocket,
+	authorization: Annotated[str | None, Header()] = None,
 	# user: SUser = Depends(get_current_user_ws),
 	uow=Depends(UnitOfWork)
 ):
 	# await AuthService.check_verificated_user_with_exc(uow=uow, user_id=user.id)
 	# await AuthService.validate_user_access_to_chat(uow=uow, user_id=user.id, chat_id=chat_id)
 	await manager.connect(chat_id, websocket)
-	await websocket.send_json({"headers": str(websocket.headers)})
+	await websocket.send_json({"headers": str(websocket.headers), "authorization": authorization})
 	try:
 		while True:
 			data = await websocket.receive_json()

app/users/dependencies.py

@@ -50,7 +50,7 @@ async def check_verificated_user_with_exc(user: SUser = Depends(get_current_user
 	return user
 
 
-def get_token_ws(websocket: WebSocket, authorization: Annotated[str | None, Header()] = None) -> str:
+def get_token_ws(authorization: Annotated[str | None, Header()] = None) -> str:
 	if authorization is None:
 		raise TokenAbsentException
 	return authorization.split()[1]