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