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