From 892b50e242e3c59a738b92dfdfee1bf1ff8932f2 Mon Sep 17 00:00:00 2001 From: Administrator <admin@example.com> Date: 星期五, 21 十月 2022 16:59:58 +0800 Subject: [PATCH] 新策略修改 --- gpcode_manager.py | 40 +++++++++++++++++++++++++++++++++------- 1 files changed, 33 insertions(+), 7 deletions(-) diff --git a/gpcode_manager.py b/gpcode_manager.py index 6c0e3fd..9a0f239 100644 --- a/gpcode_manager.py +++ b/gpcode_manager.py @@ -20,22 +20,49 @@ # 鑾峰彇鍩烘湰淇℃伅 code_datas = juejin.JueJinManager.get_gp_latest_info(gpset) codes = [] + name_codes = {} for _data in code_datas: # 姝e父鐨勮偂绁� if _data["sec_type"] == 1 and _data["sec_level"] == 1: code = _data["symbol"].split(".")[1] if code.find("30") != 0 and code.find("68") != 0: + name = _data["sec_name"] codes.append(code) + # 淇濆瓨浠g爜瀵瑰簲鐨勫悕绉� + name_codes[name] = code redis_instance = __redisManager.getRedis() # 鍒犻櫎涔嬪墠鐨� redis_instance.delete("gp_list") + redis_instance.delete("gp_list_names") for d in codes: redis_instance.sadd("gp_list", d) + redis_instance.set("gp_list_names", json.dumps(name_codes)) + + +# 鑾峰彇鍚嶇О瀵瑰簲鐨勪唬鐮� +def get_name_code(name): + redis_instance = __redisManager.getRedis() + val = redis_instance.get("gp_list_names") + if not val: + return None + val = json.loads(val) + return val.get(name) + + +def get_name_codes(): + redis_instance = __redisManager.getRedis() + val = redis_instance.get("gp_list_names") + if not val: + return None + val = json.loads(val) + return val # 娑ㄥ仠鐘佺鍧� def set_limit_up_list(gpset): + if gpset is None: + return # 淇濆瓨鍒板唴瀛樹腑 global_util.add_limit_up_codes(gpset) # 鑾峰彇鍩烘湰淇℃伅 @@ -45,7 +72,7 @@ for d in gpset: redis_instance.sadd("gp_limit_up_list", json.dumps(d)) redis_instance.expire("gp_limit_up_list", tool.get_expire()) - redis_instance.setex("gp_limit_up_list_update_time",tool.get_expire(),round( time.time()*1000)) + redis_instance.setex("gp_limit_up_list_update_time", tool.get_expire(), round(time.time() * 1000)) # 鑾峰彇娑ㄥ仠鍒楄〃 @@ -92,7 +119,7 @@ # 璁剧疆鏀剁洏浠� def set_price_pre(code, price): - codes= get_gp_list() + codes = get_gp_list() if code not in codes: return redis_instance = __redisManager.getRedis() @@ -145,11 +172,10 @@ # 鏍规嵁浣嶇疆鑾峰彇姝e湪鐩戝惉鐨勪唬鐮� def get_listen_code_by_pos(client_id, pos): redis_instance = __redisManager.getRedis() - key="listen_code-{}-{}".format(client_id, pos) + key = "listen_code-{}-{}".format(client_id, pos) value = redis_instance.get(key) # print("redis:", key,value) return value - # 璁剧疆浣嶇疆鐨勭洃鍚唬鐮� @@ -221,7 +247,7 @@ def is_listen_full(): clients = data_process.getValidL2Clients() codes = get_listen_codes() - return len(codes) >= 8*len(clients) + return len(codes) >= 8 * len(clients) # 鏄惁姝e湪鎿嶄綔 @@ -248,5 +274,5 @@ # print(is_listen_full()) # print(is_listen("002271")) # print(get_listen_code_pos("002272")) - code= get_listen_code_by_pos(2, 7) - print(code) \ No newline at end of file + code = get_listen_code_by_pos(2, 7) + print(code) -- Gitblit v1.8.0