From 6056ef8795f5620f9a2ac4891e4c24526e6a7103 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 29 四月 2025 11:07:36 +0800
Subject: [PATCH] 人为拉黑的不能自动移黑

---
 code_attribute/gpcode_manager.py |   40 ++++++++++++++++++++++++++++++++++++++++
 1 files changed, 40 insertions(+), 0 deletions(-)

diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index e1cc686..9ab6e76 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -10,6 +10,7 @@
 from db.redis_manager_delegate import RedisUtils
 from log_module import log_export
 from log_module.log import logger_pre_close_price, logger_debug
+from trade import trade_record_log_util
 from utils import tool
 import decimal
 
@@ -246,6 +247,45 @@
         return code in self.__codes_cache
 
 
+@tool.singleton
+class HumanForbiddenManager:
+    """
+    浜轰负鎷夐粦绠$悊
+    """
+    __db = 0
+    redisManager = redis_manager.RedisManager(0)
+    __redis_key = "human_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):
+        trade_record_log_util.add_common_msg(code, "浜轰负鍔犻粦", f"")
+        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):
+        trade_record_log_util.add_common_msg(code, "浜轰负绉婚粦", f"")
+        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:

--
Gitblit v1.8.0