From d6b6be5eb2ae00a8ccf46bd7d53cd7d0c1e59c72 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期五, 24 十一月 2023 19:19:37 +0800
Subject: [PATCH] 初始化独立某些方法

---
 utils/data_export_util.py |   42 +++++++++++++++++++++++++++++++-----------
 1 files changed, 31 insertions(+), 11 deletions(-)

diff --git a/utils/data_export_util.py b/utils/data_export_util.py
index cd55100..209f42a 100644
--- a/utils/data_export_util.py
+++ b/utils/data_export_util.py
@@ -11,6 +11,7 @@
 import constant
 from code_attribute import gpcode_manager
 import l2.l2_data_util
+from l2.huaxin import l2_huaxin_util
 from log_module import log, log_export
 from l2 import l2_data_source_util
 from trade import deal_big_money_manager
@@ -24,7 +25,12 @@
     # 鑾峰彇L2澶勭悊浣嶇疆淇℃伅
     process_indexs = log_export.get_l2_process_position(code, date)
     trade_indexs = log_export.get_l2_trade_position(code, date)
-    fdatas = export_l2_data(code, datas, process_indexs, trade_indexs)
+    real_position_indexes = log_export.get_real_place_order_positions(code, date)
+    deal_list = log_export.load_huaxin_deal_record(code)
+    deal_list_dict = {}
+    for d in deal_list:
+        deal_list_dict[d[0]] = d
+    fdatas = export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes, deal_list_dict)
     __save_l2_datas(code, fdatas)
 
 
@@ -34,11 +40,17 @@
     datas = local_today_datas[code]
     process_indexs = log_export.get_l2_process_position(code, date)
     trade_indexs = log_export.get_l2_trade_position(code, date)
-    fdatas = export_l2_data(code, datas, process_indexs, trade_indexs)
+    real_position_indexes = log_export.get_real_place_order_positions(code, date)
+    deal_list = log_export.load_huaxin_deal_record(code)
+    deal_list_dict = {}
+    for d in deal_list:
+        deal_list_dict[d[0]] = d
+
+    fdatas = export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes, deal_list_dict)
     return fdatas
 
 
-def export_l2_data(code, datas, process_indexs, trade_indexs):
+def export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes, deal_list_dict):
     def find_process_index(index):
         for i in range(0, len(process_indexs)):
             if process_indexs[i][0] <= index <= process_indexs[i][1]:
@@ -70,6 +82,10 @@
     for data in datas:
         index += 1
         trade_info = find_trade_index(data["index"])
+        if not trade_info:
+            # 鑾峰彇鐪熷疄涓嬪崟浣嶇疆
+            if data["index"] in real_position_indexes:
+                trade_info = [3]
         style_int = None
         if find_process_index(data["index"]) % 2 == 0:
             style_int = 0
@@ -108,31 +124,31 @@
 
         limit_price = ""
         if int(data["val"]["limitPrice"]) == 1:
-            limit_price = "娑ㄥ仠"
+            limit_price = "T"
         elif int(data["val"]["limitPrice"]) == 2:
-            limit_price = "璺屽仠"
+            limit_price = "D"
         operateDesc = ""
         if int(data["val"]["operateType"]) == 0:
             if len(limit_price) > 0:
-                operateDesc = '涔� ({})'.format(limit_price)
+                operateDesc = '涔皗}'.format(limit_price)
             else:
                 operateDesc = '涔�'
         elif int(data["val"]["operateType"]) == 1:
             if len(limit_price) > 0:
-                operateDesc = '涔版挙 ({})'.format(limit_price)
+                operateDesc = '涔版挙{}'.format(limit_price)
 
             else:
                 operateDesc = '涔版挙'
 
         elif int(data["val"]["operateType"]) == 2:
             if len(limit_price) > 0:
-                operateDesc = '鍗� ({})'.format(limit_price)
+                operateDesc = '鍗杮}'.format(limit_price)
             else:
                 operateDesc = '鍗�'
 
         elif int(data["val"]["operateType"]) == 3:
             if len(limit_price) > 0:
-                operateDesc = '鍗栨挙 ({})'.format(limit_price)
+                operateDesc = '鍗栨挙{}'.format(limit_price)
             else:
                 operateDesc = '鍗栨挙'
         format_data.append(operateDesc)
@@ -144,9 +160,13 @@
             # 涔�
             if cancel_data:
                 try:
-                    cancel_info = "{}-{}".format(cancel_data["index"], cancel_data["val"]["time"])
+                    cancel_info = "{}-{}".format(cancel_data["index"],f"{cancel_data['val']['time']}") + (f".{cancel_data['val']['tms']}" if "tms" in cancel_data["val"] else '')
                 except Exception as e:
                     logging.exception(e)
+            else:
+                if int(data["val"].get("orderNo")) in deal_list_dict:
+                    cancel_info = l2_huaxin_util.convert_time(deal_list_dict[int(data["val"].get("orderNo"))][3],
+                                                              with_ms=True)
         format_data.append(cancel_info)
         cancel_order_info = None
         if trade_info:
@@ -265,6 +285,6 @@
 
 if __name__ == "__main__":
     try:
-        export_l2_excel("000981")
+        export_l2_excel("000536")
     except Exception as e:
         logging.exception(e)

--
Gitblit v1.8.0