python 的websocket的使用
时间: 2023-06-21 21:07:16 浏览: 47
Python 提供了很多 WebSocket 模块,其中比较常用的是 `websocket` 和 `websocket-client` 模块。
下面是一个简单的 WebSocket 客户端的示例代码:
```python
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):
def run(*args):
ws.send("Hello, World!")
ws.close()
print("Thread terminating...")
thread.start_new_thread(run, ())
if __name__ == "__main__":
websocket.enableTrace(True)
ws = websocket.WebSocketApp("ws://echo.websocket.org/",
on_message = on_message,
on_error = on_error,
on_close = on_close)
ws.on_open = on_open
ws.run_forever()
```
通过 `websocket.WebSocketApp()` 方法创建一个 WebSocket 对象,然后使用 `ws.on_open` 方法来注册一个回调函数,当 WebSocket 连接成功后执行该函数。在回调函数中,可以使用 `ws.send()` 方法向服务器发送消息,使用 `ws.close()` 方法关闭连接。
你可以使用上述代码作为模板,将 `on_message()`, `on_error()`, `on_close()` 和 `on_open()` 函数替换为你自己的处理逻辑。