From 52c4cb753226d69e9ca04ff73ffa6e8ba79dd917 Mon Sep 17 00:00:00 2001 From: Administrator <admin@example.com> Date: 星期三, 06 十二月 2023 16:36:35 +0800 Subject: [PATCH] 强势时间段调整 --- code_attribute/gpcode_manager.py | 104 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 103 insertions(+), 1 deletions(-) diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py index 1b63305..87e08a3 100644 --- a/code_attribute/gpcode_manager.py +++ b/code_attribute/gpcode_manager.py @@ -2,7 +2,6 @@ 鑲$エ浠g爜绠$悊鍣� """ import json -import random import time import constant @@ -295,6 +294,109 @@ return self.__pause_buy_codes_cache +class WhiteListCodeManager: + __instance = None + __redis_manager = redis_manager.RedisManager(2) + + def __new__(cls, *args, **kwargs): + if not cls.__instance: + cls.__instance = super(WhiteListCodeManager, cls).__new__(cls, *args, **kwargs) + # 鍒濆鍖栬缃� + # 鑾峰彇浜ゆ槗绐楀彛鐨勯攣 + cls.__instance.__white_codes_cache = RedisUtils.smembers(cls.__get_redis(), "white_list_codes") + + return cls.__instance + + @classmethod + def __get_redis(cls): + return cls.__redis_manager.getRedis() + + def sync(self): + data = RedisUtils.smembers(self.__get_redis(), "white_list_codes") + self.__white_codes_cache.clear() + if data: + self.__white_codes_cache |= data + + def add_code(self, code): + self.__white_codes_cache.add(code) + RedisUtils.sadd(self.__get_redis(), "white_list_codes", code) + RedisUtils.expire(self.__get_redis(), "white_list_codes", tool.get_expire()) + + def remove_code(self, code): + self.__white_codes_cache.discard(code) + RedisUtils.srem(self.__get_redis(), "white_list_codes", code) + + def is_in(self, code): + return RedisUtils.sismember(self.__get_redis(), "white_list_codes", code) + + def is_in_cache(self, code): + return code in self.__white_codes_cache + + def list_codes(self): + return RedisUtils.smembers(self.__get_redis(), "white_list_codes") + + def list_codes_cache(self): + return self.__white_codes_cache + + def clear(self): + self.__white_codes_cache.clear() + RedisUtils.delete(self.__get_redis(), "white_list_codes") + + +class BlackListCodeManager: + __instance = None + __db = 2 + __redis_manager = redis_manager.RedisManager(2) + + def __new__(cls, *args, **kwargs): + if not cls.__instance: + cls.__instance = super(BlackListCodeManager, cls).__new__(cls, *args, **kwargs) + # 鍒濆鍖栬缃� + # 鑾峰彇浜ゆ槗绐楀彛鐨勯攣 + cls.__instance.__forbidden_trade_codes_cache = RedisUtils.smembers(cls.__get_redis(), + "forbidden-trade-codes") + + return cls.__instance + + @classmethod + def __get_redis(cls): + return cls.__redis_manager.getRedis() + + def add_code(self, code): + self.__forbidden_trade_codes_cache.add(code) + RedisUtils.sadd_async(self.__db, "forbidden-trade-codes", code) + RedisUtils.expire_async(self.__db, "forbidden-trade-codes", tool.get_expire()) + + def sync(self): + data = RedisUtils.smembers(self.__get_redis(), + "forbidden-trade-codes") + self.__forbidden_trade_codes_cache.clear() + if data: + self.__forbidden_trade_codes_cache |= data + + def remove_code(self, code): + self.__forbidden_trade_codes_cache.discard(code) + RedisUtils.srem(self.__get_redis(), "forbidden-trade-codes", code) + + def is_in(self, code): + return RedisUtils.sismember(self.__get_redis(), "forbidden-trade-codes", code) + + def is_in_cache(self, code): + return code in self.__forbidden_trade_codes_cache + + def list_codes(self): + codes = RedisUtils.smembers(self.__get_redis(), "forbidden-trade-codes") + self.__forbidden_trade_codes_cache = codes + return codes + + def list_codes_cache(self): + return self.__forbidden_trade_codes_cache + + def clear(self): + self.__forbidden_trade_codes_cache.clear() + RedisUtils.delete(self.__get_redis(), "forbidden-trade-codes") + + def __parse_codes_data(code_datas): codes = [] name_codes = {} -- Gitblit v1.8.0