最近做自动化需要接收websocket消息来进行判断,做个总结:
使用websoket-client库,其git主页:https://github.com/websocket-client/websocket-client
pip install websocket_client
代码在下面,注意:cookie要发送http登录请求去获取然后传给这边的接口
from websocket import create_connection
import json
import time
url = "ws://127.0.0.1:8080/test"
header = ['Cookie: shiro.session=ffdsftedd-d234-2233;Path=/;HttpOnly, remem……']
ws = create_connection(url, header=header)
ws.send('login') #发送握手消息
end_time = time.time() + 10
tmp_list = []
while time.time() < end_time: #接收10s数据
mes = ws.recv()
print(mes)
if json.loads(mes)["key"] == 'target': #拿自己想要的数据
tmp_list.append(json.loads(mes))
遇到的问题:
-
连接报错,如下图:WinError 10042 在getsockopt 或 setsockopt调用中指定一个未知的,无效的或不受支持的选项或层次
这个是因为socket设置了一个默认的配置,而这个配置在windows某些版本不支持(我用的是win7不支持,其它不清楚)
解决办法:
找到site-packages\websocket\_socket.py文件,搜索TCP_KEEPCNT关键字,注释相关代码,如下图
-
如何发送cookie
开始我只是把cookie值填进去,用requests登录接口获取cookie发现送的不止是cookie值,还有其它,所以最好是发送登录接口获取cookie然后再赋给websocket。