用python的websocket包写一个简单的websocket客户端

当使用Python编写WebSocket客户端时，可以使用`websocket`库来处理WebSocket通信。下面是一个简单的示例代码，演示了如何使用`websocket`库创建一个WebSocket客户端：

import websocket

def on_message(ws, message):
    print(f"Received: {message}")

def on_error(ws, error):
    print(f"Error: {error}")

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

def on_open(ws):
    print("Connection established")
    ws.send("Hello, server!")

if __name__ == "__main__":
    # 使用websocket库创建WebSocket连接
    ws = websocket.WebSocketApp("ws://echo.websocket.org/",
                                on_message=on_message,
                                on_error=on_error,
                                on_close=on_close)
    ws.on_open = on_open

    # 启动WebSocket连接
    ws.run_forever()

在上面的示例中，首先定义了四个回调函数：`on_message`用于处理接收到的消息，`on_error`用于处理错误，`on_close`用于处理连接关闭，`on_open`用于在连接建立时发送一条消息。

然后，使用`websocket.WebSocketApp`创建一个WebSocket连接，并传入回调函数。在这个例子中，我们连接到了一个公共的WebSocket测试服务器`ws://echo.websocket.org/`。

最后，通过调用`ws.run_forever()`启动WebSocket连接，程序将一直运行直到连接关闭。

在实际应用中，你可以根据需要修改回调函数的实现，以满足特定的业务需求。