From fb47d36048e94b9a506d5c153e3dd19a01e37df1 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期一, 30 十月 2023 16:30:27 +0800
Subject: [PATCH] bug修复

---
 utils/data_export_util.py |   32 ++++++++++++++++++++++++--------
 1 files changed, 24 insertions(+), 8 deletions(-)

diff --git a/utils/data_export_util.py b/utils/data_export_util.py
index fa26cd3..7de7140 100644
--- a/utils/data_export_util.py
+++ b/utils/data_export_util.py
@@ -24,7 +24,8 @@
     # 鑾峰彇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)
+    fdatas = export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes)
     __save_l2_datas(code, fdatas)
 
 
@@ -34,11 +35,12 @@
     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)
+    fdatas = export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes)
     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):
     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 +72,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
@@ -161,7 +167,7 @@
                 cancel_order_info = trade_info[2]
         format_data.append(cancel_order_info)
         format_data.append(data["val"].get("orderNo"))
-        fdatas.append((style_int, format_data))
+        fdatas.append((style_int, trade_info, format_data))
     return fdatas
 
 
@@ -179,21 +185,31 @@
     ws.write(0, 6, '绫诲瀷')
     ws.write(0, 7, '閲嶅鏁伴噺')
     ws.write(0, 8, '鎾ゅ崟鏃堕棿')
-    ws.write(0, 9, '璁㈠崟鍙�')
+    ws.write(0, 9, '澶囨敞')
+    ws.write(0, 10, '璁㈠崟鍙�')
     index = 0
-    font = xlwt.Font()
 
     cancel_style = xlwt.easyxf('pattern: pattern solid, fore_colour gray25')
     for fdata in fdatas:
         index += 1
         style_int = fdata[0]
-        data = fdatas[1]
+        trade_info = fdata[1]
+        data = fdata[2]
         style = None
         if style_int == 0:
             style = xlwt.easyxf('pattern: pattern solid')
         else:
             style = xlwt.easyxf('pattern: pattern solid, fore_colour light_yellow')
+        font = xlwt.Font()
         style.font = font
+
+        if trade_info:
+            if trade_info[0] == 0:
+                font.colour_index = 53
+            elif trade_info[0] == 1:
+                font.colour_index = 17
+            elif trade_info[0] == 2:
+                font.colour_index = 10
 
         for i in range(len(data)):
             if data[i] is None:
@@ -255,6 +271,6 @@
 
 if __name__ == "__main__":
     try:
-        export_l2_excel("600476")
+        export_l2_excel("000530")
     except Exception as e:
         logging.exception(e)

--
Gitblit v1.8.0