From 1dea3d72da9d3d2d55f2e41b611913b0901d8074 Mon Sep 17 00:00:00 2001
From: admin <admin@example.com>
Date: 星期二, 17 六月 2025 19:15:19 +0800
Subject: [PATCH] 增加测试/记录精选流入板块

---
 strategy/forbidden_plates_manager.py |   16 ++++++++--------
 1 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/strategy/forbidden_plates_manager.py b/strategy/forbidden_plates_manager.py
index 920ee96..9c8de9c 100644
--- a/strategy/forbidden_plates_manager.py
+++ b/strategy/forbidden_plates_manager.py
@@ -1,6 +1,8 @@
 """
 榛戝悕鍗曟澘鍧楃鐞�
 """
+import json
+
 from db import redis_manager_delegate as redis_manager
 from db.redis_manager_delegate import RedisUtils
 from utils import tool
@@ -9,15 +11,15 @@
 @tool.singleton
 class ForbiddenPlatesManager:
     def __init__(self):
-        self.forbidden_plates = set()
+        self.forbidden_plates = []
         self.__db = 1
         self.__redisManager = redis_manager.RedisManager(self.__db)
         self.__load_forbidden_plates()
 
     def __load_forbidden_plates(self):
-        plates = RedisUtils.smembers(self.__redisManager.getRedis(), "forbidden_plates")
+        plates = RedisUtils.get(self.__redisManager.getRedis(), "forbidden_plate_list")
         if plates:
-            self.forbidden_plates = set(plates)
+            self.forbidden_plates = json.loads(plates)
 
     def add_plate(self, plate):
         """
@@ -27,9 +29,8 @@
         """
         if plate in self.forbidden_plates:
             return
-        self.forbidden_plates.add(plate)
-        RedisUtils.sadd_async(self.__db, "forbidden_plates", plate)
-        RedisUtils.expire_async(self.__db, "forbidden_plates", tool.get_expire())
+        self.forbidden_plates.append(plate)
+        RedisUtils.setex_async(self.__db, "forbidden_plate_list",tool.get_expire(), json.dumps( self.forbidden_plates))
 
     def remove_plate(self, plate):
         """
@@ -40,8 +41,7 @@
         if plate not in self.forbidden_plates:
             return
         self.forbidden_plates.remove(plate)
-        RedisUtils.srem_async(self.__db, "forbidden_plates", plate)
-        RedisUtils.expire_async(self.__db, "forbidden_plates", tool.get_expire())
+        RedisUtils.setex_async(self.__db, "forbidden_plate_list", tool.get_expire(), json.dumps(self.forbidden_plates))
 
     def list_plates(self):
         """

--
Gitblit v1.8.0