python3发送需要双向认证的wss请求

3,862 阅读1分钟

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())