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