python .websocket 客户端代码

下面是一个使用 Python 的 websocket 客户端代码示例：

import websocket

def on_message(ws, message):
    print(message)

def on_error(ws, error):
    print(error)

def on_close(ws):
    print("### closed ###")

def on_open(ws):
    print("### open ###")
    ws.send("Hello, Server!")

if __name__ == "__main__":
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp("ws://localhost:8000/",
                              on_message = on_message,
                              on_error = on_error,
                              on_close = on_close)
    ws.on_open = on_open
    ws.run_forever()

其中：

- `on_message` 是当客户端接收到消息时执行的回调函数。
- `on_error` 是当发生错误时执行的回调函数。
- `on_close` 是当连接关闭时执行的回调函数。
- `on_open` 是当连接建立后执行的回调函数。
- `websocket.enableTrace(True)` 可以开启日志输出，方便调试。

在 `main` 函数中，我们创建了一个 `WebSocketApp` 对象，并将上述回调函数传递给它。然后，我们调用 `run_forever` 方法开始运行客户端。在 `on_open` 回调函数中，我们可以向服务器发送消息，例如这里发送了一个简单的字符串消息 "Hello, Server!"。