一、安装websocket-client
1、安装
pip install websocket-client
二、方法
1、我们先看一下,长连接调用方法:
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()
2、长连接,参数介绍:
(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的超时时间
长连接方法:
import websocket
import json
import time
try:
import thread
except ImportError:
import _thread as thread
def on_message(ws, message):
'''服务器有数据更新时,主动推送过来的数据'''
print(message)
def on_error(ws, error):
'''程序报错时,就会触发on_error事件'''
print(error)
def on_close(ws):
'''关闭websocket连接后,打印此消息'''
print("### closed ###")
def on_open(ws):
'''连接到服务器之后就会触发on_open事件,这里用于send数据'''
def run(*args):
Sign = {
"name": 1,
"nam2": 2
}
ws.send(json.dumps(Sign)) # 发送数据(必须为str类型)
time.sleep(1)
ws.close() # 关闭连接
thread.start_new_thread(run, ())
def run_1():
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_timeout=5)
# ws.run_forever()
if __name__ == "__main__":
run_1()
短连接方法(使用create_connection链接,此方法不建议使用,链接不稳定,容易断,并且连接很耗时):
import time
import json
from websocket import create_connection
class webcket:
def __init__(self):
self.url = "ws://echo.websocket.org/"
def connect(self):
'''一直链接,直到连接上就退出循环'''
while True:
try:
self.ws = create_connection(self.url)
print(self.ws)
break
except Exception as e:
print('连接异常:', e)
continue
def send_data(self):
'''连接成功后,发送数据'''
Sign = {
"name": 1,
"nam2": 2
}
self.ws.send(json.dumps(Sign)) # 发送数据(必须为str类型)
def run(self):
'''执行,然后循环获取服务器返回的数据'''
self.connect()
self.send_data()
time.sleep(1)
while True:
response = self.ws.recv()
print(f"结果:{response}")
if __name__ == '__main__':
webcket().run()