From 7d38983528a2706efd404379126df203c9e13038 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期三, 28 五月 2025 14:57:44 +0800
Subject: [PATCH] 大单解析按照代码来聚合

---
 data_parser/transaction_big_order_parser.py |   25 +++++++++++++------------
 1 files changed, 13 insertions(+), 12 deletions(-)

diff --git a/data_parser/transaction_big_order_parser.py b/data_parser/transaction_big_order_parser.py
index 31f3eb9..405432c 100644
--- a/data_parser/transaction_big_order_parser.py
+++ b/data_parser/transaction_big_order_parser.py
@@ -194,6 +194,9 @@
     ))
 
 
+__combined_df_cache = {}
+
+
 def extract_big_order_of_code(dir_path, code=None):
     """
     鎻愬彇浠g爜鐨勫ぇ鍗�
@@ -224,15 +227,23 @@
         print("鎷兼帴鏁版嵁涓嶅瓨鍦�")
         return
 
+    if not code:
+        codes = extract_big_order_codes(dir_path)
+        print("鎬讳唬鐮佹暟閲忥細", len(codes))
+        for code in codes:
+            extract_big_order_of_code(dir_path, code)
     if code:
         output_path = os.path.join(dir_path, f"big_buy_{code}.csv")
     else:
         output_path = os.path.join(dir_path, f"big_buy.csv")
 
     if os.path.exists(output_path):
-        print("璺緞宸插瓨鍦�")
+        print("璺緞宸插瓨鍦�:", output_path)
         return
-    df = pd.read_csv(combined_path)
+    df = __combined_df_cache.get(combined_path)
+    if not df:
+        df = pd.read_csv(combined_path)
+        __combined_df_cache[combined_path] = df
     df_copy = df.copy()
     if code:
         df_copy = df_copy[df_copy["SecurityID"] == int(code)]
@@ -257,16 +268,6 @@
     @param code:
     @return:
     """
-
-    def first_last(group):
-        """
-            鑾峰彇绗竴鏉℃暟鎹笌鏈�鍚庝竴鏉�
-            @param group:
-            @return:
-            """
-        return pd.Series({
-        })
-
     combined_path = os.path.join(dir_path, 'combined.csv')
     if not os.path.exists(combined_path):
         print("鎷兼帴鏁版嵁涓嶅瓨鍦�")

--
Gitblit v1.8.0