From ffc0ded48d5118f20c274ccaf65819934636b0c5 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期五, 04 七月 2025 13:34:04 +0800
Subject: [PATCH] 接口调整(L2数据急速模式/L后手动撤单率设置)

---
 utils/data_export_util.py |   59 +++++++++++++++++++++++++++++++++++++++++------------------
 1 files changed, 41 insertions(+), 18 deletions(-)

diff --git a/utils/data_export_util.py b/utils/data_export_util.py
index 02e93e9..1b0701b 100644
--- a/utils/data_export_util.py
+++ b/utils/data_export_util.py
@@ -33,7 +33,16 @@
 
 
 # 鑾峰彇L2鐨勬暟鎹�
-def get_l2_datas(code, today_datas=None, date=None, end_index=None):
+def get_l2_datas(code, today_datas=None, date=None, end_index=None, fast_mode=False):
+    """
+
+    @param code:
+    @param today_datas:
+    @param date:
+    @param end_index:
+    @param fast_mode: 鏄惁鏄�ラ�熸ā寮�
+    @return:
+    """
     __start_time = time.time()
     if date is None:
         date = tool.get_now_date_str()
@@ -57,22 +66,37 @@
         sell_no_dict = {}
         active_sell_map = {}
     else:
-        tasks = [dask.delayed(log_export.get_l2_process_position)(code, date),
-                 dask.delayed(log_export.get_l2_trade_position)(code, date),
-                 dask.delayed(log_export.get_real_place_order_positions)(code, date),
-                 dask.delayed(log_export.load_huaxin_deal_record)(code, date),
-                 dask.delayed(log_export.load_cancel_buy_reasons)(code, date),
-                 dask.delayed(log_export.load_huaxin_transaction_sell_no)(code, date),
-                 dask.delayed(log_export.load_huaxin_active_sell_map)(date),
-                 ]
-        results = dask.delayed(tasks).compute()
-        process_indexs = results[0]
-        trade_indexs = results[1]
-        real_position_indexes = results[2]
-        deal_list = results[3]
-        cancel_reasons = results[4]
-        sell_no_dict = results[5]
-        active_sell_map = results[6]
+        if not fast_mode:
+            tasks = [dask.delayed(log_export.get_l2_process_position)(code, date),
+                     dask.delayed(log_export.get_l2_trade_position)(code, date),
+                     dask.delayed(log_export.get_real_place_order_positions)(code, date),
+                     dask.delayed(log_export.load_huaxin_deal_record)(code, date),
+                     dask.delayed(log_export.load_cancel_buy_reasons)(code, date),
+                     dask.delayed(log_export.load_huaxin_transaction_sell_no)(code, date),
+                     dask.delayed(log_export.load_huaxin_active_sell_map)(date),
+                     ]
+            results = dask.delayed(tasks).compute()
+            process_indexs = results[0]
+            trade_indexs = results[1]
+            real_position_indexes = results[2]
+            deal_list = results[3]
+            cancel_reasons = results[4]
+            sell_no_dict = results[5]
+            active_sell_map = results[6]
+        else:
+            process_indexs = []
+            trade_indexs = []
+            deal_list = []
+            sell_no_dict = {}
+            active_sell_map = {}
+            tasks = [dask.delayed(log_export.get_real_place_order_positions)(code, date),
+                     dask.delayed(log_export.load_cancel_buy_reasons)(code, date),
+                     ]
+            results = dask.delayed(tasks).compute()
+            real_position_indexes = results[0]
+            cancel_reasons = results[1]
+
+
     deal_list_dict = {}
     for d in deal_list:
         deal_list_dict[str(d[0])] = d
@@ -93,7 +117,6 @@
     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, cancel_reasons, deal_big_buy_order_no_dict)
     return fdatas

--
Gitblit v1.8.0