From ae8d76a456b64c1c6c4ebf11b6ec33b7df217b1a Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期一, 18 八月 2025 13:43:46 +0800
Subject: [PATCH] bug修复
---
third_data/kpl_data_manager.py | 230 +++++++++++++++++++++++++++++++++++++++++++++++++++------
1 files changed, 206 insertions(+), 24 deletions(-)
diff --git a/third_data/kpl_data_manager.py b/third_data/kpl_data_manager.py
index b4cd314..4c094cd 100644
--- a/third_data/kpl_data_manager.py
+++ b/third_data/kpl_data_manager.py
@@ -9,7 +9,8 @@
import constant
from db.redis_manager_delegate import RedisUtils
-from log_module import async_log_util
+from log_module import async_log_util, log
+from third_data.kpl_data_constant import LimitUpDataConstant, TodayLimitUpReasonChangeManager
from utils import tool
# 寮�鐩樺暒鍘嗗彶娑ㄥ仠鏁版嵁绠＄悊
@@ -49,8 +50,12 @@
__LimitUpCodesPlateKeyManager = LimitUpCodesPlateKeyManager()
__CodesPlateKeysManager = CodesHisReasonAndBlocksManager()
__current_code_reasons_dict = {}
+ # 褰撳墠娑ㄥ仠鍘熷洜+鎺ㄨ崘鍘熷洜鐨勪唬鐮侀泦鍚�
__current_reason_codes_dict = {}
+ # 褰撳墠娑ㄥ仠鍘熷洜鐨勪唬鐮侀泦鍚�
+ __current_limit_up_reason_codes_dict = {}
__records_cache = {}
+ record_code_dict = {}
@classmethod
def __load_hist_and_blocks(cls, code):
@@ -67,7 +72,13 @@
pass
@classmethod
- def save_record(cls, day, records):
+ def save_record(cls, day, records, set_not_open=False):
+ """
+ @param day:
+ @param records:
+ @param set_not_open: 鏄惁闇�瑕佽缃偢鏉夸笌鍚�
+ @return:
+ """
# 缁熻鐐告澘
try:
last_codes = set()
@@ -85,6 +96,7 @@
# 缁熻浠ｇ爜鎵�灞炴澘鍧�
code_block_dict = {}
for data in records:
+ cls.record_code_dict[data[0]] = data
blocks = set(data[5].split("銆�"))
code = data[0]
for b in blocks:
@@ -92,15 +104,20 @@
code_block_dict[code] = set()
code_block_dict[code].add(b)
# 璁剧疆娑ㄥ仠鏁版嵁
-
if records:
cls.latest_origin_datas = records
cls.__LimitUpCodesPlateKeyManager.set_today_limit_up(
[(r[0], r[5], r[6].split('銆�') if r[6] else []) for r in records])
+ LimitUpDataConstant.set_current_limit_up_datas(records)
code_reasons_dict = {}
reason_codes_dict = {}
+ limit_up_reason_codes_dict = {}
for d in records:
+ if d[5] not in limit_up_reason_codes_dict:
+ limit_up_reason_codes_dict[d[5]] = set()
+ limit_up_reason_codes_dict[d[5]].add(d[0])
+
# 娑ㄥ仠鍘熷洜 + 鎺ㄨ崘鍘熷洜
bs = {d[5]}
if d[6]:
@@ -112,6 +129,7 @@
reason_codes_dict[b].add(d[0])
cls.__current_code_reasons_dict = code_reasons_dict
cls.__current_reason_codes_dict = reason_codes_dict
+ cls.__current_limit_up_reason_codes_dict = limit_up_reason_codes_dict
# 娑ㄥ仠鏁版嵁璁板綍
mysqldb = mysql_data.Mysqldb()
@@ -130,13 +148,17 @@
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,_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]})")
+ 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,_hot_block_code) values('{_id}','{day}','{d[5]}','{d[0]}','{d[1]}','{d[2]}','{d[6]}','{d[3]}',now(),now(),{d[10]},'{d[4]}',{d[7]},{d[9]})")
cls.__load_hist_and_blocks(code)
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]}',_hot_block_code_count={d[10]},_limit_up_high_info='{d[4]}' ,_update_time=now() where _id='{_id}'")
cls.latest_datas[_id] = d
+ if set_not_open:
+ # 闇�瑕佽缃笉鐐告澘
+ mysqldb.execute(f"update kpl_limit_up_record set _open = 0, _update_time = now() where _id='{_id}'")
+
cls.latest_datas[_id] = d
# 鑾峰彇鍘熸潵鐨勪唬鐮佹墍灞炴澘鍧�,鍒犻櫎涔嬪墠閿欒鐨勬澘鍧�
@@ -149,13 +171,13 @@
# 鏉垮潡鏇存敼杩�
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],
+ TodayLimitUpReasonChangeManager().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())
+ LimitUpDataConstant.set_history_limit_up_datas(cls.total_datas)
@classmethod
def load_total_datas(cls):
@@ -166,9 +188,12 @@
cls.__load_hist_and_blocks(d[3])
@staticmethod
- def list_all(day):
+ def list_all(day, max_limit_up_time=None):
mysqldb = mysql_data.Mysqldb()
- return mysqldb.select_all(f"select * from kpl_limit_up_record where _day='{day}'")
+ sql = f"select * from kpl_limit_up_record where _day='{day}'"
+ if max_limit_up_time:
+ sql += f" and cast(_limit_up_time as unsigned)<={max_limit_up_time}"
+ return mysqldb.select_all(sql)
@classmethod
def list_all_cache(cls, day):
@@ -230,6 +255,22 @@
return results
@classmethod
+ def get_latest_block_infos(cls, min_day=tool.date_sub(tool.get_now_date_str(), 180), code=None):
+ """
+
+ @param min_day: 榛樿鑾峰彇180澶╀箣鍓嶇殑
+ @param code: 浠ｇ爜
+ @return: 鏈�杩戠殑娑ㄥ仠鏉垮潡淇℃伅
+ """
+ sql = f"SELECT r.`_code`, r.`_day`, r.`_hot_block_name`, r.`_blocks` FROM `kpl_limit_up_record` r WHERE r.`_day`>'{min_day}'"
+ if code:
+ sql += f" AND _code='{code}'"
+ sql += " order by _create_time"
+ mysqldb = mysql_data.Mysqldb()
+ results = mysqldb.select_all(sql)
+ return results
+
+ @classmethod
def get_latest_blocks_set(cls, code):
results = cls.get_latest_infos(code, 2, False)
bs = set([b[0] for b in results])
@@ -248,10 +289,26 @@
return copy.deepcopy(cls.__current_reason_codes_dict)
@classmethod
+ def get_current_limit_up_reason_codes_dict(cls):
+ return copy.deepcopy(cls.__current_limit_up_reason_codes_dict)
+
+ @classmethod
def get_current_reasons(cls):
if cls.__current_reason_codes_dict:
return cls.__current_reason_codes_dict.keys()
return set()
+
+ @classmethod
+ def get_new_blocks(cls, day):
+ """
+ 鑾峰彇鏌愪竴澶╂柊鍑虹幇鐨勬澘鍧�(鏂版澘鍧�)
+ @param day:
+ @return:
+ """
+ sql = f"SELECT k.`_hot_block_name`, k.`_day` FROM `kpl_limit_up_record` k GROUP BY k.`_hot_block_name` HAVING k.`_day`='{day}' ORDER BY k.`_day` DESC"
+ mysqldb = mysql_data.Mysqldb()
+ results = mysqldb.select_all(sql)
+ return [x[0] for x in results]
class KPLDataManager:
@@ -302,7 +359,7 @@
@classmethod
# 鑾峰彇鏈�杩戝嚑澶╃殑鏁版嵁锛屾牴鎹棩鏈熷�掑簭杩斿洖
- def get_latest_from_file(cls, type, count):
+ def get_latest_from_file(cls, type, count, max_day=tool.get_now_date_str()):
files = os.listdir(constant.CACHE_PATH)
file_name_list = []
for f in files:
@@ -310,7 +367,6 @@
file_name_list.append((f.split("_")[0], f))
file_name_list.sort(key=lambda x: x[0], reverse=True)
- file_name_list = file_name_list[:count]
fresults = []
for file in file_name_list:
path = f"{constant.CACHE_PATH}/{file[1]}"
@@ -319,7 +375,10 @@
with open(path, 'r') as f:
lines = f.readlines()
if lines:
- fresults.append((file[0], json.loads(lines[0])))
+ if int(file[0].replace("-", "")) <= int(max_day.replace("-", "")):
+ fresults.append((file[0], json.loads(lines[0])))
+ if len(fresults) >= count:
+ break
return fresults
@@ -364,18 +423,24 @@
# 鑾峰彇鏈�杩戝嚑澶╃殑瀹炴椂娑ㄥ仠淇℃伅
# 杩斿洖鏍煎紡锛圼鏃ユ湡,鏁版嵁]锛�
-def get_current_limit_up_data_records(count):
+def get_current_limit_up_data_records(count, day=tool.get_now_date_str()):
fresults = []
- day = tool.get_now_date_str()
datas = []
if day in __limit_up_list_records_dict:
datas = __limit_up_list_records_dict[day]
else:
logger_debug.info("浠庢枃浠朵腑鑾峰彇鍓嶅嚑澶╃殑瀹炴椂娑ㄥ仠鏁版嵁")
- datas = KPLDataManager().get_latest_from_file(KPLDataType.LIMIT_UP, count + 2)
- if datas:
+ datas = KPLDataManager().get_latest_from_file(KPLDataType.LIMIT_UP, count + 2, max_day=day)
+ # 绉婚櫎姣斾粖澶╄繕澶х殑鏁版嵁
+ fdatas = []
+ for d in datas:
+ if int(d[0].replace("-", "")) > int(day.replace("-", "")):
+ continue
+ fdatas.append(d)
+ if fdatas:
# 淇濆瓨鏁版嵁
- __limit_up_list_records_dict[day] = datas
+ __limit_up_list_records_dict[day] = fdatas
+ datas = __limit_up_list_records_dict[day]
for i in range(len(datas)):
if datas[i][0] == day:
continue
@@ -400,10 +465,9 @@
__latest_current_limit_up_records = {}
-def get_latest_current_limit_up_records():
- day = tool.get_now_date_str()
+def get_latest_current_limit_up_records(day=tool.get_now_date_str(), max_day_count=15):
if day not in __latest_current_limit_up_records:
- fdatas = get_current_limit_up_data_records(15)
+ fdatas = get_current_limit_up_data_records(max_day_count)
__latest_current_limit_up_records[day] = fdatas
return __latest_current_limit_up_records.get(day)
@@ -433,12 +497,31 @@
logger_debug.info("浠诲姟淇-寮�鐩樺暒锛氭定鍋滃垪琛�")
# 澶т簬20s灏遍渶瑕佹洿鏂�
threading.Thread(target=cls.run_limit_up_task, daemon=True).start()
+ # key = "jingxuan_rank"
+ # if key not in cls.__latest_update_time_dict or time.time() - cls.__latest_update_time_dict[key] > 20:
+ # logger_debug.info("浠诲姟淇-寮�鐩樺暒锛氱簿閫夋祦鍏ュ垪琛�")
+ # # 澶т簬20s灏遍渶瑕佹洿鏂�
+ # threading.Thread(target=cls.run_market_jingxuan_in, daemon=True).start()
+ #
+ # key = "jingxuan_rank_out"
+ # if key not in cls.__latest_update_time_dict or time.time() - cls.__latest_update_time_dict[key] > 20:
+ # logger_debug.info("浠诲姟淇-寮�鐩樺暒锛氱簿閫夋祦鍑哄垪琛�")
+ # # 澶т簬20s灏遍渶瑕佹洿鏂�
+ # threading.Thread(target=cls.run_market_jingxuan_out, daemon=True).start()
+
+ key = "market_strong"
+ if key not in cls.__latest_update_time_dict or time.time() - cls.__latest_update_time_dict[key] > 20:
+ logger_debug.info("浠诲姟淇-寮�鐩樺暒锛氬競鍦哄己搴�")
+ # 澶т簬20s灏遍渶瑕佹洿鏂�
+ threading.Thread(target=cls.run_market_strong, daemon=True).start()
@classmethod
def run_limit_up_task(cls):
+ # 鍏抽棴log
+ log.close_print()
while True:
try:
- if (tool.is_trade_time() and int(tool.get_now_time_str().replace(':', '')) > int("092530")) or True:
+ if (tool.is_trade_time() and int(tool.get_now_time_str().replace(':', '')) > int("092530")):
results = kpl_api.getLimitUpInfoNew()
result = json.loads(results)
start_time = time.time()
@@ -453,6 +536,59 @@
pass
finally:
cls.__latest_update_time_dict["limit_up"] = time.time()
+ time.sleep(3)
+
+ @classmethod
+ def run_market_jingxuan_in(cls):
+ """
+ 绮鹃�夋祦鍏�
+ @return:
+ """
+ while True:
+ try:
+ if tool.is_trade_time():
+ results = kpl_api.getMarketJingXuanRealRankingInfo()
+ result = json.loads(results)
+ cls.__upload_data("jingxuan_rank", result)
+ except:
+ pass
+ finally:
+ cls.__latest_update_time_dict["jingxuan_rank"] = time.time()
+ time.sleep(3)
+
+ @classmethod
+ def run_market_jingxuan_out(cls):
+ """
+ 绮鹃�夋祦鍑�
+ @return:
+ """
+ while True:
+ try:
+ if tool.is_trade_time():
+ results = kpl_api.getMarketJingXuanRealRankingInfo(False)
+ result = json.loads(results)
+ cls.__upload_data("jingxuan_rank_out", result)
+ except:
+ pass
+ finally:
+ cls.__latest_update_time_dict["jingxuan_rank_out"] = time.time()
+ time.sleep(3)
+
+ @classmethod
+ def run_market_strong(cls):
+ """
+ 绮鹃�夋祦鍑�
+ @return:
+ """
+ while True:
+ try:
+ if tool.is_trade_time():
+ strong_value = kpl_api.getMarketStrong()
+ cls.__upload_data("market_strong", strong_value)
+ except:
+ pass
+ finally:
+ cls.__latest_update_time_dict["market_strong"] = time.time()
time.sleep(3)
@classmethod
@@ -490,15 +626,61 @@
cls.__upload_data("jingxuan_rank", result)
except:
pass
- time.sleep(3)
+ finally:
+ cls.__latest_update_time_dict["jingxuan_rank"] = time.time()
+ time.sleep(3)
+ else:
+ time.sleep(3)
threading.Thread(target=cls.run_limit_up_task, daemon=True).start()
+ threading.Thread(target=cls.run_market_strong, daemon=True).start()
# threading.Thread(target=get_bidding_money, daemon=True).start()
# threading.Thread(target=get_market_industry, daemon=True).start()
- # threading.Thread(target=get_market_jingxuan, daemon=True).start()
+ # threading.Thread(target=cls.run_market_jingxuan_in, daemon=True).start()
+ # threading.Thread(target=cls.run_market_jingxuan_out, daemon=True).start()
+
+
+@tool.singleton
+class CodeHighLevel:
+ """
+ 浠ｇ爜楂樺害绠＄悊
+ """
+ __instance = None
+ # 涓嬪崟鏉垮潡鐨勪唬鐮佽褰�
+ __code_level_dict = {}
+ __codes = set()
+
+ def __init__(self, day=tool.get_now_date_str()):
+ self.__day = day
+ self.__load_data(day)
+
+ @classmethod
+ def __load_data(cls, day):
+ fdatas = get_current_limit_up_data_records(15, day=day)
+ temp_dict = {d[0]: 2 for d in fdatas[0][1]}
+ break_codes = set()
+ for i in range(1, len(fdatas)):
+ codes = [d[0] for d in fdatas[i][1]]
+ for k in temp_dict:
+ if k in break_codes:
+ continue
+ if k in codes:
+ temp_dict[k] += 1
+ else:
+ break_codes.add(k)
+ cls.__code_level_dict = temp_dict
+
+ def get_high_level(self, code):
+ """
+ 鑾峰彇娑ㄥ仠楂樺害锛岄粯璁�1鏉�
+ @param code:
+ @return:
+ """
+ if code in self.__code_level_dict:
+ return self.__code_level_dict[code]
+ return 1
if __name__ == "__main__":
- print(get_latest_current_limit_up_records())
- print(get_latest_current_limit_up_records())
+ print(CodeHighLevel("2024-11-11").get_high_level("000833"))
input()
--
Gitblit v1.8.0