import websocket

#从ws服务器接收到消息
def on_message(ws, message):  
    print(f"Received: {message}")  
        
def on_error(ws, error):  
    print(f"Websocket错误: {error}") 
        
def on_close(ws):  
    print("Websocket已关闭") 
        
def on_open(ws):  
    print("Websocket已连接成功") 
            
# 携带token连接websocket
headers = {
    "token": loginInfo['token']
}

wsServer = 'ws://xxx.xxx.xxx.xxx:3838'
            
ws = websocket.WebSocketApp(wsServer,  header=headers,
                            on_message = on_message,  
                            on_error = on_error,  
                            on_close = on_close)
ws.on_open = on_open  
ws.run_forever()

#发送消息给ws服务器
ws.send("message")