"""
|
代码行业关键词管理
|
"""
|
|
# 涨停代码关键词板块管理
|
import copy
|
import json
|
import time
|
|
import constant
|
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, 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
|
from third_data.kpl_util import KPLPlatManager
|
from trade import trade_manager, l2_trade_util
|
|
|
# 代码精选板块管理
|
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: list, by=False):
|
if not blocks:
|
return
|
final_blocks = copy.deepcopy(blocks)
|
if len(blocks) > 2:
|
final_blocks.clear()
|
# 根据涨幅排序
|
blocks.sort(key=lambda x: x[2])
|
blocks.reverse()
|
for b in blocks:
|
if b[2] > 0 and b[1] not in constant.KPL_INVALID_BLOCKS:
|
final_blocks.append(b)
|
if len(final_blocks) < 2:
|
final_blocks = blocks
|
|
# 保存前2条数据
|
if by:
|
RedisUtils.setex_async(self.__db, f"kpl_jx_blocks_by-{code}", tool.get_expire(), json.dumps(final_blocks))
|
self.__code_by_blocks[code] = (final_blocks, time.time())
|
else:
|
RedisUtils.setex_async(self.__db, f"kpl_jx_blocks-{code}", tool.get_expire(), json.dumps(final_blocks))
|
self.__code_blocks[code] = (final_blocks, time.time())
|
|
# 获取精选板块
|
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:
|
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:
|
jx_blocks_info = self.get_jx_blocks_cache(code)
|
if not jx_blocks_info:
|
blocks = kpl_api.getCodeJingXuanBlocks(code)
|
self.save_jx_blocks(code, blocks)
|
async_log_util.info(logger_kpl_block_can_buy, f"{code}:获取到精选板块-{blocks}")
|
else:
|
# 还没涨停的需要更新精选板块 更新精选板块
|
if abs(float(buy_1_price) - float(limit_up_price)) >= 0.001:
|
# 非涨停状态
|
UPDATE_TIME_SPACE = 5 * 60
|
time_diff = tool.trade_time_sub(tool.get_now_time_str(), "09:30:00")
|
if time_diff < 0:
|
UPDATE_TIME_SPACE = 60 * 60
|
else:
|
UPDATE_TIME_SPACE = int(time_diff / 30) + 60
|
if UPDATE_TIME_SPACE > 5 * 60:
|
UPDATE_TIME_SPACE = 5 * 60
|
|
if time.time() - jx_blocks_info[1] > UPDATE_TIME_SPACE:
|
# 距离上次更新时间过去了5分钟
|
blocks = kpl_api.getCodeJingXuanBlocks(code)
|
self.save_jx_blocks(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()
|
|
__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.__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 RedisUtils.smembers(self.__get_redis(), "kpl_forbidden_plates")
|
|
def list_all_cache(self):
|
return self.__kpl_forbidden_plates_cache
|
|
|
class LimitUpCodesPlateKeyManager:
|
# 今日涨停原因
|
today_limit_up_reason_dict = {}
|
today_total_limit_up_reason_dict = {}
|
total_code_keys_dict = {}
|
total_key_codes_dict = {}
|
__redisManager = redis_manager.RedisManager(1)
|
|
def __get_redis(self):
|
return self.__redisManager.getRedis()
|
|
# 获取今日涨停数据，格式：[(代码,涨停原因)]
|
def set_today_limit_up(self, datas):
|
temp_dict = {}
|
if datas:
|
for item in datas:
|
temp_dict[item[0]] = item[1]
|
self.today_limit_up_reason_dict = temp_dict
|
if datas:
|
for item in datas:
|
self.__set_total_keys(item[0])
|
self.set_today_total_limit_up(datas)
|
|
# 设置今日历史涨停数据
|
def set_today_total_limit_up(self, datas):
|
for item in datas:
|
code = item[0]
|
self.today_total_limit_up_reason_dict[code] = item[1]
|
|
# 今日涨停原因变化
|
def set_today_limit_up_reason_change(self, code, from_reason, to_reason):
|
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)
|
|
# 设置代码的今日涨停原因
|
def __set_total_keys(self, code):
|
keys = set()
|
# keys_his = self.__get_redis().smembers(f"kpl_limit_up_reason_his-{code}")
|
# keys |= keys_his
|
if code in self.today_limit_up_reason_dict:
|
if self.today_limit_up_reason_dict.get(code) not in constant.KPL_INVALID_BLOCKS:
|
keys.add(self.today_limit_up_reason_dict.get(code))
|
self.total_code_keys_dict[code] = keys
|
for k in keys:
|
if k not in self.total_key_codes_dict:
|
self.total_key_codes_dict[k] = set()
|
self.total_key_codes_dict[k].add(code)
|
|
logger_kpl_limit_up.info("{}板块关键词:{}", code, keys)
|
|
# 根据传入的关键词与涨停代码信息匹配身位
|
|
def get_codes_by_key_without_mine(self, key, code):
|
# 只比较今日涨停原因
|
codes_set = set()
|
if key in self.total_key_codes_dict:
|
codes_set |= self.total_key_codes_dict[key]
|
codes_set.discard(code)
|
return codes_set
|
|
# 涨停原因匹配关键字(和涨停列表中的涨停原因做对比),返回:{关键词:代码集合}
|
def match_limit_up_reason_keys(self, code, keys):
|
fresult = {}
|
for k in keys:
|
if k in self.total_key_codes_dict:
|
codes = set(self.total_key_codes_dict[k])
|
codes.discard(code)
|
if codes:
|
fresult[k] = codes
|
return fresult
|
|
|
# 实时开盘啦市场数据
|
class RealTimeKplMarketData:
|
# 精选前5
|
top_5_reason_list = []
|
# 行业前5
|
top_5_industry_list = []
|
#
|
top_5_key_dict = {}
|
total_reason_dict = {}
|
total_industry_dict = {}
|
__KPLPlateForbiddenManager = KPLPlateForbiddenManager()
|
__LimitUpCodesPlateKeyManager = LimitUpCodesPlateKeyManager()
|
__KPLPlatManager = KPLPlatManager()
|
|
@classmethod
|
def set_top_5_reasons(cls, datas):
|
temp_list = []
|
for d in datas:
|
cls.total_reason_dict[d[1]] = d
|
# 排序
|
for i in range(0, len(datas)):
|
if datas[i][1] not in constant.KPL_INVALID_BLOCKS:
|
# （名称,净流入金额,排名）
|
temp_list.append((datas[i][1], datas[i][3], len(temp_list)))
|
# 只获取前10个
|
if len(temp_list) > 10:
|
break
|
if datas[i][3] < 3 * 10000 * 10000:
|
break
|
|
for temp in temp_list:
|
names = cls.__KPLPlatManager.get_same_plat_names_by_id(temp[0])
|
for name in names:
|
if name == temp[1]:
|
continue
|
temp_list.append((name, temp[1], temp[2]))
|
cls.top_5_reason_list = temp_list
|
cls.__reset_top_5_dict()
|
|
@classmethod
|
def set_top_5_industry(cls, datas):
|
for d in datas:
|
cls.total_industry_dict[d[1]] = d
|
temp_list = []
|
for i in range(0, len(datas)):
|
if datas[i][1] in constant.KPL_INVALID_BLOCKS:
|
continue
|
temp_list.append((datas[i][1], datas[i][2], len(temp_list)))
|
if len(temp_list) > 10:
|
break
|
if datas[i][2] < 3 * 10000 * 10000:
|
break
|
cls.top_5_industry_list = temp_list
|
cls.__reset_top_5_dict()
|
|
@classmethod
|
def __reset_top_5_dict(cls):
|
temp_dict = {}
|
for t in cls.top_5_industry_list:
|
temp_dict[t[0]] = t
|
for t in cls.top_5_reason_list:
|
temp_dict[t[0]] = t
|
cls.top_5_key_dict = temp_dict
|
|
# 获取能够买的行业关键字set
|
@classmethod
|
def get_can_buy_key_set(cls):
|
temp_set = cls.top_5_key_dict.keys()
|
return temp_set
|
|
# 通过关键字判断能买的代码数量
|
@classmethod
|
def get_can_buy_codes_count(cls, code, key):
|
# 判断行业涨停票数量，除开自己必须大于1个
|
temp_codes = LimitUpCodesPlateKeyManager.total_key_codes_dict.get(key)
|
if temp_codes is None:
|
temp_codes = set()
|
else:
|
temp_codes = set(temp_codes)
|
temp_codes.discard(code)
|
if len(temp_codes) < 1:
|
# 后排才能挂单
|
return 0, "身位不为后排"
|
|
forbidden_plates = cls.__KPLPlateForbiddenManager.list_all_cache()
|
if key in forbidden_plates:
|
return 0, "不买该板块"
|
|
# 10:30以前可以挂2个单
|
if int(tool.get_now_time_str().replace(':', '')) < int("100000"):
|
return 2, "10:00以前可以挂2个单"
|
# 10:30以后
|
if key not in cls.top_5_key_dict:
|
return 0, "净流入没在前5"
|
if cls.top_5_key_dict[key][1] > 3 * 10000 * 10000:
|
return 2, "净流入在前5且大于3亿"
|
else:
|
return 1, "净流入在前5"
|
|
@classmethod
|
def is_in_top(cls, keys):
|
reasons = cls.get_can_buy_key_set()
|
forbidden_plates = cls.__KPLPlateForbiddenManager.list_all_cache()
|
reasons = reasons - forbidden_plates
|
temp_set = keys & reasons
|
if temp_set:
|
return True, temp_set
|
else:
|
return False, None
|
|
|
# 代码历史涨停原因与板块管理
|
class CodesHisReasonAndBlocksManager:
|
__redisManager = redis_manager.RedisManager(1)
|
# 历史涨停原因
|
__history_limit_up_reason_dict = {}
|
# 板块
|
__blocks_dict = {}
|
|
def __get_redis(self):
|
return self.__redisManager.getRedis()
|
|
def set_history_limit_up_reason(self, code, reasons):
|
self.__history_limit_up_reason_dict[code] = set(reasons)
|
RedisUtils.setex(self.__get_redis(), f"kpl_his_limit_up_reason-{code}", tool.get_expire(),
|
json.dumps(list(reasons)))
|
|
# 如果返回值不为None表示已经加载过历史原因了
|
def get_history_limit_up_reason(self, code):
|
reasons = self.__history_limit_up_reason_dict.get(code)
|
if reasons is None:
|
# 从内存中加载
|
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
|
if code in self.__history_limit_up_reason_dict:
|
return self.__history_limit_up_reason_dict.get(code)
|
else:
|
return None
|
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)
|
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 = RedisUtils.get(self.__get_redis(), f"kpl_blocks-{code}")
|
if val is not None:
|
val = set(json.loads(val))
|
self.__blocks_dict[code] = val
|
if code in self.__blocks_dict:
|
return self.__blocks_dict.get(code)
|
else:
|
return None
|
else:
|
return reasons
|
|
def get_total_keys(self, code):
|
reasons = self.get_history_limit_up_reason(code)
|
if reasons is None:
|
reasons = set()
|
blocks = self.get_blocks(code)
|
if blocks is None:
|
blocks = set()
|
return reasons | blocks
|
|
|
# 目标代码板块关键词管理
|
class TargetCodePlateKeyManager:
|
__redisManager = redis_manager.RedisManager(1)
|
__CodesPlateKeysManager = CodesHisReasonAndBlocksManager()
|
__KPLCodeJXBlockManager = KPLCodeJXBlockManager()
|
|
def __get_redis(self):
|
return self.__redisManager.getRedis()
|
|
# 返回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 = 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()
|
industry = global_util.code_industry_map.get(code)
|
if industry:
|
k3 = {industry}
|
|
k4 = set()
|
jingxuan_block_info = self.__KPLCodeJXBlockManager.get_jx_blocks_cache(code)
|
if not jingxuan_block_info:
|
jingxuan_block_info = self.__KPLCodeJXBlockManager.get_jx_blocks_cache(code, by=True)
|
if jingxuan_block_info:
|
jingxuan_blocks = jingxuan_block_info[0]
|
k4 |= set([x[1] for x in jingxuan_blocks])
|
for k in [k1, k11, k2, k3, k4]:
|
keys |= k
|
|
# 排除无效的涨停原因
|
keys = keys - set(constant.KPL_INVALID_BLOCKS)
|
|
return keys, k1, k11, k2, k3, k4
|
|
|
class CodePlateKeyBuyManager:
|
# 无板块
|
BLOCK_TYPE_NONE = -1
|
# 一般板块
|
BLOCK_TYPE_COMMON = 0
|
# 强势板块
|
BLOCK_TYPE_STRONG = 1
|
# 猛拉板块
|
BLOCK_TYPE_SOON_LIMIT_UP = 2
|
# 潜伏板块
|
BLOCK_TYPE_START_UP = 3
|
|
__TargetCodePlateKeyManager = TargetCodePlateKeyManager()
|
__LimitUpCodesPlateKeyManager = LimitUpCodesPlateKeyManager()
|
__CodesHisReasonAndBlocksManager = CodesHisReasonAndBlocksManager()
|
__CodesTradeStateManager = trade_manager.CodesTradeStateManager()
|
__can_buy_compute_result_dict = {}
|
|
@classmethod
|
def __remove_from_l2(cls, code, msg):
|
# 根据身位移除代码
|
# return
|
# 下过单的代码不移除
|
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, current_limit_up_block_codes_dict):
|
|
# 独苗判断
|
block_codes = current_limit_up_block_codes_dict.get(block)
|
if not block_codes:
|
return False, True, ""
|
elif len(block_codes) == 1 and code in block_codes:
|
return False, 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)
|
current_open_limit_up_codes = kpl_block_util.get_shsz_open_limit_up_codes_current(code, block,
|
current_limit_up_datas)
|
|
max_rank = 2
|
# 如果是强势板块（有1个开1/有非主板涨停/有高位板）可以买到老三
|
msg_list = []
|
for bc in block_codes:
|
if bc in current_open_limit_up_codes:
|
max_rank = 3
|
msg_list.append(f"{bc}开1")
|
break
|
elif bc in yesterday_current_limit_up_codes:
|
max_rank = 3
|
msg_list.append(f"{bc}高位板")
|
break
|
elif bc.find("00") != 0 and bc.find("60") != 0:
|
max_rank = 3
|
msg_list.append(f"{bc}创业板/科创板")
|
break
|
|
# 判断身位
|
|
# 需要排除的老大的代码
|
exclude_first_codes = set() # HighIncreaseCodeManager().list_all()
|
|
# 获取主板开1的代码
|
|
# 剔除高位板
|
if open_limit_up_codes and yesterday_current_limit_up_codes:
|
open_limit_up_codes -= yesterday_current_limit_up_codes
|
|
# 获取主板实时身位,剔除高位板
|
# 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,
|
# exclude_first_codes,
|
# len(
|
# open_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)
|
|
if record_shsz_rank < len(open_limit_up_codes) + max_rank:
|
return True, False, f"【{block}】前排代码：{record_shsz_rank_codes}"
|
else:
|
return True, False, f"【{block}】前排代码：{record_shsz_rank_codes} 超过{len(open_limit_up_codes) + max_rank}个"
|
|
# 过时的代码
|
# 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}：top10涨停板块，主板开1({open_limit_up_codes}),属于主板前龙{len(open_limit_up_codes) + 1}(实时身位-{current_shsz_rank}:{front_current_shsz_rank_codes}/{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}：top4涨停板块，非主板开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: 今日实时涨停
|
# latest_2_day_limit_up_datas：最近2天的实时涨停（不含今日）
|
# limit_up_record_datas：今日历史涨停
|
# yesterday_current_limit_up_codes ： 昨日涨停代码
|
# before_blocks_dict：历史涨停原因
|
# 返回板块的计算结果[(板块名称,是否可买,是否是独苗,信息)]
|
|
@classmethod
|
def get_can_buy_block(cls, code, current_limit_up_datas, limit_up_record_datas, yesterday_current_limit_up_codes,
|
before_blocks_dict, current_limit_up_block_codes_dict):
|
# 加载涨停代码的目标板块
|
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 = []
|
|
# 获取目标代码板块
|
keys, k1, k11, k2, k3, k4 = cls.__TargetCodePlateKeyManager.get_plate_keys(code)
|
# log.logger_kpl_debug.info("{}关键词：今日-{},今日历史-{},历史-{},二级行业-{},代码板块-{}", 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 True:
|
# 获取
|
if k4:
|
keys |= k4
|
keys = keys - constant.KPL_INVALID_BLOCKS
|
|
# log.logger_kpl_debug.info("{}最终关键词：{}", code, keys)
|
|
# 涨停列表中匹配关键词，返回（板块:代码集合），代码集合中已经排除自身
|
|
fresults = []
|
if not keys:
|
return fresults
|
code_limit_up_reason_dict = {}
|
load_code_block()
|
for block in keys:
|
can_buy, unique, 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,
|
current_limit_up_block_codes_dict)
|
fresults.append((block, can_buy, unique, msg))
|
return fresults
|
|
# 是否可以下单
|
# 返回：是否可以下单,消息,板块类型
|
@classmethod
|
def can_buy(cls, code):
|
if constant.TEST:
|
return ["测试"], True, cls.BLOCK_TYPE_NONE
|
# if True:
|
# # 测试
|
# return True, "不判断板块身位"
|
return cls.__can_buy_compute_result_dict.get(code)
|
|
# 返回:(可以买的板块列表, 是否是独苗, 消息简介)
|
@classmethod
|
def __compute_can_buy_blocks(cls, code, current_limit_up_datas, limit_up_record_datas,
|
yesterday_current_limit_up_codes, before_blocks_dict,
|
current_limit_up_block_codes_dict):
|
blocks_compute_results = cls.get_can_buy_block(code, current_limit_up_datas,
|
limit_up_record_datas, yesterday_current_limit_up_codes,
|
before_blocks_dict, current_limit_up_block_codes_dict)
|
if not blocks_compute_results:
|
return False, True, "没有找到板块"
|
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(cls.__CodesTradeStateManager.get_codes_by_trade_states_cache(
|
{trade_manager.TRADE_STATE_BUY_SUCCESS}))
|
codes = codes_delegate | codes_success
|
# 统计成交代码的板块
|
trade_codes_blocks_dict = {}
|
# 已经成交的板块
|
trade_success_blocks_count = {}
|
trade_delegate_blocks_count = {}
|
for c in codes:
|
keys_, k1_, k11_, k2_, k3_, k4_ = cls.__TargetCodePlateKeyManager.get_plate_keys(c)
|
# 实时涨停原因 + 推荐原因
|
if not k1_:
|
trade_codes_blocks_dict[c] = k4_
|
else:
|
trade_codes_blocks_dict[c] = k1_
|
# 统计板块中的代码
|
trade_block_codes_dict = {}
|
for c in trade_codes_blocks_dict:
|
for b in trade_codes_blocks_dict[c]:
|
if c in codes_success:
|
if b not in trade_success_blocks_count:
|
trade_success_blocks_count[b] = set()
|
trade_success_blocks_count[b].add(c)
|
if c in codes_delegate:
|
if b not in trade_delegate_blocks_count:
|
trade_delegate_blocks_count[b] = set()
|
trade_delegate_blocks_count[b].add(c)
|
|
if b not in trade_block_codes_dict:
|
trade_block_codes_dict[b] = set()
|
trade_block_codes_dict[b].add(c)
|
|
# ---------------------------------加载已经下单/成交的代码信息------------end-------------
|
|
#
|
can_buy_blocks = []
|
unique_count = 0
|
msg_list = []
|
for r in blocks_compute_results:
|
# r的数据结构(板块,是否可以买,是否独苗,消息)
|
if r[2]:
|
unique_count += 1
|
if r[1]:
|
if r[0] in trade_success_blocks_count and len(trade_success_blocks_count[r[0]]) > 0:
|
msg_list.append(f"【{r[0]}】有成交代码：{trade_success_blocks_count[r[0]]}")
|
continue
|
if r[0] in trade_delegate_blocks_count and len(trade_delegate_blocks_count[r[0]]) >= 2:
|
msg_list.append(f"【{r[0]}】已挂单：{trade_delegate_blocks_count[r[0]]}")
|
continue
|
can_buy_blocks.append(r[0])
|
msg_list.append(r[3])
|
# 所有板块都是独苗
|
if unique_count == len(blocks_compute_results):
|
return can_buy_blocks, True, ",".join(msg_list)
|
return can_buy_blocks, False, ",".join(msg_list)
|
|
# 更新代码板块判断是否可以买的结果
|
@classmethod
|
def update_can_buy_blocks(cls, code, current_limit_up_datas, limit_up_record_datas,
|
yesterday_current_limit_up_codes,
|
before_blocks_dict, current_limit_up_block_codes_dict):
|
can_buy_blocks, unique, msg = cls.__compute_can_buy_blocks(code, current_limit_up_datas, limit_up_record_datas,
|
yesterday_current_limit_up_codes,
|
before_blocks_dict,
|
current_limit_up_block_codes_dict)
|
# 保存板块计算结果
|
cls.__can_buy_compute_result_dict[code] = (can_buy_blocks, unique, msg)
|
|
|
if __name__ == "__main__":
|
pass
|
