websocket链接python有很多封装好的库:websocket-client、websockets、aiowebsocket 这里用的websokets
此次接口要求:1、需要双向认证 2、wss协议 3、发送数据和接受数据都需要序列化和反序列化(probuff)
# encoding = utf-8
import asyncio
import pathlib
import ssl
import websockets
import base64
from pb import conn_ready_pb2
import uuid
import datetime,time
import json
from google.protobuf import json_format
# 服务端声明CLIENT_AUTH
ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
#Purpose.CLIENT_AUTH该值表示该上下文可用于对Web客户端进行身份验证(因此,它将用于创建服务器端套接字),carfile服务端证书
ssl_context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH,cafile="D://my_project//Ajcloud//ca.crt")
# certfile和keyfile参数为指定包含证书被用来识别连接的本地端可选的文件,客户端证书
ssl_context.load_cert_chain(certfile="D://my_project//Ajcloud//mgr.crt", keyfile="D://my_project//Ajcloud//mgr.key")
ssl_context.verify_mode = ssl.CERT_REQUIRED
async def hello():
uri = "wss://xxxxxxxxxx"
async with websockets.connect(
uri, ssl=ssl_context
) as websocket:
message = "xxxxx"
reg = conn_ready_pb2.ConnReadyRequest()
res = conn_ready_pb2.ConnReadyResponse()
reg.header.message_id = str(uuid.uuid1())
datetime_now = datetime.datetime.now()
date_stamp = str(int(time.mktime(datetime_now.timetuple())))
data_microsecond = str("%06d" % datetime_now.microsecond)[0:3]
date_stamp = date_stamp + data_microsecond
reg.header.message_ts = date_stamp
r = reg.SerializeToString()#序列化
b = base64.b64encode(r) # 将二进制转为base64编码
b2 = str(b, 'utf-8')
message = {"xxx": "xxx", "yyyy": b2}
await websocket.send(json.dumps(message))
print(f"> {message}")
greeting = await websocket.recv()
payload = json.loads(greeting)["payload"]
# base64转码成二进制
payload =base64.b64decode(payload)
res.ParseFromString(payload)#反序列化
json_result = json_format.MessageToJson(res) # pb转成json格式
print(json_result)
print(f"< {greeting}")
asyncio.get_event_loop().run_until_complete(hello())