From 48fb7a00951f91bdc707e5dd2d196e5bccb752c3 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期三, 18 六月 2025 18:41:30 +0800
Subject: [PATCH] 异常保护

---
 utils/data_export_util.py |   78 ++++++++++++++++++++++++++++++++++++---
 1 files changed, 72 insertions(+), 6 deletions(-)

diff --git a/utils/data_export_util.py b/utils/data_export_util.py
index 6c3a590..4de83e7 100644
--- a/utils/data_export_util.py
+++ b/utils/data_export_util.py
@@ -1,6 +1,7 @@
 """
 鏁版嵁瀵煎嚭宸ュ叿
 """
+import copy
 import json
 import logging
 import os
@@ -12,10 +13,12 @@
 from code_attribute import gpcode_manager
 import l2.l2_data_util
 from l2.huaxin import l2_huaxin_util
+from l2.l2_transaction_data_manager import BigOrderDealManager
 from log_module import log, log_export
-from l2 import l2_data_source_util
+from l2 import l2_data_source_util, l2_data_util
 from log_module.log import logger_debug
 from trade import deal_big_money_manager
+from trade.buy_money_count_setting import BuyMoneyUtil
 from utils import tool
 
 # 缂撳瓨L2鏁版嵁锛屾牸寮忥細{"鏃ユ湡":{鏁版嵁}}
@@ -29,7 +32,7 @@
 
 
 # 鑾峰彇L2鐨勬暟鎹�
-def get_l2_datas(code, today_datas=None, date=None, max_time=None):
+def get_l2_datas(code, today_datas=None, date=None, end_index=None):
     __start_time = time.time()
     if date is None:
         date = tool.get_now_date_str()
@@ -49,11 +52,13 @@
         trade_indexs = []
         real_position_indexes = []
         deal_list = []
+        cancel_reasons = {}
     else:
         process_indexs = log_export.get_l2_process_position(code, date)
         trade_indexs = log_export.get_l2_trade_position(code, date)
         real_position_indexes = log_export.get_real_place_order_positions(code, date)
         deal_list = log_export.load_huaxin_deal_record(code, date)
+        cancel_reasons = log_export.load_cancel_buy_reasons(code, date)
     deal_list_dict = {}
     for d in deal_list:
         deal_list_dict[str(d[0])] = d
@@ -64,8 +69,22 @@
     active_sell_set = active_sell_map.get(code)
     if not active_sell_set:
         active_sell_set = set()
+    # 濡傛灉鏈夋埅鑷充綅缃�
+    if end_index:
+        datas = copy.deepcopy(datas)
+        for i in range(len(datas)):
+            if datas[i]["index"] == end_index:
+                datas = datas[:i]
+                break
+    # 鎴愪氦鐨勫ぇ鍗曚俊鎭痆(涔板崟鍙�,鎬昏偂鏁�,鎬绘垚浜ら,鎴愪氦寮�濮嬫椂闂�,鎴愪氦缁撴潫鏃堕棿)]
+    deal_big_buy_order_list = copy.deepcopy(BigOrderDealManager().get_total_buy_data_list(code))
+    deal_big_buy_order_no_dict = {}
+    if deal_big_buy_order_list:
+        for d in deal_big_buy_order_list:
+            deal_big_buy_order_no_dict[d[0]] = d
+
     fdatas = export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes, deal_list_dict, sell_nos,
-                            active_sell_set)
+                            active_sell_set, cancel_reasons, deal_big_buy_order_no_dict)
     return fdatas
 
 
@@ -79,7 +98,7 @@
 
 
 def export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes, deal_list_dict, sell_nos,
-                   active_sell_nos):
+                   active_sell_nos, cancel_reasons, deal_big_buy_order_no_dict):
     def find_process_index(index):
         for i in range(0, len(process_indexs)):
             if process_indexs[i][0] <= index <= process_indexs[i][1]:
@@ -91,6 +110,11 @@
             if trade_indexs[i][1] == index:
                 return trade_indexs[i]
         return None
+
+    # 鏈�鏂扮殑涓�鏉℃暟鎹�
+    latest_data = datas[-1]
+    latest_time_str_with_ms = l2.l2_data_util.L2DataUtil.get_time_with_ms(latest_data["val"])
+
     # 鏁版嵁棰勫鐞�
     num_operate_map = {}
     l2.l2_data_util.load_num_operate_map(num_operate_map, code, datas)
@@ -109,16 +133,31 @@
     fdatas = []
     # 鏁版嵁澶灏遍渶瑕佽繃婊ゆ帀灏忛噾棰�
     is_data_too_large = len(datas) > 20000
+
+    limit_up_price = gpcode_manager.get_limit_up_price_as_num(code)
+    if limit_up_price:
+        # 闇�瑕佽闃呯殑鐗规畩鐨勯噺
+        special_volumes = BuyMoneyUtil.get_possible_buy_volumes(limit_up_price)
+        special_volumes = [x//100 for x in special_volumes]
+    else:
+        special_volumes = []
+
     for data in datas:
         index += 1
         if is_data_too_large and data['val']['num'] * float(data['val']['price']) < 5000:
-            if data["index"] not in real_position_indexes:
+            # 灏忓崟
+            if data["index"] not in real_position_indexes and data['val']['num'] not in special_volumes:
                 continue
 
         # 鍏堢Щ闄�
         if data['val']['orderNo'] in active_sell_nos:
             # 杩囨护涓诲姩鍗�
             continue
+
+        # 绉婚櫎鎴愪氦澶у崟鍦↙2涓凡鏈夌殑璁㈠崟
+        if l2.l2_data_util.L2DataUtil.is_limit_up_price_buy(data["val"]):
+            if int(data['val']['orderNo']) in deal_big_buy_order_no_dict:
+                deal_big_buy_order_no_dict.pop(int(data['val']['orderNo']))
 
         trade_info = find_trade_index(data["index"])
         if not trade_info:
@@ -197,7 +236,8 @@
         if int(data["val"]["operateType"]) == 0:
             cancel_data = l2.l2_data_util.local_today_canceled_buyno_map.get(code).get(str(data["val"]["orderNo"]))
             # 涔�
-            if cancel_data:
+            if cancel_data and latest_data["index"] >= cancel_data["index"]:
+                # 濡傛灉鎾ゅ崟绱㈠紩涓嶈兘姣旀渶杩戞暟鎹储寮曡繕澶�
                 try:
                     left_num = data["val"]["num"] - cancel_data["val"]["num"]
                     if left_num > 0:
@@ -210,8 +250,11 @@
             else:
                 deal_info = deal_list_dict.get(str(data["val"].get("orderNo")))
                 if deal_info and len(deal_info) >= 5:
+                    # 鎴愪氦鏃堕棿涓嶈兘姣斿綋鍓嶇储寮曟椂闂村ぇ
                     cancel_info = l2_huaxin_util.convert_time(deal_info[4],
                                                               with_ms=True)
+                    if tool.trade_time_sub_with_ms(cancel_info, latest_time_str_with_ms) > 0:
+                        cancel_info = None
         format_data.append(cancel_info)
         cancel_order_info = None
         if trade_info:
@@ -224,9 +267,32 @@
             elif trade_info[0] == 2:
                 # font.colour_index = 10
                 cancel_order_info = trade_info[2]
+        if not cancel_order_info:
+            if data["index"] in cancel_reasons:
+                cancel_order_info = cancel_reasons[data["index"]]
+
         format_data.append(cancel_order_info)
         format_data.append(data["val"].get("orderNo"))
         fdatas.append((style_int, trade_info, format_data))
+
+    if tool.is_sh_code(code) and deal_big_buy_order_no_dict:
+        # 涓婅瘉锛屾湁涓诲姩鎴愪氦鐨勫ぇ鍗�
+        active_deal_list = [deal_big_buy_order_no_dict[order_no] for order_no in deal_big_buy_order_no_dict]
+        active_deal_list.sort(key=lambda x: x[3])
+        for i in range(len(active_deal_list)):
+            d = active_deal_list[i]
+            format_data = [0]
+            format_data.append(l2_huaxin_util.convert_time(d[3], with_ms=True))
+            format_data.append('')
+            format_data.append(f"{round(d[2] / 10000, 1)}涓�")
+            format_data.append(round(d[2] / d[1], 2))
+            format_data.append(d[1] // 100)
+            format_data.append('涓诲姩涔�')
+            format_data.append(1)
+            format_data.append(l2_huaxin_util.convert_time(d[4], with_ms=True))
+            format_data.append('')
+            format_data.append(d[0])
+            fdatas.insert(i, (0, None, format_data))
     logger_debug.info("寰幆缁勮鏁版嵁瀹屾垚")
     return fdatas
 

--
Gitblit v1.8.0