安装
pip install websocket-client
先来看一下，长连接调用方式：
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()
长连接，参数介绍：
（1）url: websocket的地址。
（2）header: 客户发送websocket握手请求的请求头，{'head1:value1','head2:value2'}。
（3）on_open：在建立Websocket握手时调用的可调用对象，这个方法只有一个参数，就是该类本身。
（4）on_message：这个对象在接收到服务器返回的消息时调用。有两个参数，一个是该类本身，一个是我们从服务器获取的字符串（utf-8格式）。
（5）on_error：这个对象在遇到错误时调用，有两个参数，第一个是该类本身，第二个是异常对象。
（6）on_close：在遇到连接关闭的情况时调用，参数只有一个，就是该类本身。
（7）on_cont_message：这个对象在接收到连续帧数据时被调用，有三个参数，分别是：类本身，从服务器接受的字符串（utf-8），连续标志。
（8）on_data：当从服务器接收到消息时被调用，有四个参数，分别是：该类本身，接收到的字符串（utf-8），数据类型，连续标志。
（9）keep_running：一个二进制的标志位，如果为True，这个app的主循环将持续运行，默认值为True。
（10）get_mask_key：用于产生一个掩码。
（11）subprotocols：一组可用的子协议，默认为空。
长连接关键方法：ws.run_forever(ping_interval=60,ping_timeout=5)
如果不断开关闭websocket连接，会一直阻塞下去。另外这个函数带两个参数，如果传的话，启动心跳包发送。
ping_interval:自动发送“ping”命令，每个指定的时间(秒),如果设置为0，则不会自动发送。
ping_timeout:如果没有收到pong消息，则为超时(秒)。
ws.run_forever(ping_interval=60,ping_timeout=5)#ping_interval心跳发送间隔时间#ping_timeout 设置，发送ping到收到pong的超时时间
我们看源代码，会发现这样一断代码：
ping的超时时间，要大于ping间隔时间
if not ping_timeout or ping_timeout <= 0:
ping_timeout = None
if ping_timeout and ping_interval and ping_interval <= ping_timeout:
raise WebSocketException("Ensure ping_interval > ping_timeout")
长连接：
示例1：
import websocket
try:
import thread
except ImportError:
import _thread as thread
import time
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("hello1")
time.sleep(1)
ws.close()
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(ping_interval=60,ping_timeout=5)
示例2：
import websocket
from threading import Thread
import time
import sys
class MyApp(websocket.WebSocketApp):
def on_message(self, message):
print(message)
def on_error(self, error):
print(error)
def on_close(self):
print("### closed ###")
def on_open(self):
def run(*args):
for i in range(3):
# send the message, then wait
# so thread doesn't exit and socket
# isn't closed
self.send("Hello %d" % i)
time.sleep(1)
time.sleep(1)
self.close()
print("Thread terminating...")
Thread(target=run).start()
if __name__ == "__main__":
websocket.enableTrace(True)
if len(sys.argv) < 2:
host = "ws://echo.websocket.org/"
else:
host = sys.argv[1]
ws = MyApp(host)
ws.run_forever()
短连接：
from websocket import create_connection
ws = create_connection("ws://echo.websocket.org/")
print("Sending 'Hello, World'...")
ws.send("Hello, World")
print("Sent")
print("Receiving...")
result = ws.recv()
print("Received '%s'" % result)
ws.close()
以上就是Python WebSocket长连接心跳与短连接的示例的详细内容，更多关于Python WebSocket连接的资料请关注易盾网络其它相关文章！