From 5afbce25e8903e4ceae59abf393b844d23fd2d45 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 24 十二月 2024 17:06:53 +0800
Subject: [PATCH] 手动移黑机制修改

---
 code_attribute/gpcode_manager.py |   39 +++++++++++++++++++++++++++++++++++++--
 1 files changed, 37 insertions(+), 2 deletions(-)

diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index 501d851..eeffff7 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -210,6 +210,42 @@
         return self.__want_buy_codes_cache
 
 
+@tool.singleton
+class HumanRemoveForbiddenManager:
+    """
+    璁や负绉婚粦绠$悊
+    """
+    __db = 0
+    redisManager = redis_manager.RedisManager(0)
+    __redis_key = "human_remove_forbidden_codes"
+
+    __codes_cache = set()
+
+    def __init__(self):
+        self.__load_data()
+
+    @classmethod
+    def __get_redis(cls):
+        return cls.redisManager.getRedis()
+
+    def __load_data(self):
+        self.__codes_cache = RedisUtils.smembers(self.__get_redis(), self.__redis_key)
+        if self.__codes_cache is None:
+            self.__codes_cache = set()
+
+    def add_code(self, code):
+        self.__codes_cache.add(code)
+        RedisUtils.sadd_async(self.__db, self.__redis_key, code)
+        RedisUtils.expire_async(self.__db, self.__redis_key, tool.get_expire())
+
+    def remove_code(self, code):
+        self.__codes_cache.discard(code)
+        RedisUtils.srem_async(self.__db, self.__redis_key, code)
+
+    def is_in_cache(self, code):
+        return code in self.__codes_cache
+
+
 # 鏆傚仠涓嬪崟浠g爜绠$悊
 # 涓庨粦鍚嶅崟鐨勫尯鍒槸鏆傚仠浜ゆ槗浠g爜鍙槸涓嶄氦鏄擄紝涓嶈兘绉婚櫎L2鐩戞帶浣�
 class PauseBuyCodesManager:
@@ -868,5 +904,4 @@
         RedisUtils.delete(redis_instance, "first_code_record", auto_free=False)
         RedisUtils.delete(redis_instance, "first_code_limited_up_record", auto_free=False)
     finally:
-        RedisUtils.realse(redis_instance)
-
+        RedisUtils.realse(redis_instance)
\ No newline at end of file

--
Gitblit v1.8.0