From 045a5aa6434da6e83c3d850b17e7e58cd7b55ef5 Mon Sep 17 00:00:00 2001 From: Administrator <admin@example.com> Date: 星期一, 15 五月 2023 15:46:32 +0800 Subject: [PATCH] 开盘啦板块影响交易逻辑 --- third_data/kpl_data_manager.py | 109 +++++++++++++++++++++++++++++++++++++++++++++++------- 1 files changed, 95 insertions(+), 14 deletions(-) diff --git a/third_data/kpl_data_manager.py b/third_data/kpl_data_manager.py index 27ea980..8904b81 100644 --- a/third_data/kpl_data_manager.py +++ b/third_data/kpl_data_manager.py @@ -5,17 +5,63 @@ import tool # 寮�鐩樺暒鍘嗗彶娑ㄥ仠鏁版嵁绠$悊 -from db import mysql_data +from db import mysql_data, redis_manager from l2 import code_price_manager +from log import logger_kpl_limit_up_reason_change from third_data import kpl_util +from third_data.code_plate_key_manager import LimitUpCodesPlateKeyManager -INVALID_BLOCKS = ["涓�瀛f姤澧為暱", "浜屽鎶ュ闀�", "涓夊鎶ュ闀�", "鍥涘鎶ュ闀�", "涓氱哗澧為暱", "涓姤澧為暱", "骞存姤澧為暱", "骞存姤棰勫", "鏃�", "娆℃柊鑲�", "ST鎽樺附", "瓒呰穼", "鑲℃潈杞", - "骞惰喘閲嶇粍"] + +# 浠g爜瀵瑰簲鐨勬定鍋滃師鍥犱繚瀛� +class KPLCodeLimitUpReasonManager: + __redisManager = redis_manager.RedisManager(3) + + def __get_redis(self): + return self.__redisManager.getRedis() + + def save_reason(self, code, reason): + self.__get_redis().setex(f"kpl_limitup_reason-{code}", tool.get_expire(), reason) + + def list_all(self): + keys = self.__get_redis().keys("kpl_limitup_reason-*") + dict_ = {} + for k in keys: + val = self.__get_redis().get(k) + dict_[k.split("-")[1]] = val + return dict_ + + +class KPLPlatManager: + def save_plat(self, _id, name): + if not _id: + return + mysqldb = mysql_data.Mysqldb() + key = f"{_id}-{name}" + results = mysqldb.select_one(f"select * from kpl_plate where _name='{name}'") + if not results: + mysqldb.execute(f"insert into kpl_plate(_id,_name,_key) values({_id},'{name}','{key}')") + + def get_plat(self, name): + mysqldb = mysql_data.Mysqldb() + results = mysqldb.select_one(f"select * from kpl_plate where _name='{name}'") + if results: + return results[0] + return None + + def get_same_plat_names(self, name): + mysqldb = mysql_data.Mysqldb() + plate = self.get_plat(name) + if not plate: + return {name} + results = mysqldb.select_all(f"select _name from kpl_plate where _id='{plate}'") + return set([r[0] for r in results]) class KPLLimitUpDataRecordManager: total_datas = None latest_datas = {} + __kplPlatManager = KPLPlatManager() + __LimitUpCodesPlateKeyManager = LimitUpCodesPlateKeyManager() @classmethod def save_record(cls, day, records): @@ -28,22 +74,33 @@ if not code_block_dict.get(code): code_block_dict[code] = set() code_block_dict[code].add(b) + # 璁剧疆娑ㄥ仠鏁版嵁 + if records: + cls.__LimitUpCodesPlateKeyManager.set_today_limit_up([(r[0], r[5]) for r in records]) # 娑ㄥ仠鏁版嵁璁板綍 mysqldb = mysql_data.Mysqldb() + # 缁熻娑ㄥ仠鍘熷洜鍜屾蹇典唬鐮� + plats = {} for d in records: - # (浠g爜, 鍚嶇О, 棣栨娑ㄥ仠鏃堕棿, 鏈�杩戞定鍋滄椂闂�, 鍑犳澘, 娑ㄥ仠鍘熷洜, 鏉垮潡, 瀹為檯娴侀��, 涓诲姏鍑�棰�) + plats[d[5]] = d[9] + for p in plats: + cls.__kplPlatManager.save_plat(plats[p], p) + + for d in records: + # (浠g爜, 鍚嶇О, 棣栨娑ㄥ仠鏃堕棿, 鏈�杩戞定鍋滄椂闂�, 鍑犳澘, 娑ㄥ仠鍘熷洜, 鏉垮潡, 瀹為檯娴侀��, 涓诲姏鍑�棰�,娑ㄥ仠鍘熷洜浠g爜,娑ㄥ仠鍘熷洜浠g爜鏁伴噺) code = d[0] _id = f"{day}_{code}_{d[5]}" result = mysqldb.select_one("select * from kpl_limit_up_record where _id='{}'".format(_id)) if not result: mysqldb.execute( - f"insert into kpl_limit_up_record(_id,_day,_hot_block_name,_code,_code_name,_limit_up_time,_blocks,_latest_limit_up_time,_update_time,_create_time) values('{_id}','{day}','{d[5]}','{d[0]}','{d[1]}','{d[2]}','{d[6]}','{d[3]}',now(),now())") + f"insert into kpl_limit_up_record(_id,_day,_hot_block_name,_code,_code_name,_limit_up_time,_blocks,_latest_limit_up_time,_update_time,_create_time,_hot_block_code_count,_limit_up_high_info,_zylt_val) values('{_id}','{day}','{d[5]}','{d[0]}','{d[1]}','{d[2]}','{d[6]}','{d[3]}',now(),now(),{d[10]},'{d[4]}',{d[7]})") + else: if _id in cls.latest_datas and json.dumps(cls.latest_datas.get(_id)) != json.dumps(d): mysqldb.execute( - f"update kpl_limit_up_record set _latest_limit_up_time='{d[3]}',_limit_up_time='{d[2]}' ,_update_time=now() where _id='{_id}'") + f"update kpl_limit_up_record set _latest_limit_up_time='{d[3]}',_limit_up_time='{d[2]}',_hot_block_code_count={d[10]},_limit_up_high_info='{d[4]}' ,_update_time=now() where _id='{_id}'") cls.latest_datas[_id] = d cls.latest_datas[_id] = d @@ -53,6 +110,14 @@ for dd in old_datas: if dd[2] not in code_block_dict[code]: mysqldb.execute(f"delete from kpl_limit_up_record where _id='{dd[0]}'") + logger_kpl_limit_up_reason_change.info(f"code-{dd[3]}:{dd[2]}-{code_block_dict[code]}") + # 鏉垮潡鏇存敼杩� + mysqldb.execute( + f"update kpl_limit_up_record set _hot_block_change = f'{dd[2]}' where _day='{dd[1]}' and _code='{code}'") + + cls.__LimitUpCodesPlateKeyManager.set_today_limit_up_reason_change(code, dd[2], + code_block_dict[code]) + if dd[0] in cls.latest_datas: cls.latest_datas.pop(dd[0]) cls.total_datas = KPLLimitUpDataRecordManager.list_all(tool.get_now_date_str()) @@ -60,6 +125,7 @@ @classmethod def load_total_datas(cls): cls.total_datas = KPLLimitUpDataRecordManager.list_all(tool.get_now_date_str()) + cls.__LimitUpCodesPlateKeyManager.set_today_total_limit_up([(r[3], r[2]) for r in cls.total_datas]) @staticmethod def list_all(day): @@ -93,7 +159,7 @@ @staticmethod def get_latest_blocks(code): wheres = [] - for b in INVALID_BLOCKS: + for b in constant.KPL_INVALID_BLOCKS: wheres.append(f"hb.`_hot_block_name` != '{b}'") wheres = " and ".join(wheres) sql = f"SELECT GROUP_CONCAT(_hot_block_name) FROM (SELECT hb.`_hot_block_name`,hb.`_day` FROM `kpl_limit_up_record` hb WHERE hb.`_code`='{code}' AND {wheres} ORDER BY hb.`_day` DESC LIMIT 10) a GROUP BY a._day ORDER BY a._day DESC LIMIT 1" @@ -101,11 +167,22 @@ return mysqldb.select_one(sql) # 鑾峰彇浠g爜鏈�杩戠殑鏉垮潡锛岃繑鍥瀃(鏉垮潡,鏃ユ湡)] - @staticmethod - def get_latest_infos(code, count): - sql = f"SELECT GROUP_CONCAT(_hot_block_name),`_day` FROM (SELECT hb.`_hot_block_name`,hb.`_day` FROM `kpl_limit_up_record` hb WHERE hb.`_code`='{code}' ORDER BY hb.`_day` DESC LIMIT 10) a GROUP BY a._day ORDER BY a._day DESC LIMIT {count}" + @classmethod + def get_latest_infos(cls, code, count, contains_today=True): + # 鍙幏鍙栨渶杩�180澶╃殑鏁版嵁 + min_day = tool.date_sub(tool.get_now_date_str(), 180) + sql = f"SELECT GROUP_CONCAT(_hot_block_name),`_day`,_blocks FROM (SELECT hb.`_hot_block_name`,hb.`_day`,hb._blocks FROM `kpl_limit_up_record` hb WHERE hb.`_code`='{code}' and hb.`_day` > '{min_day}' ORDER BY hb.`_day` DESC LIMIT 10) a GROUP BY a._day ORDER BY a._day DESC LIMIT {count}" mysqldb = mysql_data.Mysqldb() - return mysqldb.select_all(sql) + results = mysqldb.select_all(sql) + if results and not contains_today and results[0][1] == tool.get_now_date_str(): + return results[1:] + return results + + @classmethod + def get_latest_blocks_set(cls, code): + results = cls.get_latest_infos(code, 10, False) + bs = set([b[0] for b in results]) + return bs class KPLDataManager: @@ -163,10 +240,14 @@ line = lines[0] result = json.loads(line) list_ = kpl_util.parseDaBanData(result, kpl_util.DABAN_TYPE_LIMIT_UP) - KPLLimitUpDataRecordManager.save_record(day, list_) + # KPLLimitUpDataRecordManager.save_record(day, list_) + for r in list_: + print(r[-1], r[5]) + KPLPlatManager().save_plat(r[-1], r[5]) - print(day, list_) + # print(day, list_) if __name__ == "__main__": - print(KPLLimitUpDataRecordManager.get_latest_infos("000950", 4)) + ds = set(["1", "2", "3"]) + print(ds.pop()) -- Gitblit v1.8.0