From 1e607e90c08fa2b7a765360665cc392539e3ddec Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期五, 04 八月 2023 13:20:55 +0800
Subject: [PATCH] redis连接池归还

---
 third_data/code_plate_key_manager.py |   40 +++++++++++++++++++++-------------------
 1 files changed, 21 insertions(+), 19 deletions(-)

diff --git a/third_data/code_plate_key_manager.py b/third_data/code_plate_key_manager.py
index a34dc57..80dbcb7 100644
--- a/third_data/code_plate_key_manager.py
+++ b/third_data/code_plate_key_manager.py
@@ -29,7 +29,7 @@
         if blocks is None:
             return
         # 淇濆瓨鍓�2鏉℃暟鎹�
-        RedisUtils.setex(self.__get_redis(),f"kpl_jx_blocks-{code}", tool.get_expire(), json.dumps(blocks))
+        RedisUtils.setex(self.__get_redis(), f"kpl_jx_blocks-{code}", tool.get_expire(), json.dumps(blocks))
         self.__code_blocks[code] = blocks
 
     # 鑾峰彇绮鹃�夋澘鍧�
@@ -92,7 +92,7 @@
     # 浠婃棩娑ㄥ仠鍘熷洜鍙樺寲
     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)
-        self.__get_redis().expire(f"kpl_limit_up_reason_his-{code}", tool.get_expire())
+        RedisUtils.expire(self.__get_redis(), f"kpl_limit_up_reason_his-{code}", tool.get_expire())
         self.__set_total_keys(code)
 
     # 璁剧疆浠g爜鐨勪粖鏃ユ定鍋滃師鍥�
@@ -259,7 +259,8 @@
 
     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)))
+        RedisUtils.setex(self.__get_redis(), f"kpl_his_limit_up_reason-{code}", tool.get_expire(),
+                         json.dumps(list(reasons)))
         logger_kpl_debug.debug(f"璁剧疆鍘嗗彶娑ㄥ仠鍘熷洜锛歿code}-{reasons}")
 
     # 濡傛灉杩斿洖鍊间笉涓篘one琛ㄧず宸茬粡鍔犺浇杩囧巻鍙插師鍥犱簡
@@ -267,7 +268,7 @@
         reasons = self.__history_limit_up_reason_dict.get(code)
         if reasons is None:
             # 浠庡唴瀛樹腑鍔犺浇
-            val = self.__get_redis().get(f"kpl_his_limit_up_reason-{code}")
+            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
@@ -280,13 +281,13 @@
 
     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)))
+        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 = self.__get_redis().get(f"kpl_blocks-{code}")
+            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
@@ -397,7 +398,8 @@
             for k in k2:
                 if k not in constant.KPL_INVALID_BLOCKS:
                     keys.add(k)
-        if not keys:
+        # 濮嬬粓鑾峰彇绮鹃�夋澘鍧�
+        if True:
             # 鑾峰彇
             jx_blocks = cls.__KPLCodeJXBlockManager.get_jx_blocks(code)
             if jx_blocks:
@@ -406,7 +408,7 @@
 
         # 娑ㄥ仠鍒楄〃涓尮閰嶅叧閿瘝锛岃繑鍥烇紙鏉垮潡:浠g爜闆嗗悎锛夛紝浠g爜闆嗗悎涓凡缁忔帓闄よ嚜韬�
         if not keys:
-            return cls.BLOCK_TYPE_NONE, None, "灏氭湭鎵惧埌娑ㄥ仠鍘熷洜"
+            return None, "灏氭湭鎵惧埌娑ㄥ仠鍘熷洜"
         code_limit_up_reason_dict = {}
         load_code_block()
         msg_list = []
@@ -421,11 +423,11 @@
             msg_list.append(f"\n瀹炴椂top10: {top_4_current}(娑ㄥ仠鏁伴噺锛歿len(current_limit_up_datas)})")
             msg_list.append(f"鍘嗗彶top20: {top_8_record}")
 
-            # 鑾峰彇涓绘澘瀹炴椂韬綅
+            # 鑾峰彇涓绘澘瀹炴椂韬綅,鍓旈櫎楂樹綅鏉�
             current_shsz_rank = kpl_block_util.get_code_current_rank(code, block, current_limit_up_datas,
-                                                                     code_limit_up_reason_dict, shsz=True)
+                                                                     code_limit_up_reason_dict, yesterday_current_limit_up_codes, shsz=True)
             record_shsz_rank = kpl_block_util.get_code_record_rank(code, block, limit_up_record_datas,
-                                                                   code_limit_up_reason_dict, shsz=True)
+                                                                   code_limit_up_reason_dict, yesterday_current_limit_up_codes, shsz=True)
             # 鑾峰彇涓绘澘鍘嗗彶韬綅
             if is_top_4:
                 pen_limit_up_codes = kpl_block_util.get_shsz_open_limit_up_codes(code, block, limit_up_record_datas,
@@ -436,7 +438,7 @@
                             pen_limit_up_codes) + 1:
                         # 灞炰簬榫�1,榫�2
                         can_buy_blocks.append((block,
-                                               f"{block}锛歵op4娑ㄥ仠鏉垮潡锛屼富鏉垮紑1({pen_limit_up_codes}),灞炰簬涓绘澘鍓嶉緳{len(pen_limit_up_codes) + 1}(瀹炴椂韬綅-{current_shsz_rank})"))
+                                               f"{block}锛歵op10娑ㄥ仠鏉垮潡锛屼富鏉垮紑1({pen_limit_up_codes}),灞炰簬涓绘澘鍓嶉緳{len(pen_limit_up_codes) + 1}(瀹炴椂韬綅-{current_shsz_rank})"))
                         continue
                     else:
                         msg_list.append(
@@ -467,7 +469,7 @@
                             f"鏉垮潡-{block}: 涓嶆槸top4娑ㄥ仠鏉垮潡锛屾弧瓒崇簿閫�/琛屼笟娴佸叆瑕佹眰,涓嶄负涓绘澘榫�1锛堝疄鏃惰韩浣�-{current_shsz_rank},鍘嗗彶韬綅-{record_shsz_rank}锛�")
                         continue
         if len(can_buy_blocks) == len(keys):
-            blocks = "/".join([x[0] for x in can_buy_blocks])
+            blocks = [x[0] for x in can_buy_blocks]
             blocks_msg = "\n".join([x[1] for x in can_buy_blocks])
             return blocks, blocks_msg
 
@@ -481,10 +483,10 @@
         if constant.TEST:
             return True, "", cls.BLOCK_TYPE_NONE
 
-        block, block_msg = cls.get_can_buy_block(code, current_limit_up_datas,
-                                                 limit_up_record_datas, yesterday_current_limit_up_codes,
-                                                 before_blocks_dict)
-        if block is None:
+        blocks, block_msg = cls.get_can_buy_block(code, current_limit_up_datas,
+                                                  limit_up_record_datas, yesterday_current_limit_up_codes,
+                                                  before_blocks_dict)
+        if not blocks:
             return False, block_msg
 
         # ---------------------------------鍒ゆ柇鐩爣浠g爜鐨勬澘鍧�-------------------start------------
@@ -565,11 +567,11 @@
         # ---------------------------------鍔犺浇宸茬粡涓嬪崟/鎴愪氦鐨勪唬鐮佷俊鎭�------------end-------------
 
         msg_list = []
-        for key in [block]:
+        for key in blocks:
             # 鏉垮潡涓凡缁忔湁鎴愪氦鐨勫氨涓嶄笅鍗曚簡
             if key in trade_success_blocks_count:
                 success_codes_count = len(trade_success_blocks_count[key])
-                if success_codes_count >= 2:
+                if success_codes_count >= 1:
                     msg_list.append(f"銆恵key}銆戜腑宸茬粡鏈墈success_codes_count}涓垚浜や唬鐮�")
                     log.logger_kpl_debug.debug(f"{code}锛氭澘鍧楋紙{key}锛夊凡缁忔湁鎴愪氦銆恵trade_success_blocks_count[key]}銆�")
                     continue

--
Gitblit v1.8.0