python websocket客户端 工具类

以下是一个简单的 Python WebSocket 客户端工具类的示例代码：

import websocket

class WebSocketClient:
    def __init__(self, url):
        self.url = url
        self.ws = None
    
    def on_open(self):
        print("WebSocket connection opened.")
    
    def on_message(self, message):
        print(f"Received message: {message}")
    
    def on_close(self):
        print("WebSocket connection closed.")
    
    def on_error(self, error):
        print(f"Error: {error}")
    
    def connect(self):
        self.ws = websocket.WebSocketApp(self.url,
                                         on_open=self.on_open,
                                         on_message=self.on_message,
                                         on_close=self.on_close,
                                         on_error=self.on_error)
        self.ws.run_forever()

    def send(self, message):
        if self.ws:
            self.ws.send(message)

你可以通过以下方式使用 `WebSocketClient` 类：

ws = WebSocketClient("ws://example.com:8000")
ws.connect()

此时，`WebSocketClient` 将连接到指定的 WebSocket URL 并等待接收消息。如果你想发送消息，可以使用 `send` 方法：

ws.send("Hello, WebSocket!")

注意：要使用此示例代码，你需要安装 `websocket` 模块。你可以使用以下命令安装：

pip install websocket-client