From e5c8cf73f8fe0209faad73af74e9d119e681a3ec Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期一, 21 十月 2024 17:29:41 +0800
Subject: [PATCH] 精选流出修改/修改扫入规则的量比
---
code_attribute/gpcode_manager.py | 97 ++++++++++++++++++++++++++++++++++--------------
1 files changed, 68 insertions(+), 29 deletions(-)
diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index c09906c..98f25de 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -36,7 +36,7 @@
def get_code_name(cls, code):
if code in cls.__code_name_dict:
return cls.__code_name_dict[code]
- fresults = cls.__mysqldb.select_one(f"select code_name from code_name where code = {code}")
+ fresults = cls.__mysqldb.select_one(f"select code_name from code_name where code = '{code}'")
if fresults:
cls.__code_name_dict[code] = fresults[0]
return fresults[0]
@@ -45,11 +45,11 @@
@classmethod
def add_code_name(cls, code, name):
cls.__code_name_dict[code] = name
- fresults = cls.__mysqldb.select_one(f"select code_name from code_name where code = {code}")
+ fresults = cls.__mysqldb.select_one(f"select code_name from code_name where code = '{code}'")
if fresults:
- cls.__mysqldb.execute(f"update code_name set code_name ='{name}', update_time = now() where code= {code}")
+ cls.__mysqldb.execute(f"update code_name set code_name ='{name}', update_time = now() where code= '{code}'")
else:
- cls.__mysqldb.execute(f"insert into code_name(code,code_name,update_time ) values({code},'{name}',now())")
+ cls.__mysqldb.execute(f"insert into code_name(code,code_name,update_time ) values('{code}','{name}',now())")
@classmethod
def add_first_code_name(cls, code, name):
@@ -69,8 +69,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):
@@ -95,27 +93,6 @@
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
# 鎯宠涔扮殑浠ｇ爜
@@ -385,6 +362,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 = {}
@@ -595,8 +628,7 @@
# 璁剧疆鏀剁洏浠�
@classmethod
def set_price_pre(cls, code, price, force=False):
- codes = get_gp_list()
- if code not in codes and not FirstCodeManager().is_in_first_record_cache(code) and not force:
+ if code in cls.__price_pre_cache and not force:
return
price = round(float(price), 2)
logger_pre_close_price.info(f"{code}-{price}")
@@ -619,6 +651,13 @@
return limit_up_price
+def get_limit_up_price_as_num(code):
+ limit_up_price = get_limit_up_price(code)
+ if limit_up_price:
+ return round(float(limit_up_price), 2)
+ return None
+
+
def get_limit_up_price_cache(code):
if code in __limit_up_price_dict:
return __limit_up_price_dict[code]
--
Gitblit v1.8.0