From b51b2ae184fad5aaf37a78903987e064f192d430 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期一, 26 五月 2025 11:35:20 +0800
Subject: [PATCH] 大单解析修改

---
 code_attribute/gpcode_manager.py |   82 +++++++++++++++++++++++++++++++++++++++-
 1 files changed, 79 insertions(+), 3 deletions(-)

diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index 501d851..9ab6e76 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -9,7 +9,8 @@
 from db.mysql_data_delegate import Mysqldb
 from db.redis_manager_delegate import RedisUtils
 from log_module import log_export
-from log_module.log import logger_pre_close_price
+from log_module.log import logger_pre_close_price, logger_debug
+from trade import trade_record_log_util
 from utils import tool
 import decimal
 
@@ -210,6 +211,81 @@
         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
+
+
+@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:
@@ -383,6 +459,7 @@
             # 鑾峰彇浜ゆ槗绐楀彛鐨勯攣
             cls.__instance.__forbidden_trade_codes_cache = RedisUtils.smembers(cls.__get_redis(),
                                                                                "forbidden-trade-codes")
+            logger_debug.info(f"鍔犺浇鍔犻粦鍒楄〃锛歿cls.__instance.__forbidden_trade_codes_cache}")
 
         return cls.__instance
 
@@ -868,5 +945,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