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

---
 third_data/code_plate_key_manager.py |  457 ++++++++++++++++++++++++++++++++++----------------------
 1 files changed, 274 insertions(+), 183 deletions(-)

diff --git a/third_data/code_plate_key_manager.py b/third_data/code_plate_key_manager.py
index 7eb3c70..1db7f34 100644
--- a/third_data/code_plate_key_manager.py
+++ b/third_data/code_plate_key_manager.py
@@ -6,57 +6,134 @@
 import json
 
 import constant
-from third_data import kpl_block_util
+from db.redis_manager_delegate import RedisUtils
+from third_data import kpl_block_util, kpl_api
 from utils import global_util, tool
-from log_module import log
-from db import redis_manager
+from log_module import log, async_log_util
+from db import redis_manager_delegate as redis_manager
 
-from log_module.log import logger_kpl_limit_up, logger_kpl_block_can_buy, logger_kpl_debug
+from log_module.log import logger_kpl_limit_up, logger_kpl_block_can_buy
 from third_data.kpl_util import KPLPlatManager
-from trade import trade_manager
+from trade import trade_manager, l2_trade_util
 
 
 # 浠g爜绮鹃�夋澘鍧楃鐞�
 class KPLCodeJXBlockManager:
+    __db = 3
     __redisManager = redis_manager.RedisManager(3)
     __code_blocks = {}
+    # 澶囩敤
+    __code_by_blocks = {}
+
+    __instance = None
+
+    def __new__(cls, *args, **kwargs):
+        if not cls.__instance:
+            cls.__instance = super(KPLCodeJXBlockManager, cls).__new__(cls, *args, **kwargs)
+        return cls.__instance
 
     def __get_redis(self):
         return self.__redisManager.getRedis()
 
-    def save_jx_blocks(self, code, blocks):
+    def save_jx_blocks(self, code, blocks, by=False):
         if blocks is None:
             return
+        if len(blocks) > 2:
+            blocks = blocks[:2]
+
         # 淇濆瓨鍓�2鏉℃暟鎹�
-        self.__get_redis().setex(f"kpl_jx_blocks-{code}", tool.get_expire(), json.dumps(blocks))
-        self.__code_blocks[code] = blocks
+        if by:
+            RedisUtils.setex_async(self.__db, f"kpl_jx_blocks_by-{code}", tool.get_expire(), json.dumps(blocks))
+            self.__code_by_blocks[code] = blocks
+        else:
+            RedisUtils.setex_async(self.__db, f"kpl_jx_blocks-{code}", tool.get_expire(), json.dumps(blocks))
+            self.__code_blocks[code] = blocks
 
     # 鑾峰彇绮鹃�夋澘鍧�
-    def get_jx_blocks(self, code):
-        if code in self.__code_blocks:
-            return self.__code_blocks[code]
-        val = self.__get_redis().get(f"kpl_jx_blocks-{code}")
-        if val is None:
-            return None
+    def get_jx_blocks(self, code, by=False):
+        if by:
+            if code in self.__code_by_blocks:
+                return self.__code_by_blocks[code]
+            val = RedisUtils.get(self.__get_redis(), f"kpl_jx_blocks_by-{code}")
+            if val is None:
+                return None
+            else:
+                val = json.loads(val)
+                self.__code_by_blocks[code] = val
+            return self.__code_by_blocks[code]
         else:
-            val = json.loads(val)
-            self.__code_blocks[code] = val
-        return self.__code_blocks[code]
+            if code in self.__code_blocks:
+                return self.__code_blocks[code]
+            val = RedisUtils.get(self.__get_redis(), f"kpl_jx_blocks-{code}")
+            if val is None:
+                return None
+            else:
+                val = json.loads(val)
+                self.__code_blocks[code] = val
+            return self.__code_blocks[code]
+
+    def get_jx_blocks_cache(self, code, by=False):
+        if by:
+            return self.__code_by_blocks.get(code)
+        else:
+            return self.__code_blocks.get(code)
+
+    # 浠庣綉缁滀笂鍔犺浇绮鹃�夋澘鍧�
+    def load_jx_blocks(self, code, buy_1_price, limit_up_price):
+        if limit_up_price and buy_1_price:
+            # 澶勭悊涔�1,鍗�1淇℃伅
+            pre_close_price = round(float(limit_up_price) / 1.1, 2)
+            # 濡傛灉娑ㄥ箙澶т簬7%灏辫鍙栨澘鍧�
+            price_rate = (buy_1_price - pre_close_price) / pre_close_price
+            if price_rate > 0.07:
+                if not self.get_jx_blocks_cache(code):
+                    blocks = kpl_api.getCodeJingXuanBlocks(code)
+                    self.save_jx_blocks(code, blocks)
+                    async_log_util.info(logger_kpl_block_can_buy,f"{code}:鑾峰彇鍒扮簿閫夋澘鍧�-{blocks}")
+            elif price_rate > 0.03:
+                # 娣诲姞澶囩敤鏉垮潡
+                if not self.get_jx_blocks_cache(code, by=True):
+                    blocks = kpl_api.getCodeJingXuanBlocks(code)
+                    self.save_jx_blocks(code, blocks, by=True)
+                    async_log_util.info(logger_kpl_block_can_buy, f"{code}:鑾峰彇鍒扮簿閫夋澘鍧�(澶囩敤)-{blocks}")
+
 
 
 # 寮�鐩樺暒绂佹浜ゆ槗鏉垮潡绠$悊
 class KPLPlateForbiddenManager:
     __redisManager = redis_manager.RedisManager(3)
+    __kpl_forbidden_plates_cache = set()
 
-    def __get_redis(self):
-        return self.__redisManager.getRedis()
+    __instance = None
+
+    def __new__(cls, *args, **kwargs):
+        if not cls.__instance:
+            cls.__instance = super(KPLPlateForbiddenManager, cls).__new__(cls, *args, **kwargs)
+            cls.__load_datas()
+        return cls.__instance
+
+    @classmethod
+    def __load_datas(cls):
+        __redis = cls.__get_redis()
+        try:
+            __kpl_forbidden_plates_cache = RedisUtils.smembers(__redis, "kpl_forbidden_plates")
+        finally:
+            RedisUtils.realse(__redis)
+
+    @classmethod
+    def __get_redis(cls):
+        return cls.__redisManager.getRedis()
 
     def save_plate(self, plate):
-        self.__get_redis().sadd("kpl_forbidden_plates", plate)
-        self.__get_redis().expire("kpl_forbidden_plates", tool.get_expire())
+        self.__kpl_forbidden_plates_cache.add(plate)
+        RedisUtils.sadd(self.__get_redis(), "kpl_forbidden_plates", plate)
+        RedisUtils.expire(self.__get_redis(), "kpl_forbidden_plates", tool.get_expire())
 
     def list_all(self):
-        return self.__get_redis().smembers("kpl_forbidden_plates")
+        return RedisUtils.smembers(self.__get_redis(), "kpl_forbidden_plates")
+
+    def list_all_cache(self):
+        return self.__kpl_forbidden_plates_cache
 
 
 class LimitUpCodesPlateKeyManager:
@@ -90,8 +167,8 @@
 
     # 浠婃棩娑ㄥ仠鍘熷洜鍙樺寲
     def set_today_limit_up_reason_change(self, code, from_reason, to_reason):
-        self.__get_redis().sadd(f"kpl_limit_up_reason_his-{code}", from_reason)
-        self.__get_redis().expire(f"kpl_limit_up_reason_his-{code}", tool.get_expire())
+        RedisUtils.sadd(self.__get_redis(), f"kpl_limit_up_reason_his-{code}", from_reason)
+        RedisUtils.expire(self.__get_redis(), f"kpl_limit_up_reason_his-{code}", tool.get_expire())
         self.__set_total_keys(code)
 
     # 璁剧疆浠g爜鐨勪粖鏃ユ定鍋滃師鍥�
@@ -216,7 +293,7 @@
             # 鍚庢帓鎵嶈兘鎸傚崟
             return 0, "韬綅涓嶄负鍚庢帓"
 
-        forbidden_plates = cls.__KPLPlateForbiddenManager.list_all()
+        forbidden_plates = cls.__KPLPlateForbiddenManager.list_all_cache()
         if key in forbidden_plates:
             return 0, "涓嶄拱璇ユ澘鍧�"
 
@@ -234,11 +311,9 @@
     @classmethod
     def is_in_top(cls, keys):
         reasons = cls.get_can_buy_key_set()
-        log.logger_kpl_debug.debug("甯傚満娴佸叆鍓�5:{}", reasons)
-        forbidden_plates = cls.__KPLPlateForbiddenManager.list_all()
+        forbidden_plates = cls.__KPLPlateForbiddenManager.list_all_cache()
         reasons = reasons - forbidden_plates
         temp_set = keys & reasons
-        log.logger_kpl_debug.debug("甯傚満娴佸叆鍓�5鍖归厤缁撴灉:{}", temp_set)
         if temp_set:
             return True, temp_set
         else:
@@ -258,15 +333,15 @@
 
     def set_history_limit_up_reason(self, code, reasons):
         self.__history_limit_up_reason_dict[code] = set(reasons)
-        self.__get_redis().setex(f"kpl_his_limit_up_reason-{code}", tool.get_expire(), json.dumps(list(reasons)))
-        logger_kpl_debug.debug(f"璁剧疆鍘嗗彶娑ㄥ仠鍘熷洜锛歿code}-{reasons}")
+        RedisUtils.setex(self.__get_redis(), f"kpl_his_limit_up_reason-{code}", tool.get_expire(),
+                         json.dumps(list(reasons)))
 
     # 濡傛灉杩斿洖鍊间笉涓篘one琛ㄧず宸茬粡鍔犺浇杩囧巻鍙插師鍥犱簡
     def get_history_limit_up_reason(self, code):
         reasons = self.__history_limit_up_reason_dict.get(code)
         if reasons is None:
             # 浠庡唴瀛樹腑鍔犺浇
-            val = self.__get_redis().get(f"kpl_his_limit_up_reason-{code}")
+            val = RedisUtils.get(self.__get_redis(), f"kpl_his_limit_up_reason-{code}")
             if val is not None:
                 val = set(json.loads(val))
                 self.__history_limit_up_reason_dict[code] = val
@@ -277,15 +352,19 @@
         else:
             return reasons
 
+    def get_history_limit_up_reason_cache(self, code):
+        reasons = self.__history_limit_up_reason_dict.get(code)
+        return reasons
+
     def set_blocks(self, code, blocks):
         self.__blocks_dict[code] = set(blocks)
-        self.__get_redis().setex(f"kpl_blocks-{code}", tool.get_expire(), json.dumps(list(blocks)))
+        RedisUtils.setex(self.__get_redis(), f"kpl_blocks-{code}", tool.get_expire(), json.dumps(list(blocks)))
 
     def get_blocks(self, code):
         reasons = self.__blocks_dict.get(code)
         if reasons is None:
             # 浠庡唴瀛樹腑鍔犺浇
-            val = self.__get_redis().get(f"kpl_blocks-{code}")
+            val = RedisUtils.get(self.__get_redis(), f"kpl_blocks-{code}")
             if val is not None:
                 val = set(json.loads(val))
                 self.__blocks_dict[code] = val
@@ -310,19 +389,20 @@
 class TargetCodePlateKeyManager:
     __redisManager = redis_manager.RedisManager(1)
     __CodesPlateKeysManager = CodesHisReasonAndBlocksManager()
+    __KPLCodeJXBlockManager = KPLCodeJXBlockManager()
 
     def __get_redis(self):
         return self.__redisManager.getRedis()
 
-    # 杩斿洖key闆嗗悎(鎺掗櫎鏃犳晥鏉垮潡),浠婃棩娑ㄥ仠鍘熷洜,浠婃棩鍘嗗彶娑ㄥ仠鍘熷洜,鍘嗗彶娑ㄥ仠鍘熷洜,浜岀骇,鏉垮潡
+    # 杩斿洖key闆嗗悎(鎺掗櫎鏃犳晥鏉垮潡),浠婃棩娑ㄥ仠鍘熷洜,浠婃棩鍘嗗彶娑ㄥ仠鍘熷洜,鍘嗗彶娑ㄥ仠鍘熷洜,浜岀骇,绮鹃�夋澘鍧�
     def get_plate_keys(self, code):
         keys = set()
         k1 = set()
         if code in LimitUpCodesPlateKeyManager.today_total_limit_up_reason_dict:
             k1 = {LimitUpCodesPlateKeyManager.today_total_limit_up_reason_dict[code]}
-        # 鍔犺浇浠婃棩鍘嗗彶鍘熷洜
-        k11 = self.__get_redis().smembers(f"kpl_limit_up_reason_his-{code}")
-        k2 = self.__CodesPlateKeysManager.get_history_limit_up_reason(code)
+        # 鍔犺浇浠婃棩鍘嗗彶鍘熷洜,鏆傛椂涓嶉渶瑕佸巻鍙插師鍥犱簡
+        k11 = set()  # RedisUtils.smembers(self.__get_redis(), f"kpl_limit_up_reason_his-{code}")
+        k2 = self.__CodesPlateKeysManager.get_history_limit_up_reason_cache(code)
         if k2 is None:
             k2 = set()
         k3 = set()
@@ -330,9 +410,13 @@
         if industry:
             k3 = {industry}
 
-        k4 = self.__CodesPlateKeysManager.get_blocks(code)
-        if k4 is None:
-            k4 = set()
+        k4 = set()
+        jingxuan_blocks = self.__KPLCodeJXBlockManager.get_jx_blocks_cache(code)
+        if not jingxuan_blocks:
+            jingxuan_blocks = self.__KPLCodeJXBlockManager.get_jx_blocks_cache(code, by=True)
+        if jingxuan_blocks:
+            jingxuan_blocks = jingxuan_blocks[:2]
+            k4 |= set([x[1] for x in jingxuan_blocks])
         for k in [k1, k11, k2, k3, k4]:
             keys |= k
 
@@ -357,7 +441,64 @@
     __TargetCodePlateKeyManager = TargetCodePlateKeyManager()
     __LimitUpCodesPlateKeyManager = LimitUpCodesPlateKeyManager()
     __CodesHisReasonAndBlocksManager = CodesHisReasonAndBlocksManager()
-    __KPLCodeJXBlockManager = KPLCodeJXBlockManager()
+    __CodesTradeStateManager = trade_manager.CodesTradeStateManager()
+    __can_buy_compute_result_dict = {}
+
+    @classmethod
+    def __remove_from_l2(cls, code, msg):
+        # 鏍规嵁韬綅绉婚櫎浠g爜
+        # return
+        # 涓嬭繃鍗曠殑浠g爜涓嶇Щ闄�
+        if trade_manager.CodesTradeStateManager().get_trade_state_cache(code) != trade_manager.TRADE_STATE_NOT_TRADE:
+            # 鍙涓嬭繃鍗曠殑灏变笉绉婚櫎
+            return
+        l2_trade_util.forbidden_trade(code, msg=msg)
+        logger_kpl_block_can_buy.info(msg)
+
+    @classmethod
+    def __is_block_can_buy(cls, code, block, current_limit_up_datas, code_limit_up_reason_dict,
+                           yesterday_current_limit_up_codes, limit_up_record_datas):
+        # log.logger_kpl_debug.info(f"鍒ゆ柇鏉垮潡鏄惁鍙拱锛歿block}")
+        # is_top_8_record, top_8_record = kpl_block_util.is_record_top_block(code, block, limit_up_record_datas,
+        #                                                                    yesterday_current_limit_up_codes, 50)
+        # is_top_4_current, top_4_current = kpl_block_util.is_current_top_block(code, block, current_limit_up_datas,
+        #                                                                       yesterday_current_limit_up_codes, 50)
+        # is_top_4 = is_top_8_record and is_top_4_current
+        # msg_list.append(f"\n瀹炴椂top10(娑ㄥ仠鏁伴噺锛歿len(current_limit_up_datas)})")
+        # msg_list.append(f"鍘嗗彶top20(娑ㄥ仠鏁伴噺锛歿len(top_8_record)})")
+
+        # 鑾峰彇涓绘澘瀹炴椂韬綅,鍓旈櫎楂樹綅鏉�
+        current_shsz_rank, front_current_shsz_rank_codes = kpl_block_util.get_code_current_rank(code, block,
+                                                                                                current_limit_up_datas,
+                                                                                                code_limit_up_reason_dict,
+                                                                                                yesterday_current_limit_up_codes,
+                                                                                                shsz=True)
+        record_shsz_rank, record_shsz_rank_codes = kpl_block_util.get_code_record_rank(code, block,
+                                                                                       limit_up_record_datas,
+                                                                                       code_limit_up_reason_dict,
+                                                                                       yesterday_current_limit_up_codes,
+                                                                                       shsz=True)
+        # 鑾峰彇涓绘澘鍘嗗彶韬綅
+
+        open_limit_up_codes = kpl_block_util.get_shsz_open_limit_up_codes(code, block, limit_up_record_datas,
+                                                                          code_limit_up_reason_dict)
+        if open_limit_up_codes:
+            # 涓绘澘寮�1
+            if current_shsz_rank < len(open_limit_up_codes) + 1 and record_shsz_rank < len(open_limit_up_codes) + 2:
+                # 灞炰簬榫�1,榫�2
+                return True, f"{tool.get_now_time_str()} {block}锛歵op10娑ㄥ仠鏉垮潡锛屼富鏉垮紑1({open_limit_up_codes}),灞炰簬涓绘澘鍓嶉緳{len(open_limit_up_codes) + 1}(瀹炴椂韬綅-{current_shsz_rank}/{len(current_limit_up_datas)})"
+            else:
+                if record_shsz_rank >= len(open_limit_up_codes) + 1:
+                    cls.__remove_from_l2(code, f"{code}鏍规嵁韬綅绂佹涔板叆锛氥�恵block}銆戝巻鍙茶韩浣峽record_shsz_rank}")
+                return False, f"鏉垮潡-{block}: top4娑ㄥ仠鏉垮潡锛屼富鏉垮紑1锛坽open_limit_up_codes}锛�,涓嶄负涓绘澘鍓嶉緳{len(open_limit_up_codes) + 1}锛堝疄鏃惰韩浣�-{current_shsz_rank}:{front_current_shsz_rank_codes},鍘嗗彶韬綅-{record_shsz_rank}锛�"
+        else:
+            if current_shsz_rank == 0 and record_shsz_rank < 2:
+                return True, f"{tool.get_now_time_str()} {block}锛歵op4娑ㄥ仠鏉垮潡锛岄潪涓绘澘寮�1锛屽睘浜庨緳1锛屽疄鏃舵定鍋滃垪琛ㄦ暟閲�({len(current_limit_up_datas)})"
+            else:
+                if record_shsz_rank >= 2:
+                    cls.__remove_from_l2(code, f"{code}鏍规嵁韬綅绂佹涔板叆锛氥�恵block}銆戝巻鍙茶韩浣峽record_shsz_rank}")
+
+                return False, f"鏉垮潡-{block}: top4娑ㄥ仠鏉垮潡锛岄潪涓绘澘寮�1,涓嶄负涓绘澘榫�1锛堝疄鏃惰韩浣�-{current_shsz_rank}:{front_current_shsz_rank_codes},鍘嗗彶韬綅-{record_shsz_rank}锛�"
 
     # 鑾峰彇鍙互涔扮殑鏉垮潡
     # current_limit_up_datas: 浠婃棩瀹炴椂娑ㄥ仠
@@ -377,154 +518,72 @@
                     code_limit_up_reason_dict[d[3]] = d[2]
             return code_limit_up_reason_dict
 
-        now_time = int(tool.get_now_time_str().replace(":", ""))
-        times = [100000, 103000, 110000, 133000, 150000]
-        time_index = 0
-        for i in range(len(times)):
-            if now_time < times[i]:
-                time_index = i
-                break
+        if current_limit_up_datas is None:
+            current_limit_up_datas = []
+
         # 鑾峰彇鐩爣浠g爜鏉垮潡
         keys, k1, k11, k2, k3, k4 = cls.__TargetCodePlateKeyManager.get_plate_keys(code)
-        log.logger_kpl_debug.info("{}鍏抽敭璇嶏細浠婃棩-{},浠婃棩鍘嗗彶-{},鍘嗗彶-{},浜岀骇琛屼笟-{},浠g爜鏉垮潡-{}", code, k1, k11, k2, k3, k4)
+        # log.logger_kpl_debug.info("{}鍏抽敭璇嶏細浠婃棩-{},浠婃棩鍘嗗彶-{},鍘嗗彶-{},浜岀骇琛屼笟-{},浠g爜鏉垮潡-{}", code, k1, k11, k2, k3, k4)
         keys = set()
         if k1:
             for k in k1:
                 if k not in constant.KPL_INVALID_BLOCKS:
                     keys.add(k)
-        if not keys:
-            for k in k2:
-                if k not in constant.KPL_INVALID_BLOCKS:
-                    keys.add(k)
-        if not keys:
+        # 濮嬬粓鑾峰彇绮鹃�夋澘鍧�
+        if True:
             # 鑾峰彇
-            jx_blocks = cls.__KPLCodeJXBlockManager.get_jx_blocks(code)
-            if jx_blocks:
-                keys |= set([k[1] for k in jx_blocks[:2]])
-        log.logger_kpl_debug.info("{}鏈�缁堝叧閿瘝锛歿}", code, keys)
+            if k4:
+                keys |= k4
+                keys = keys - constant.KPL_INVALID_BLOCKS
+
+        # log.logger_kpl_debug.info("{}鏈�缁堝叧閿瘝锛歿}", code, keys)
 
         # 娑ㄥ仠鍒楄〃涓尮閰嶅叧閿瘝锛岃繑鍥烇紙鏉垮潡:浠g爜闆嗗悎锛夛紝浠g爜闆嗗悎涓凡缁忔帓闄よ嚜韬�
         if not keys:
-            return cls.BLOCK_TYPE_NONE, None, "灏氭湭鎵惧埌娑ㄥ仠鍘熷洜"
+            return None, "灏氭湭鎵惧埌娑ㄥ仠鍘熷洜"
         code_limit_up_reason_dict = {}
         load_code_block()
         msg_list = []
-        for block in keys:
-            is_top_8_record, top_8_record = kpl_block_util.is_record_top_block(code, block, limit_up_record_datas,
-                                                                               yesterday_current_limit_up_codes, 20)
-            is_top_4_current, top_4_current = kpl_block_util.is_current_top_block(code, block, current_limit_up_datas,
-                                                                                  yesterday_current_limit_up_codes, 10)
-            is_top_4 = is_top_8_record and is_top_4_current
-            msg_list.append(f"\n瀹炴椂top10: {top_4_current}(娑ㄥ仠鏁伴噺锛歿len(current_limit_up_datas)})")
-            msg_list.append(f"鍘嗗彶top20: {top_8_record}")
 
-            # 鑾峰彇涓绘澘瀹炴椂韬綅
-            current_shsz_rank = kpl_block_util.get_code_current_rank(code, block, current_limit_up_datas,
-                                                                     code_limit_up_reason_dict, shsz=True)
-            record_shsz_rank = kpl_block_util.get_code_record_rank(code, block, limit_up_record_datas,
-                                                                   code_limit_up_reason_dict, shsz=True)
-            # 鑾峰彇涓绘澘鍘嗗彶韬綅
-            if is_top_4:
-                pen_limit_up_codes = kpl_block_util.get_shsz_open_limit_up_codes(code, block, limit_up_record_datas,
-                                                                                 code_limit_up_reason_dict)
-                if pen_limit_up_codes:
-                    # 涓绘澘寮�1
-                    if current_shsz_rank < len(pen_limit_up_codes) + 1 and record_shsz_rank < len(pen_limit_up_codes) + 1:
-                        # 灞炰簬榫�1,榫�2
-                        return block, f"{block}锛歵op4娑ㄥ仠鏉垮潡锛屼富鏉垮紑1({pen_limit_up_codes}),灞炰簬涓绘澘鍓嶉緳{len(pen_limit_up_codes) + 1}(瀹炴椂韬綅-{current_shsz_rank})"
-                    else:
-                        msg_list.append(
-                            f"鏉垮潡-{block}: top4娑ㄥ仠鏉垮潡锛屼富鏉垮紑1锛坽pen_limit_up_codes}锛�,涓嶄负涓绘澘鍓嶉緳{len(pen_limit_up_codes) + 1}锛堝疄鏃惰韩浣�-{current_shsz_rank},鍘嗗彶韬綅-{record_shsz_rank}锛�")
-                        continue
-                else:
-                    if current_shsz_rank == 0 and record_shsz_rank < 2:
-                        return block, f"{block}锛歵op4娑ㄥ仠鏉垮潡锛岄潪涓绘澘寮�1锛屽睘浜庨緳1"
-                    else:
-                        msg_list.append(
-                            f"鏉垮潡-{block}: top4娑ㄥ仠鏉垮潡锛岄潪涓绘澘寮�1,涓嶄负涓绘澘榫�1锛堝疄鏃惰韩浣�-{current_shsz_rank},鍘嗗彶韬綅-{record_shsz_rank}锛�")
-                        continue
+        can_buy_blocks = []
+        for block in keys:
+
+            can_buy, msg = cls.__is_block_can_buy(code, block, current_limit_up_datas, code_limit_up_reason_dict,
+                                                  yesterday_current_limit_up_codes, limit_up_record_datas)
+            if can_buy:
+                can_buy_blocks.append((block, msg))
             else:
-                # 鏄惁婊¤冻琛屼笟绮鹃�夋祦鍏ヨ姹�
-                is_in_top_input = RealTimeKplMarketData.is_in_top(set([block]))[0]
-                if not is_in_top_input:
-                    msg_list.append(
-                        f"鏉垮潡-{block}: 闈瀟op4娑ㄥ仠鏉垮潡锛屼笉婊¤冻绮鹃��/琛屼笟娴佸叆瑕佹眰")
-                    continue
-                else:
-                    # 鏄惁涓轰富鏉块緳1锛堝疄鏃堕緳1锛屽巻鍙查緳2浠ュ唴锛�
-                    if current_shsz_rank == 0 and record_shsz_rank < 2:
-                        return block, f"{block}锛氫笉鏄痶op4娑ㄥ仠鏉垮潡锛屾弧瓒崇簿閫�/琛屼笟娴佸叆瑕佹眰锛屾弧瓒充富鏉块緳1"
-                    else:
-                        msg_list.append(
-                            f"鏉垮潡-{block}: 涓嶆槸top4娑ㄥ仠鏉垮潡锛屾弧瓒崇簿閫�/琛屼笟娴佸叆瑕佹眰,涓嶄负涓绘澘榫�1锛堝疄鏃惰韩浣�-{current_shsz_rank},鍘嗗彶韬綅-{record_shsz_rank}锛�")
-                        continue
+                msg_list.append(msg)
+        if len(can_buy_blocks) == len(keys):
+            blocks = [x[0] for x in can_buy_blocks]
+            blocks_msg = "\n".join([x[1] for x in can_buy_blocks])
+            return blocks, blocks_msg
+
         return None, "\n".join(msg_list)
 
     # 鏄惁鍙互涓嬪崟
     # 杩斿洖锛氭槸鍚﹀彲浠ヤ笅鍗�,娑堟伅,鏉垮潡绫诲瀷
     @classmethod
-    def can_buy(cls, code, current_limit_up_datas, limit_up_record_datas, yesterday_current_limit_up_codes,
-                before_blocks_dict):
+    def can_buy(cls, code):
         if constant.TEST:
-            return True, "", cls.BLOCK_TYPE_NONE
+            return True, cls.BLOCK_TYPE_NONE
+        # if True:
+        #     # 娴嬭瘯
+        #     return True, "涓嶅垽鏂澘鍧楄韩浣�"
+        return cls.__can_buy_compute_result_dict.get(code)
 
-        block, block_msg = cls.get_can_buy_block(code, current_limit_up_datas,
-                                                 limit_up_record_datas, yesterday_current_limit_up_codes,
-                                                 before_blocks_dict)
-        if block is None:
+    @classmethod
+    def __compute_can_buy_blocks(cls, code, current_limit_up_datas, limit_up_record_datas,
+                                 yesterday_current_limit_up_codes, before_blocks_dict):
+
+        blocks, block_msg = cls.get_can_buy_block(code, current_limit_up_datas,
+                                                  limit_up_record_datas, yesterday_current_limit_up_codes,
+                                                  before_blocks_dict)
+        if not blocks:
             return False, block_msg
-
-        # ---------------------------------鍒ゆ柇鐩爣浠g爜鐨勬澘鍧�-------------------start------------
-        # 鍒ゆ柇鍖归厤鍑虹殑娑ㄥ仠鍘熷洜锛屽垽鏂槸鍚︽湁宸茬粡涓嬪崟鐨勭エ
-        # reason_need_buy_dict = {}
-        # for k in match_limit_up_result:
-        #     codes = match_limit_up_result[k]
-        #     final_codes_keys = [keys]
-        #     for code_ in codes:
-        #         temp_key_set = set()
-        #         temp_key_set |= cls.__CodesHisReasonAndBlocksManager.get_total_keys(code_)
-        #         temp = cls.__LimitUpCodesPlateKeyManager.total_code_keys_dict.get(code_)
-        #         if temp:
-        #             temp_key_set |= temp
-        #         # 浜岀骇
-        #         industry = global_util.code_industry_map.get(code_)
-        #         if industry:
-        #             temp_key_set.add(industry)
-        #
-        #         final_codes_keys.append(temp_key_set)
-        #     # 姹傚叡鍚岀殑鍏抽敭璇�
-        #     intersection = set(final_codes_keys[0])
-        #     for s in final_codes_keys:
-        #         intersection &= s
-        #     log.logger_kpl_debug.info("{}鐨勬澘鍧楁眰浜ら泦锛歿}-{}", code, k, intersection)
-        #
-        #     # 姹傚叕鍏辩殑鏉垮潡鏄惁鍦ㄦ祦鍏ュ墠5涓�
-        #     is_in, valid_keys = RealTimeKplMarketData.is_in_top(intersection)
-        #     if is_in:
-        #         reason_need_buy_dict[k] = (is_in, valid_keys)
-        # ---------------------------------鍒ゆ柇鐩爣浠g爜鐨勬澘鍧�-------------------end------------
-
-        # 鑾峰彇鏉垮潡鍙互涓嬪崟鐨勪釜鏁�
-        # can_buy_codes_count_dict = {}
-        #
-        # for key__ in match_limit_up_result:
-        #     can_buy_count, msg = RealTimeKplMarketData.get_can_buy_codes_count(code, key__)
-        #     can_buy_codes_count_dict[key__] = can_buy_count
-
-        # has_available_key = False
-        # for key in can_buy_codes_count_dict:
-        #     if can_buy_codes_count_dict[key] > 0:
-        #         has_available_key = True
-        #         break
-        # if not has_available_key:
-        #     return False, f"鍖归厤鍒扮殑銆恵','.join(match_limit_up_result.keys())}銆戞病鍦ㄧ簿閫�/琛屼笟鍙互涔板叆鐨勬澘鍧椾腑"
-
-        # ---------------------------------鍔犺浇宸茬粡涓嬪崟/鎴愪氦鐨勪唬鐮佷俊鎭�------------start-------------
-        # match_reasons = match_limit_up_result.keys()
-        # 鍒ゆ柇鍖归厤鍒扮殑鍘熷洜鏄惁宸茬粡鏈変笅鍗�/涔板叆鎴愬姛鐨勪唬鐮�
-        codes_delegate = set(trade_manager.get_codes_by_trade_states(
+        codes_delegate = set(cls.__CodesTradeStateManager.get_codes_by_trade_states_cache(
             {trade_manager.TRADE_STATE_BUY_DELEGATED, trade_manager.TRADE_STATE_BUY_PLACE_ORDER}))
-        codes_success = set(trade_manager.get_codes_by_trade_states(
+        codes_success = set(cls.__CodesTradeStateManager.get_codes_by_trade_states_cache(
             {trade_manager.TRADE_STATE_BUY_SUCCESS}))
 
         codes = codes_delegate | codes_success
@@ -536,7 +595,7 @@
         for c in codes:
             keys_, k1_, k11_, k2_, k3_, k4_ = cls.__TargetCodePlateKeyManager.get_plate_keys(c)
             # 瀹炴椂娑ㄥ仠鍘熷洜
-            trade_codes_blocks_dict[c] = k1_
+            trade_codes_blocks_dict[c] = k1_ | k4_
         # 缁熻鏉垮潡涓殑浠g爜
         trade_block_codes_dict = {}
         for c in trade_codes_blocks_dict:
@@ -550,35 +609,67 @@
                 trade_block_codes_dict[b].add(c)
 
         # ---------------------------------鍔犺浇宸茬粡涓嬪崟/鎴愪氦鐨勪唬鐮佷俊鎭�------------end-------------
-
         msg_list = []
-        for key in [block]:
+        for key in blocks:
             # 鏉垮潡涓凡缁忔湁鎴愪氦鐨勫氨涓嶄笅鍗曚簡
             if key in trade_success_blocks_count:
                 success_codes_count = len(trade_success_blocks_count[key])
-                if success_codes_count >= 2:
-                    msg_list.append(f"銆恵key}銆戜腑宸茬粡鏈墈success_codes_count}涓垚浜や唬鐮�")
-                    log.logger_kpl_debug.debug(f"{code}锛氭澘鍧楋紙{key}锛夊凡缁忔湁鎴愪氦銆恵trade_success_blocks_count[key]}銆�")
-                    continue
-                # 10:30浠ュ悗涔�1涓�
-                if int(tool.get_now_time_str().replace(":", "")) > int("103000") and success_codes_count >= 1:
+                if success_codes_count >= 1:
                     msg_list.append(f"銆恵key}銆戜腑宸茬粡鏈墈success_codes_count}涓垚浜や唬鐮�")
                     log.logger_kpl_debug.debug(f"{code}锛氭澘鍧楋紙{key}锛夊凡缁忔湁鎴愪氦銆恵trade_success_blocks_count[key]}銆�")
                     continue
             return True, block_msg
-            # 鏉垮潡鍙互涓嬪崟鏁伴噺
-            # if trade_block_codes_dict.get(key) is None or len(trade_block_codes_dict.get(key)) < \
-            #         can_buy_codes_count_dict[key]:
-            #     order_count = len(trade_block_codes_dict.get(key)) if key in trade_block_codes_dict else 0
-            #     logger_kpl_block_can_buy.info(
-            #         f"code={code}锛氥�恵key}銆戝彲浠ヤ笅鍗曪紝鐜版湁鏁伴噺锛歿order_count} 鏈�澶ф暟閲忥細{can_buy_codes_count_dict[key]}")
-            #     return True, f"鍙互涓嬪崟锛屾澘鍧�:銆恵key}銆�,鏉垮潡涓凡缁忎笅鍗曠殑鏁伴噺锛歿order_count}"
-            # else:
-            #     order_count = len(trade_block_codes_dict.get(key))
-            #     msg_list.append(f"銆恵key}銆戜腑涓嬪崟浠g爜鏁伴噺{order_count}/鍏佽涓嬪崟鏁伴噺{can_buy_codes_count_dict[key]}")
 
         return False, ",".join(msg_list)
 
+    # 鏇存柊浠g爜鏉垮潡鍒ゆ柇鏄惁鍙互涔扮殑缁撴灉
+    @classmethod
+    def update_can_buy_blocks(cls, code, current_limit_up_datas, limit_up_record_datas,
+                              yesterday_current_limit_up_codes,
+                              before_blocks_dict):
+        can_buy, msg = cls.__compute_can_buy_blocks(code, current_limit_up_datas, limit_up_record_datas,
+                                                    yesterday_current_limit_up_codes,
+                                                    before_blocks_dict)
+        # 淇濆瓨鏉垮潡璁$畻缁撴灉
+        cls.__can_buy_compute_result_dict[code] = (can_buy, msg)
+
+    # 鍒ゆ柇鏄惁涓虹湡鑰佸ぇ
+    @classmethod
+    def __is_real_first_limit_up(cls, code, block, current_limit_up_datas, limit_up_record_datas,
+                                 yesterday_current_limit_up_codes,
+                                 before_blocks_dict):
+        # 鍔犺浇娑ㄥ仠浠g爜鐨勭洰鏍囨澘鍧�
+        def load_code_block():
+            if limit_up_record_datas:
+                for d in limit_up_record_datas:
+                    if d[2] in constant.KPL_INVALID_BLOCKS and d[3] in before_blocks_dict:
+                        code_limit_up_reason_dict[d[3]] = list(before_blocks_dict.get(d[3]))[0]
+                    else:
+                        code_limit_up_reason_dict[d[3]] = d[2]
+            return code_limit_up_reason_dict
+
+        if current_limit_up_datas is None:
+            current_limit_up_datas = []
+        if limit_up_record_datas is None:
+            limit_up_record_datas = []
+        code_limit_up_reason_dict = {}
+        load_code_block()
+        can_buy, msg = cls.__is_block_can_buy(code, block, current_limit_up_datas, code_limit_up_reason_dict,
+                                              yesterday_current_limit_up_codes, limit_up_record_datas)
+        return can_buy, msg
+
+    @classmethod
+    def is_need_cancel(cls, code, limit_up_reason, current_limit_up_datas, limit_up_record_datas,
+                       yesterday_current_limit_up_codes,
+                       before_blocks_dict):
+        can_buy, msg = cls.__is_real_first_limit_up(code, limit_up_reason, current_limit_up_datas,
+                                                    limit_up_record_datas,
+                                                    yesterday_current_limit_up_codes,
+                                                    before_blocks_dict)
+        if not can_buy:
+            logger_kpl_block_can_buy.warning(f"{code} 鏍规嵁娑ㄥ仠鍘熷洜锛坽limit_up_reason}锛夊尮閰嶄笉鑳戒拱")
+        return not can_buy
+
 
 if __name__ == "__main__":
     pass

--
Gitblit v1.8.0