From 2c707e6083d8d60b088a7f91a0020370e65c8c7f Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 29 十月 2024 17:10:24 +0800
Subject: [PATCH] 昨日跌幅过大不扫入/添加日志记录
---
code_attribute/gpcode_manager.py | 95 +++++++++++++++++++++++++++++++++++------------
1 files changed, 70 insertions(+), 25 deletions(-)
diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index ff4f641..64bd83d 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -22,6 +22,17 @@
class CodesNameManager:
__mysqldb = Mysqldb()
__code_name_dict = {}
+ __instance = None
+
+ def __new__(cls, *args, **kwargs):
+ if not cls.__instance:
+ cls.__instance = super(CodesNameManager, cls).__new__(cls, *args, **kwargs)
+ cls.__load_data()
+ return cls.__instance
+
+ @classmethod
+ def __load_data(cls):
+ cls.__code_name_dict = cls.list_code_name_dict()
@classmethod
def list_code_name_dict(cls):
@@ -69,8 +80,6 @@
cls.__instance.redisManager = redis_manager.RedisManager(0)
cls.__instance.__first_code_record_cache = RedisUtils.smembers(cls.__instance.__get_redis(),
"first_code_record")
- cls.__instance.__first_code_limited_up_record_cache = RedisUtils.smembers(cls.__instance.__get_redis(),
- "first_code_limited_up_record")
return cls.__instance
def __get_redis(self):
@@ -96,31 +105,11 @@
def is_in_first_record_cache(self, code):
return code in self.__first_code_record_cache
- # 鍔犲叆棣栨澘娑ㄥ仠杩囦唬鐮侀泦鍚�
- def add_limited_up_record(self, codes):
- hasChanged = False
- for code in codes:
- if code not in self.__first_code_limited_up_record_cache:
- RedisUtils.sadd_async(self.__db, "first_code_limited_up_record", code)
- hasChanged = True
- self.__first_code_limited_up_record_cache.add(code)
- if hasChanged:
- RedisUtils.expire_async(self.__db, "first_code_limited_up_record", tool.get_expire())
-
- # 鏄惁娑ㄥ仠杩�
- def is_limited_up(self, code):
- if RedisUtils.sismember(self.__get_redis(), "first_code_limited_up_record", code):
- return True
- else:
- return False
-
- def is_limited_up_cache(self, code):
- return code in self.__first_code_limited_up_record_cache
-
# 鎯宠涔扮殑浠ｇ爜
class WantBuyCodesManager:
__instance = None
+ __db = 0
redisManager = redis_manager.RedisManager(0)
__redis_key = "want_buy_codes"
@@ -149,7 +138,7 @@
def remove_code(self, code):
self.__want_buy_codes_cache.discard(code)
- RedisUtils.srem(self.__get_redis(), self.__redis_key, code)
+ RedisUtils.srem_async(self.__db, self.__redis_key, code)
def sync(self):
codes = self.list_code()
@@ -385,6 +374,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_async(self.__db, "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 = {}
@@ -499,7 +544,7 @@
# 鑾峰彇浠ｇ爜鐨勫悕绉�
def get_code_name(code):
- return CodesNameManager.get_code_name(code)
+ return CodesNameManager().get_code_name(code)
def get_name_codes():
--
Gitblit v1.8.0