From 64acc2bcd5e531f02be2da919e97b7af6bc86e4a Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 27 二月 2024 16:52:21 +0800
Subject: [PATCH] zeromq集成测试

---
 third_data/kpl_block_util.py |   17 +++++++++--------
 1 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/third_data/kpl_block_util.py b/third_data/kpl_block_util.py
index 5d83433..3972d15 100644
--- a/third_data/kpl_block_util.py
+++ b/third_data/kpl_block_util.py
@@ -18,10 +18,11 @@
     time_str = datetime.datetime.now().strftime("%Y-%m-%d") + " 09:30:00"
     timestamp = time.mktime(time.strptime(time_str, '%Y-%m-%d %H:%M:%S'))
     limit_up_codes = set()
-    for k in limit_up_record_datas:
-        if code_block_dict.get(k[3]) == block:
-            if int(k[5]) < timestamp:
-                limit_up_codes.add(k[3])
+    if limit_up_record_datas:
+        for k in limit_up_record_datas:
+            if code_block_dict.get(k[3]) == block:
+                if int(k[5]) < timestamp:
+                    limit_up_codes.add(k[3])
     return limit_up_codes
 
 
@@ -167,10 +168,10 @@
 
 # 鑾峰彇褰撴棩瀹炴椂韬綅
 # before_blocks_dict鏍煎紡浣峽"浠g爜":set("鏉垮潡")}
-def get_code_current_rank(code, block, current_limit_up_datas, code_limit_up_reason_dict,
-                          yesterday_current_limit_up_codes, exclude_codes, open_limit_up_count, shsz=False):
+def get_code_current_rank(code, block, current_limit_up_datas, code_limit_up_reasons_dict,
+                          yesterday_current_limit_up_codes, exclude_codes, open_limit_up_count, shsz=False,
+                          limit_up_time=time.time()):
     block_codes_infos = []
-    limit_up_time = time.time()
     for k in current_limit_up_datas:
         if k[0] == code:
             # 鑾峰彇褰撳墠浠g爜娑ㄥ仠鏃堕棿
@@ -180,7 +181,7 @@
         # 鍓旈櫎楂樹綅鏉�
         if k[0] in yesterday_current_limit_up_codes:
             continue
-        if code_limit_up_reason_dict.get(k[0]) == block:
+        if code_limit_up_reasons_dict.get(k[0]) and block in code_limit_up_reasons_dict.get(k[0]):
             if k[0] != code:
                 # 浠g爜.娑ㄥ仠鏃堕棿
                 block_codes_infos.append((k[0], int(k[2])))

--
Gitblit v1.8.0