"""
|
股票代码管理器
|
"""
|
from code_attribute.history_k_data_util import HistoryKDatasUtils
|
from db import redis_manager_delegate as redis_manager
|
from db.redis_manager_delegate import RedisUtils
|
from utils import tool, constant
|
import decimal
|
|
__redisManager = redis_manager.RedisManager(constant.REDIS_DB)
|
__db = constant.REDIS_DB
|
|
|
class CodesNameManager:
|
__code_name_dict = {}
|
|
@classmethod
|
def set_code_name(cls, code, name):
|
cls.__code_name_dict[code] = name
|
|
@classmethod
|
def get_code_name(cls, code):
|
return cls.__code_name_dict.get(code)
|
|
@classmethod
|
def request_code_name(cls, code):
|
datas = HistoryKDatasUtils.get_gp_latest_info([code])
|
if datas:
|
name = datas[0]['sec_name']
|
cls.set_code_name(code, name)
|
|
|
# 获取二板代码
|
def get_second_gp_list():
|
codes = RedisUtils.smembers(__redisManager.getRedis(), "gp_list")
|
return codes
|
|
|
class CodePrePriceManager:
|
__price_pre_cache = {}
|
__redisManager = redis_manager.RedisManager(constant.REDIS_DB)
|
|
# 获取收盘价
|
@classmethod
|
def get_price_pre(cls, code):
|
result = RedisUtils.get(cls.__redisManager.getRedis(), "price-pre-{}".format(code))
|
if result is not None:
|
return float(result)
|
return None
|
|
# 获取缓存
|
@classmethod
|
def get_price_pre_cache(cls, code):
|
if code in cls.__price_pre_cache:
|
return float(cls.__price_pre_cache[code])
|
val = cls.get_price_pre(code)
|
if val:
|
cls.__price_pre_cache[code] = val
|
return val
|
|
# 设置收盘价
|
@classmethod
|
def set_price_pre(cls, code, price, force=False):
|
RedisUtils.setex(cls.__redisManager.getRedis(), "price-pre-{}".format(code), tool.get_expire(), str(price))
|
cls.__price_pre_cache[code] = float(price)
|
|
|
__limit_up_price_dict = {}
|
|
|
# 获取涨停价
|
def get_limit_up_price(code):
|
# 读取内存中的值
|
if code in __limit_up_price_dict:
|
return __limit_up_price_dict[code]
|
price = CodePrePriceManager.get_price_pre_cache(code)
|
if price is None:
|
return None
|
limit_up_price = tool.to_price(decimal.Decimal(str(price)) * decimal.Decimal(f"{tool.get_limit_up_rate(code)}"))
|
__limit_up_price_dict[code] = limit_up_price
|
return limit_up_price
|
|
def get_price_pre_cache(code):
|
price = CodePrePriceManager.get_price_pre_cache(code)
|
return price
|
|
|
def set_price_pre(code, price):
|
CodePrePriceManager.set_price_pre(code, price, force=True)
|
|
|
def request_price_pre(codes):
|
k_results = HistoryKDatasUtils.get_gp_latest_info(codes, "pre_close,sec_id")
|
for kr in k_results:
|
set_price_pre(kr['sec_id'], round(kr['pre_close'], 2))
|
|
|
def get_limit_up_price_cache(code):
|
if code in __limit_up_price_dict:
|
return __limit_up_price_dict[code]
|
return None
|
|
|
# 获取跌停价
|
def get_limit_down_price(code):
|
price = CodePrePriceManager.get_price_pre_cache(code)
|
if price is None:
|
return None
|
return tool.to_price(decimal.Decimal(str(price)) * decimal.Decimal(f"{tool.get_limit_down_rate(code)}"))
|
