From 6b002b4a90fd73db0daca682aec7a8b1318f1bff Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期日, 19 十一月 2023 23:23:42 +0800
Subject: [PATCH] 炸板过后的票下午买入需要看板块不是独苗

---
 third_data/kpl_data_manager.py |   20 ++++++++++++++++++++
 1 files changed, 20 insertions(+), 0 deletions(-)

diff --git a/third_data/kpl_data_manager.py b/third_data/kpl_data_manager.py
index d90a373..f1805c5 100644
--- a/third_data/kpl_data_manager.py
+++ b/third_data/kpl_data_manager.py
@@ -45,6 +45,8 @@
     __kplPlatManager = KPLPlatManager()
     __LimitUpCodesPlateKeyManager = LimitUpCodesPlateKeyManager()
     __CodesPlateKeysManager = CodesHisReasonAndBlocksManager()
+    __current_code_reason_dict = {}
+    __current_reason_codes_dict = {}
 
     @classmethod
     def __load_hist_and_blocks(cls, code):
@@ -75,6 +77,16 @@
         if records:
             cls.latest_origin_datas = records
             cls.__LimitUpCodesPlateKeyManager.set_today_limit_up([(r[0], r[5]) for r in records])
+
+        code_reason_dict = {}
+        reason_code_dict = {}
+        for d in records:
+            code_reason_dict[d[0]] = d[5]
+            if d[5] not in reason_code_dict:
+                reason_code_dict[d[5]] = set()
+            reason_code_dict[d[5]].add(d[0])
+        cls.__current_code_reason_dict = code_reason_dict
+        cls.__current_reason_codes_dict = reason_code_dict
 
         # 娑ㄥ仠鏁版嵁璁板綍
         mysqldb = mysql_data.Mysqldb()
@@ -188,6 +200,14 @@
         bs = set([b[0] for b in results])
         return bs
 
+    @classmethod
+    def get_current_block(cls, code):
+        return cls.__current_code_reason_dict.get(code)
+
+    @classmethod
+    def get_current_codes_by_block(cls,block):
+        return cls.__current_reason_codes_dict.get(block)
+
 
 class KPLDataManager:
     __latest_datas = {}

--
Gitblit v1.8.0