Todo: 集成多平台 解决因SaiNiu线程抢占资源问题 本地提交测试环境打包 和 正式打包脚本与正式环境打包bat 提交Python32环境包 改进多日志文件生成情况修改打包日志细节
This commit is contained in:
173
Utils/PythonNew32/DOEM.py
Normal file
173
Utils/PythonNew32/DOEM.py
Normal file
@@ -0,0 +1,173 @@
|
||||
import ctypes
|
||||
import asyncio
|
||||
import websockets
|
||||
import hashlib
|
||||
import time
|
||||
import json
|
||||
|
||||
# 加载DLL文件
|
||||
sainiu_api = ctypes.CDLL('SaiNiuApi.dll')
|
||||
|
||||
# 定义函数参数类型和返回值类型
|
||||
sainiu_api.Access_ServerStart.argtypes = [
|
||||
ctypes.c_int,
|
||||
ctypes.c_bool,
|
||||
ctypes.c_bool,
|
||||
ctypes.c_bool,
|
||||
ctypes.c_bool,
|
||||
ctypes.c_int,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_bool,
|
||||
ctypes.c_bool
|
||||
]
|
||||
sainiu_api.Access_ServerStart.restype = ctypes.c_char_p # 返回值类型为文本型（字符串）
|
||||
|
||||
# 调用函数时传入的参数
|
||||
port = 3030 # WS以及HTTP的请求端口，默认3030
|
||||
web_socket = True # 是否开启WebSocket
|
||||
http_server = True # 是否开启HTTP请求
|
||||
remote = False # 是否开启远程请求
|
||||
log = True # 是否开启日志功能
|
||||
ws_max = 0 # 设置WS客户端最大连接数，0 = 不限制
|
||||
sign_key = b'111111' # 签名key，留空表示无需签名请求 调试时建议修改为空
|
||||
sha256 = True # sign签名方式，假 = md5，真 = hmac - sha256
|
||||
version_tip = True # 是否开启SDK版本与DLL不一致时的刷新提示
|
||||
|
||||
# 启动服务器
|
||||
result = sainiu_api.Access_ServerStart(
|
||||
port,
|
||||
web_socket,
|
||||
http_server,
|
||||
remote,
|
||||
log,
|
||||
ws_max,
|
||||
sign_key,
|
||||
sha256,
|
||||
version_tip
|
||||
)
|
||||
|
||||
print(f"Access_ServerStart 服务器启动: {result.decode('gbk')}")
|
||||
|
||||
# 计算sign值
|
||||
def sign_GetSign(signKey, post, time, sha256=False):
|
||||
# 拼接字符串
|
||||
data = post + str(time) + signKey
|
||||
|
||||
# 将字符串转换为字节集
|
||||
byte_data = data.encode('utf-8')
|
||||
|
||||
# 根据sha256参数选择哈希算法
|
||||
if sha256:
|
||||
hash_object = hashlib.sha256(byte_data)
|
||||
else:
|
||||
hash_object = hashlib.md5(byte_data)
|
||||
|
||||
# 获取十六进制表示并转换为小写
|
||||
sign = hash_object.hexdigest().lower()
|
||||
|
||||
return sign
|
||||
|
||||
async def connect_to_websocket():
|
||||
while True:
|
||||
try:
|
||||
# 获取当前的13位时间戳（毫秒级）
|
||||
current_time_ws = int(time.time() * 1000)
|
||||
|
||||
# 计算签名，参数post参数为"ws"不可修改，signKey为自定义密匙，sha256参数为True选择sha256算法,为False则选择md5算法，需要与Access_StartServer的参数一致
|
||||
ws_sign = sign_GetSign(sign_key.decode('utf-8'), "ws", current_time_ws, sha256)
|
||||
|
||||
# 构建URI
|
||||
uri = f"ws://127.0.0.1:{port}?time={current_time_ws}&sign={ws_sign}"
|
||||
|
||||
async with websockets.connect(uri) as websocket:
|
||||
print("已连接到ws服务器")
|
||||
|
||||
# 再次获取当前的13位时间戳（毫秒级）以避免签名超时
|
||||
current_time_json = int(time.time() * 1000)
|
||||
|
||||
# 计算JSON数据的签名，参数post参数为"init"
|
||||
json_sign = sign_GetSign(sign_key.decode('utf-8'), "Init", current_time_json, sha256)
|
||||
|
||||
# 发送JSON数据
|
||||
message = {
|
||||
"type": "Invoke_SaiNiu",
|
||||
"traceId": "Access_Init",
|
||||
"codeType": "Access",
|
||||
"post": "Init",
|
||||
"data": {
|
||||
"AccessType": 0,
|
||||
"AccessId": "AccessId", #替换成授权的AccessId,
|
||||
"AccessKey": "AccessKey", #替换成授权的AccessKey
|
||||
"ForceLogin": True
|
||||
},
|
||||
"time": current_time_json,
|
||||
"sign": json_sign
|
||||
}
|
||||
|
||||
await websocket.send(json.dumps(message))
|
||||
print(f"Sent message: {message}")
|
||||
|
||||
try:
|
||||
while True:
|
||||
# 接收来自服务器的消息
|
||||
response = await websocket.recv()
|
||||
print(f"接收来自服务器的消息: {response}")
|
||||
|
||||
# 解析响应消息
|
||||
response_data = json.loads(response)
|
||||
|
||||
# 判断traceId是否匹配
|
||||
if response_data.get("traceId") == "Access_Init":
|
||||
# 再次获取当前的13位时间戳（毫秒级）以避免签名超时
|
||||
current_time_sys_init = int(time.time() * 1000)
|
||||
|
||||
# 计算新的JSON数据的签名，参数post参数为"SysInit"
|
||||
sys_init_sign = sign_GetSign(sign_key.decode('utf-8'), "SysInit", current_time_sys_init, sha256)
|
||||
|
||||
# 发送新的JSON数据
|
||||
new_message = {
|
||||
"type": "Invoke_SaiNiu",
|
||||
"traceId": "0470bf9489729b2e8a2126a04ab3e272",#自定义返回的traceId
|
||||
"codeType": "Access",
|
||||
"post": "SysInit",
|
||||
"data": {
|
||||
"number": 20,
|
||||
"port": 3030,
|
||||
"token": True,
|
||||
"connect": True,
|
||||
"disconnect": True,
|
||||
"newMessage": True,
|
||||
"groupMessage": True,
|
||||
"notice": True,
|
||||
"event": True,
|
||||
"assistant": True
|
||||
},
|
||||
"time": current_time_sys_init,
|
||||
"sign": sys_init_sign
|
||||
}
|
||||
|
||||
await websocket.send(json.dumps(new_message))
|
||||
print(f"Sent message: {new_message}")
|
||||
except websockets.ConnectionClosed:
|
||||
print("服务器关闭连接")
|
||||
except Exception as e:
|
||||
print(f"出现错误：{e}")
|
||||
except websockets.InvalidURI as e:
|
||||
print(f"ws地址无效: {e} 请检查WebSocket地址")
|
||||
break
|
||||
except ConnectionRefusedError as e:
|
||||
print(f"连接被拒绝： {e} 服务器可能未运行或URI不正确。5秒后重试..")
|
||||
await asyncio.sleep(5)
|
||||
except OSError as e:
|
||||
print(f"操作系统错误: {e} 服务器可能无法访问。5秒后重试..")
|
||||
await asyncio.sleep(5)
|
||||
except Exception as e:
|
||||
print(f"发生意外错误：{e} Sign验证错误。5秒后重试..")
|
||||
await asyncio.sleep(5)
|
||||
|
||||
# 运行异步主函数
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(connect_to_websocket())
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user