Python实现websocket Client设置headers

代码逻辑

import asyncio  #python 异步IO标准库 
from websockets.sync.client import connect # 需安装websockets库

# 部分信息需要通过URL传递
wss_uri = "wss://{服务域名}:{portal端口}/{uri}"
# 验证信息通过请求头传递
headers = {
    "Upgrade": "websocket",
    "Connection": "Upgrade",
    "Sec-WebSocket-Version": "13",
    "Sec-WebSocket-Key": "base64-encoded-secret-key",
    "csrf-token": "xxxx",
    "cookie": "xxxx"
}
# 向服务端发送消息
async def receive_message(websocket): 
    message = websocket.recv()
    print("Received message:", message)

# 接收服务器的响应并打印输出
async def send_message(websocket, message):
    websocket.send(message)
    print("send message success:")

# 主函数，实现简单客户端websocket 部分逻辑
async def client_websocket():
    with connect(wss_uri, additional_headers=headers) as websocket:
        await send_message(websocket, "Hello World")
        while True:
            await receive_message(websocket)  

asyncio.run(client_websocket())