"""
|
目标代码管理
|
"""
|
import decimal
|
|
from code_attribute.history_k_data_util import JueJinHttpApi
|
from log_module.log import printlog
|
from utils import tool
|
|
# 有效代码的缓存
|
# 数据格式：{"日期":[]}
|
__valid_codes_cache = {}
|
# 正股到可转债代码的索引
|
__valid_underlying_code_map_cache = {}
|
|
# 可转债代码到正股代码的索引
|
__valid_cb_to_underlying_code_map_cache = {}
|
|
# 可转债昨日收盘价
|
__valid_pre_close_price_map_cache = {}
|
|
|
def __request_valid_codes_info():
|
"""
|
请求有效的代码信息
|
:return:
|
"""
|
results = JueJinHttpApi.get_exchanges_codes("SHSE,SZSE", sec_types=[8], skip_suspended=True, skip_st=True,
|
fields="symbol, sec_type, sec_id,sec_name, underlying_symbol, delisted_date, sec_level, pre_close")
|
fresults = []
|
for r in results:
|
if r['sec_name'].find('转') == -1 or r['sec_level'] != 1 or int(tool.get_now_date_str('%Y%m%d')) >= int(
|
r['delisted_date'].strftime('%Y%m%d')):
|
continue
|
fresults.append(r)
|
# 排除同一股票有2个转债
|
cb_count_dict = {}
|
exclude_codes = []
|
for r in fresults:
|
if r['underlying_symbol'] not in cb_count_dict:
|
cb_count_dict[r['underlying_symbol']] = []
|
cb_count_dict[r['underlying_symbol']].append(r['symbol'])
|
if len(cb_count_dict[r['underlying_symbol']]) > 1:
|
printlog(r['underlying_symbol'], cb_count_dict[r['underlying_symbol']])
|
exclude_codes.extend(cb_count_dict[r['underlying_symbol']])
|
ffresults = []
|
for x in fresults:
|
if x['symbol'] in exclude_codes:
|
continue
|
ffresults.append(x)
|
return ffresults
|
|
|
def load_valid_codes_info(day=tool.get_now_date_str()):
|
"""
|
加载有效数据
|
:param day:
|
:return:
|
"""
|
if day in __valid_codes_cache:
|
return __valid_codes_cache.get(day)
|
results = __request_valid_codes_info()
|
if results:
|
__valid_codes_cache[day] = results
|
__valid_underlying_code_map_cache[day] = {}
|
__valid_pre_close_price_map_cache[day] = {}
|
__valid_cb_to_underlying_code_map_cache[day] = {}
|
for r in results:
|
__valid_underlying_code_map_cache[day][r['underlying_symbol'].split('.')[1]] = r['sec_id']
|
__valid_cb_to_underlying_code_map_cache[day][r['sec_id']] = r['underlying_symbol'].split('.')[1]
|
__valid_pre_close_price_map_cache[day][r['sec_id']] = r['pre_close']
|
|
|
def get_subscript_underlying_codes():
|
"""
|
获取需要订阅的正股代码
|
:return:
|
"""
|
day = tool.get_now_date_str()
|
if not __valid_codes_cache.get(day):
|
load_valid_codes_info(tool.get_now_date_str())
|
ffresults = __valid_codes_cache.get(tool.get_now_date_str())
|
return [x['underlying_symbol'].split('.')[1] for x in ffresults]
|
|
|
def get_subscript_cb_codes():
|
"""
|
获取需要订阅的代码
|
:return:
|
"""
|
day = tool.get_now_date_str()
|
if not __valid_codes_cache.get(day):
|
load_valid_codes_info(tool.get_now_date_str())
|
ffresults = __valid_codes_cache.get(tool.get_now_date_str())
|
return [x['sec_id'] for x in ffresults]
|
|
|
def get_underlying_to_cb_code_map():
|
"""
|
获取股票代码-可转债代码的map
|
:param code:
|
:return:
|
"""
|
day = tool.get_now_date_str()
|
if day not in __valid_underlying_code_map_cache:
|
load_valid_codes_info(day)
|
|
return __valid_underlying_code_map_cache.get(day)
|
|
|
def get_cb_to_underlying_code_map():
|
"""
|
获取可转债代码-股票代码的map
|
:param code:
|
:return:
|
"""
|
day = tool.get_now_date_str()
|
if day not in __valid_cb_to_underlying_code_map_cache:
|
load_valid_codes_info(day)
|
|
return __valid_cb_to_underlying_code_map_cache.get(day)
|
|
|
def get_cb_code(code):
|
"""
|
获取可转债代码
|
:param code:
|
:return:
|
"""
|
map = get_underlying_to_cb_code_map()
|
if map:
|
return map.get(code)
|
return None
|
|
|
def get_underlying_code(code):
|
map = get_cb_to_underlying_code_map()
|
if map:
|
return map.get(code)
|
return None
|
|
|
def get_limit_up_price(cb_code):
|
"""
|
获取可转债的涨停价
|
:param cb_code:
|
:return:
|
"""
|
day = tool.get_now_date_str()
|
pre_close_price_map = __valid_pre_close_price_map_cache.get(day)
|
if not pre_close_price_map:
|
return None
|
price_close = pre_close_price_map.get(cb_code)
|
price_close = tool.to_price(decimal.Decimal(str(price_close)))
|
|
return str(tool.to_price(price_close * decimal.Decimal("1.2"), "0.000"))
|
|
|
def get_limit_down_price(cb_code):
|
"""
|
获取可转债的跌停价
|
:param cb_code:
|
:return:
|
"""
|
day = tool.get_now_date_str()
|
pre_close_price_map = __valid_pre_close_price_map_cache.get(day)
|
if not pre_close_price_map:
|
return None
|
price_close = pre_close_price_map.get(cb_code)
|
price_close = tool.to_price(decimal.Decimal(str(price_close)))
|
|
return str(tool.to_price(price_close * decimal.Decimal("0.8"), "0.000"))
|
|
|
|
if __name__ == '__main__':
|
load_valid_codes_info()
|
print(get_limit_up_price("110060"), get_limit_down_price("110060"))
|
