websocket-client
websocket-client 是 websocket 客户端,提供了对ws低级API的访问。通过导入 websocket 库使用,websocket 库是基于事件驱动的设计模式,通过定义回调函数来处理接收到的消息、错误和连接关闭等事件。
优势:
- 兼容多个 Python 版本,包括 Python 2.7 和 Python 3.x。
- 简单易用,入门门槛较低。
- 提供了基本的 WebSocket 功能,可以满足一般需求。
劣势:
- 功能相对较少,不支持一些高级特性,如异步操作和性能优化。
"""用 websocket 创建长连接"""
import timeimport websocket
from gevent import threaddef on_message(ws, message):# 处理收到的消息print("Received: " + message)def on_error(ws, error):# 处理错误print("Error: " + str(error))def on_close(ws):# 关闭连接print("Connection closed")def on_open(ws):# 连接成功后的操作def run(*args):# 发送心跳包或其他持续性操作while True:ws.send("Ping")time.sleep(1) # 每隔一秒发送一次thread.start_new_thread(run, ())if __name__ == "__main__":# 创建 WebSocket 连接websocket.enableTrace(True)ws = websocket.WebSocketApp('ws://your-websocket-url',on_message=on_message,on_error=on_error,on_close=on_close)ws.on_open = on_open# 运行 WebSocket 客户端ws.run_forever()
websockets
websockets 具有 server端和 client端,采用异步操作模式,与 asyncio 模块无缝集成,可以实现高性能的 WebSocket 服务器和客户端。
python版本仅支持3.6及更高版本
优势:
- 高性能和可扩展性
- 提供了丰富的特性:支持心跳包、自定义协议和 SSL/TLS 加密等高级功能。
劣势:
- 不兼容 Python 2.x 版本。
"""服务端"""
import asyncio
import websocketsasync def handle(ws, path):name = await ws.recv()print(f"接收: {name}")greeting = f"已收到 {name}!"await ws.send(greeting)print(f"发送: {greeting}")if __name__ == '__main__':s = websockets.serve(handle, "127.0.0.1", 9451)event_loop = asyncio.get_event_loop()event_loop.run_until_complete(s)event_loop.run_forever()
"""客户端"""
import asyncio
import websocketsasync def hello():uri = "ws://127.0.0.1:9451"async with websockets.connect(uri) as websocket:name = input("发送: ")await websocket.send(name)greeting = await websocket.recv()print(f"接收: {greeting}")if __name__ == '__main__':event_loop = asyncio.get_event_loop()event_loop.run_until_complete(hello())