From d957e0dc6dcfb15304a6499e1b9ddee4e18be531 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期四, 02 一月 2025 18:10:23 +0800
Subject: [PATCH] 拆单初步规划/禁止交易板块管理

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

diff --git a/third_data/code_plate_key_manager.py b/third_data/code_plate_key_manager.py
index 6d41e73..12fcdb5 100644
--- a/third_data/code_plate_key_manager.py
+++ b/third_data/code_plate_key_manager.py
@@ -222,7 +222,10 @@
 
 # 寮�鐩樺暒绂佹浜ゆ槗鏉垮潡绠$悊
 class KPLPlateForbiddenManager:
-    __redisManager = redis_manager.RedisManager(3)
+    """
+    涓嶈兘涔扮殑鏉垮潡绠$悊
+    """
+    __redis_manager = redis_manager.RedisManager(3)
     __kpl_forbidden_plates_cache = set()
 
     __instance = None
@@ -237,17 +240,22 @@
     def __load_datas(cls):
         __redis = cls.__get_redis()
         try:
-            __kpl_forbidden_plates_cache = RedisUtils.smembers(__redis, "kpl_forbidden_plates")
+            cls.__kpl_forbidden_plates_cache = RedisUtils.smembers(__redis, "kpl_forbidden_plates")
         finally:
             RedisUtils.realse(__redis)
 
     @classmethod
     def __get_redis(cls):
-        return cls.__redisManager.getRedis()
+        return cls.__redis_manager.getRedis()
 
     def save_plate(self, plate):
         self.__kpl_forbidden_plates_cache.add(plate)
         RedisUtils.sadd(self.__get_redis(), "kpl_forbidden_plates", plate)
+        RedisUtils.expire(self.__get_redis(), "kpl_forbidden_plates", tool.get_expire())
+
+    def delete_plate(self, plate):
+        self.__kpl_forbidden_plates_cache.discard(plate)
+        RedisUtils.srem(self.__get_redis(), "kpl_forbidden_plates", plate)
         RedisUtils.expire(self.__get_redis(), "kpl_forbidden_plates", tool.get_expire())
 
     def list_all(self):
@@ -255,6 +263,11 @@
 
     def list_all_cache(self):
         return self.__kpl_forbidden_plates_cache
+
+    def is_in_cache(self, plate):
+        if self.__kpl_forbidden_plates_cache and plate in self.__kpl_forbidden_plates_cache:
+            return True
+        return False
 
 
 class LimitUpCodesPlateKeyManager:
@@ -416,7 +429,7 @@
                     strong = cls.get_market_strong()
                     if strong is None:
                         strong = 60
-                    THRESHOLD_MONEY = int((1 - strong/200) * data[3])
+                    THRESHOLD_MONEY = int((1 - strong / 200) * data[3])
             # if count >= MAX_COUNT:
             #     break
         # 璁板綍绮鹃�夋祦鍑烘棩蹇�

--
Gitblit v1.8.0