py-test/test_websocket.py

29 lines
712 B
Python

import websocket
# WebSocket 地址
url = "ws://192.168.1.41:9516/ws/" # 或 wss://192.168.1.41:9516/ws/
def on_message(ws, message):
print("收到消息:", message)
def on_error(ws, error):
print("发生错误:", error)
def on_close(ws, close_status_code, close_msg):
print("连接关闭")
def on_open(ws):
print("连接成功")
# 连接成功后发一条测试消息
ws.send("hello server")
if __name__ == "__main__":
websocket.enableTrace(True) # 打开调试日志,看握手过程
ws_app = websocket.WebSocketApp(
url,
on_message=on_message,
on_error=on_error,
on_close=on_close,
on_open=on_open
)
ws_app.run_forever()