From 9d9323c582bea82196b6813fa7331bea8494b5ea Mon Sep 17 00:00:00 2001
From: admin <admin@example.com>
Date: 星期四, 16 一月 2025 17:23:42 +0800
Subject: [PATCH] 功能完善
---
trade/middle_api_protocol.py | 17 ++++++++++++++++-
1 files changed, 16 insertions(+), 1 deletions(-)
diff --git a/trade/middle_api_protocol.py b/trade/middle_api_protocol.py
index 3c1e135..8a0298f 100644
--- a/trade/middle_api_protocol.py
+++ b/trade/middle_api_protocol.py
@@ -7,11 +7,26 @@
SERVER_HOST = '43.138.167.68'
SERVER_PORT = 10008
+SERVER_PUSH_PORT = 10009
def request(data_json, host=SERVER_HOST, port=SERVER_PORT):
if type(data_json) == set:
data_json = list(data_json)
+ data_bytes = socket_util.load_header(json.dumps(data_json).encode('utf-8'))
+ sk = socket_util.create_socket(host, port)
+ try:
+ sk.sendall(data_bytes)
+ result_str, header_str = socket_util.recv_data(sk)
+ result_json = json.loads(result_str)
+ if result_json["code"] != 0:
+ raise Exception(result_json["msg"])
+ return result_json.get("data")
+ finally:
+ sk.close()
+
+
+def push(data_json, host=SERVER_HOST, port=SERVER_PUSH_PORT):
data_bytes = socket_util.load_header(json.dumps(data_json).encode('utf-8'))
sk = socket_util.create_socket(host, port)
try:
@@ -69,7 +84,7 @@
# ------------------------------娑堟伅鎺ㄩ��------------------------------------
def load_push_msg(data):
- fdata = {"type": "push_msg", "data": {"ctype": "push_msg", "data": data}}
+ fdata = {"type": "push_msg", "data": data}
return fdata
--
Gitblit v1.8.0