routers/websockets_new.py

from fastapi import APIRouter , Request ,HTTPException , WebSocket
from controllers import websocket_controller as wc
from controllers import ws_controller as w
import logging
import aiohttp

router = APIRouter(prefix="/websockets")

@router.websocket("/ws")
async def get_data(websocket:WebSocket):
    await websocket.accept()
    
    json = await websocket.receive_json()
    access_token = json['access_token']
    brand_name = json['brand_name'] if json.get('brand_name') is not None else None
    if access_token is None:
        await websocket.send_text("Access Token Invalid OR NULL !!!")
        websocket.close()
    # access_token = await websocket.receive_text()
    # brand_name = "test"
    logging.info(f"brand_name: f{brand_name}")
    logging.info(f"access_token : {access_token}")
    await w.websocket_main(access_token ,websocket,brand_name)