From 178d119f552136d7e119ddab50a01e7f5d642186 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期四, 22 八月 2024 23:38:40 +0800
Subject: [PATCH] 绿名单 /股票板块修改
---
code_attribute/gpcode_manager.py | 56 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 56 insertions(+), 0 deletions(-)
diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index ff4f641..3e1834b 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -385,6 +385,62 @@
RedisUtils.delete(self.__get_redis(), "forbidden-trade-codes")
+class GreenListCodeManager:
+ """
+ 缁垮悕鍗曪細涔板叆鍗冲姞绾紝鎾ゅ崟涓嶇Щ绾�
+ """
+ __instance = None
+ __db = 2
+ __redis_manager = redis_manager.RedisManager(2)
+ __codes_set = set()
+
+ def __new__(cls, *args, **kwargs):
+ if not cls.__instance:
+ cls.__instance = super(GreenListCodeManager, cls).__new__(cls, *args, **kwargs)
+ # 鍒濆鍖栬缃�
+ # 鑾峰彇浜ゆ槗绐楀彛鐨勯攣
+ cls.__codes_set = RedisUtils.smembers(cls.__get_redis(), "green-trade-codes")
+ return cls.__instance
+
+ @classmethod
+ def __get_redis(cls):
+ return cls.__redis_manager.getRedis()
+
+ def add_code(self, code):
+ self.__codes_set.add(code)
+ RedisUtils.sadd_async(self.__db, "green-trade-codes", code)
+ RedisUtils.expire_async(self.__db, "green-trade-codes", tool.get_expire())
+
+ def sync(self):
+ data = RedisUtils.smembers(self.__get_redis(),
+ "green-trade-codes")
+ self.__codes_set.clear()
+ if data:
+ self.__codes_set |= data
+
+ def remove_code(self, code):
+ self.__codes_set.discard(code)
+ RedisUtils.srem(self.__get_redis(), "green-trade-codes", code)
+
+ def is_in(self, code):
+ return RedisUtils.sismember(self.__get_redis(), "green-trade-codes", code)
+
+ def is_in_cache(self, code):
+ return code in self.__codes_set
+
+ def list_codes(self):
+ codes = RedisUtils.smembers(self.__get_redis(), "green-trade-codes")
+ self.__codes_set = codes
+ return codes
+
+ def list_codes_cache(self):
+ return self.__codes_set
+
+ def clear(self):
+ self.__codes_set.clear()
+ RedisUtils.delete(self.__get_redis(), "green-trade-codes")
+
+
def __parse_codes_data(code_datas):
codes = []
name_codes = {}
--
Gitblit v1.8.0