From 97ec997e855faee25f8f2e35734f87afa5af6f95 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 27 八月 2024 17:15:21 +0800
Subject: [PATCH] 9:25之前的订阅数据处理
---
code_attribute/gpcode_manager.py | 74 ++++++++++++++++++++++++++++++++++---
1 files changed, 68 insertions(+), 6 deletions(-)
diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index c09906c..3e1834b 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):
@@ -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 = {}
@@ -595,8 +651,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 +674,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