B
|
ånhfÅ. ã @ s¾ d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
|
Z
|
m Z ddl m Z mZ ddlmZ G dd dZG d d
|
d
|
ZG d d d eZed krºe eddh¡Ze e dS )u
|
åå²K线æå¡
|
é N)Ú
|
RedisUtils)Úlogger_request_apiÚprintlog)ÚtoolÚmiddle_api_protocol)Úredis_manager_delegatec @ sp e Zd ZdZedd Zedd Zedd Zedd Zed
|
d Z ed d Z
|
edd Z edd Z dS )Ú JueJinHttpApiz http://193.112.35.168:10009/c C s2 dd }d}d }|rjt ||¡}t ¡ }zt |¡}W d t ¡ | }|dkrft d| dt| ¡ X n@| j | }tj ||d} | j
|
}
|
t |
|
¡} | d d krª| d
|
}|r*t |tkrøxh|D ]0} t | tkrÔqÂx | D ]} || | | | < qÚW qÂW n.t |tkr&x |D ]} ||| || < q W |S d S d S ) Nc S sF t | tkrB| d¡dkrB| d¡dkrB| d¡dkrBtj | ¡S | S )NÚTéÿÿÿÿú:ú+)ÚtypeÚstrÚfindÚdatetimeÚ fromisoformat)Úval© r ú=D:\workspace\trade_sell\code_atrribute\history_k_data_util.pyÚ deformat_date s ,
|
z.JueJinHttpApi.__request.<locals>.deformat_dateTé u æéAPI请æ±æ¶é´ï¼ú-)ÚjsonÚcoder Údata)r Ú load_juejinÚtimeÚrequestr ÚinfoÚintÚ_JueJinHttpApi__BASE_URLÚrequestsÚpostÚtextr Úloadsr ÚlistÚdict)ÚclsÚpath_strÚ data_jsonr ZDELEGATEÚfdataZ_JueJinHttpApi__start_timeZ_JueJinHttpApi__use_timeÚurlÚresponseÚresultZ
|
resultJsonÚdÚkr r r Z __request s:
|
|
|
|
zJueJinHttpApi.__requestc C s | d||d¡S )NÚget_instruments)ÚsymbolsÚfields)Ú_JueJinHttpApi__request)r' r1 r2 r r r r0 B s z JueJinHttpApi.get_instrumentsc C s | d|||||d¡S )NÚ history_n)ÚsymbolÚ frequencyÚcountÚadjustr2 )r3 )r' r5 r6 r7 r8 r2 r r r r4 F s zJueJinHttpApi.history_nc C s | d||d¡S )NÚcurrent)r1 r2 )r3 )r' r1 r2 r r r r9 K s zJueJinHttpApi.currentc C s | d|||||d¡S )Nr0 )Ú exchangesÚ sec_typesÚskip_suspendedÚskip_str2 )r3 )r' r: r; r< r= r2 r r r Úget_exchanges_codesO s z!JueJinHttpApi.get_exchanges_codesc C s | d||d¡S )NÚget_previous_trading_date)ÚexchangeÚdate)r3 )r' r@ rA r r r r? U s z'JueJinHttpApi.get_previous_trading_datec C s | d||d¡S )NÚget_next_trading_date)r@ rA )r3 )r' r@ rA r r r rB Y s z#JueJinHttpApi.get_next_trading_datec C s | d|||d¡S )NÚget_trading_dates)r@ Ú
|
start_dateÚend_date)r3 )r' r@ rD rE r r r rC ] s z JueJinHttpApi.get_trading_datesN) Ú__name__Ú
|
__module__Ú __qualname__r Ú classmethodr3 r0 r4 r9 r> r? rB rC r r r r r s * r c @ s e Zd Ze d¡Zedd Zedd ZedddZ edd d
|
Z
|
ed d Z ed d Z edd Z edd Zedd ZdS )Ú JueJinApir c C sV | j ¡ }z:tj|ddd}tj|ddd}tj|ddd}|||fS t |¡ X d S )Nzjuejin-account-idF)Z auto_freezjuejin-strategy-idz juejin-token)Ú_JueJinApi__redisManagerÚgetRedisr ÚgetZrealse)r' ÚredisÚ
|
account_idZ strategy_idÚtokenr r r ÚgetJueJinAccountInfog s
|
|
z JueJinApi.getJueJinAccountInfoc C sL g }xB|D ]:}t |¡r*| d |¡¡ q
|
t |¡r
|
| d |¡¡ q
|
W |S )NzSZSE.{}zSHSE.{})r Ú
|
is_sz_codeÚappendÚformatÚ
|
is_sh_code)r' Úcodesr% r. r r r Ú get_juejin_code_list_with_prefixr s
|
|
|
z*JueJinApi.get_juejin_code_list_with_prefixNc C s` |sg S | |¡}tjrH| ¡ \}}}t |¡ tjd |¡|d}|S tjd |¡|dS d S )Nú,)r1 r2 ) rW ÚconstantÚJUEJIN_LOCAL_APIrQ ÚgmapiÚ set_tokenr0 Újoinr )r' rV r2 r1 rO Ús_idrP r r r r Úget_gp_latest_info| s
|
|
z JueJinApi.get_gp_latest_infoc C sf | |g¡}tjrF| ¡ \}}}t |¡ tj|d d|d|d}|S tj|d d|d|d}|S d S )Nr Z1dé )r5 r6 r7 r8 r2 )rW rY rZ rQ r[ r\ r4 r ) r' r r7 r2 r1 rO r^ rP Úresultsr r r Úget_history_tick_n s
|
z JueJinApi.get_history_tick_nc C sb |sg S | |¡}tjrF| ¡ \}}}t |¡ tjd |¡d}|S tjd |¡dd}|S d S )NrX )r1 Ú )r1 r2 ) rW rY rZ rQ r[ r\ r9 r] r )r' rV r1 rO r^ rP r r r r Úget_gp_current_info s
|
|
z JueJinApi.get_gp_current_infoc C sN t jr4| ¡ \}}}t |¡ tj|dgddddS tj|dgddddS d S )Nr` TzLsymbol,sec_type,sec_id,sec_name,listed_date,sec_level,is_suspended,pre_close)r: r; r< r= r2 )rY rZ rQ r[ r\ r0 r r> )r' r: rO r^ rP r r r r> ¦ s
|
z JueJinApi.get_exchanges_codesc C s: t jr*| ¡ \}}}t |¡ t d|¡S t d|¡S d S )NÚSHSE)rY rZ rQ r[ r\ r? r )r' rA rO r^ rP r r r r? ² s
|
|
z#JueJinApi.get_previous_trading_datec C s: t jr*| ¡ \}}}t |¡ t d|¡S t d|¡S d S )Nre )rY rZ rQ r[ r\ rB r )r' rA rO r^ rP r r r rB ¼ s
|
|
z JueJinApi.get_next_trading_datec C s> t jr,| ¡ \}}}t |¡ t d||¡S t d||¡S d S )Nre )rY rZ rQ r[ r\ rC r )r' rD rE rO r^ rP r r r rC Å s
|
|
zJueJinApi.get_trading_dates)N)N)rF rG rH Ú redis_managerÚ RedisManagerrK rI rQ rW r_ rb rd r> r? rB rC r r r r rJ c s
|
|
|
rJ c @ s´ e Zd Zi Zi Zed ddZed ddZedd Zedd Z ed
|
d Z
|
ed d Z edd Z edd Z edd Zedd Zedd Zedd Zedd ZdS ) ÚHistoryKDatasUtilsNc C s t ||¡S )N)rJ r_ )r' rV r2 r r r r_ Ó s z%HistoryKDatasUtils.get_gp_latest_infoc C s t |||¡S )N)rJ rb )r' r r7 r2 r r r rb × s z%HistoryKDatasUtils.get_history_tick_nc C s
|
t |¡S )N)rJ rd )r' rV r r r rd Û s z&HistoryKDatasUtils.get_gp_current_infoc C s
|
t |¡S )N)rJ r? )r' rA r r r r? à s z,HistoryKDatasUtils.get_previous_trading_datec C s2 || j kr| j |¡S | |¡}|r.|| j |< |S )N)Ú0_HistoryKDatasUtils__previous_trading_date_cacherM r? )r' rA r* r r r Ú get_previous_trading_date_cacheä s
|
|
|
z2HistoryKDatasUtils.get_previous_trading_date_cachec C s: t ¡ }g }x(t|D ] }| |¡}| |¡ |}qW |S )N)r Úget_now_date_strÚrangerj rS )r' Ú day_countZnow_dayÚdaysÚiZpdayr r r Úget_latest_trading_dateî s
|
|
z*HistoryKDatasUtils.get_latest_trading_datec C s> t ¡ d| }|| jkr&| j| S | |¡}|| j|< |S )Nr )r rk Ú._HistoryKDatasUtils__latest_trading_date_cacherp )r' rm Úkeyrn r r r Ú get_latest_trading_date_cacheø s
|
|
|
|
z0HistoryKDatasUtils.get_latest_trading_date_cachec C s
|
t |¡S )N)rJ rB )r' rA r r r rB s z(HistoryKDatasUtils.get_next_trading_datec C s t ||¡S )N)rJ rC )r' rD rE r r r rC s z$HistoryKDatasUtils.get_trading_datesc C sH | |¡}g }x4|D ],}|d d¡d }|d }| ||f¡ qW |S )Nr5 Ú.r` Úprice)rd ÚsplitrS )r' rV r ZpricesÚitemr ru r r r Ú get_now_price
|
s
|
|
z HistoryKDatasUtils.get_now_pricec C s¾ | |¡}i }x4|D ],}|d }t t t|d ¡¡}|||< qW | |¡}g }xN|D ]F}|d }|d } t| |¡}
|
t | |
|
d |
|
d} |
|
|| f¡ qXW |j dd d | ¡ |S )
|
NÚsec_idÚ pre_closer r` éd é c S s | d S )Nr` r )Útupr r r Ú<lambda>& ó z9HistoryKDatasUtils.get_codes_limit_rate.<locals>.<lambda>)rr ) r_ r Úto_priceÚdecimalÚDecimalr rx ÚfloatrM ÚroundrS ÚsortÚreverse) r' rV ÚdatasZpre_price_dictr r rz Z
|
now_pricesZ f_resultsru Z pre_priceÚrater r r Úget_codes_limit_rate s"
|
|
|
|
z'HistoryKDatasUtils.get_codes_limit_ratec C sN | ||¡}|d d }x |D ]}||d kr |d }q W |d d | | S )Nr Úcloser
|
)rb )r' r r7 r Z low_pricer r r r Úget_lowest_price_rate* s
|
z(HistoryKDatasUtils.get_lowest_price_ratec C sB | |¡}i }x.|D ]&}|d d¡d }|d }|||< qW |S )Nr5 rt r` Úsec_name)r_ rv )r' rV r ra r r Ú code_namer r r Úget_gp_codes_names3 s
|
|
z%HistoryKDatasUtils.get_gp_codes_names)N)N)rF rG rH ri rq rI r_ rb rd r? rj rp rs rB rC rx r r r r r r r rh Ï s"
|
|
|
rh Ú__main__Z000422Z600610)Ú__doc__r r r r r! rY Údb.redis_manager_delegater Úlog_module.logr r Úutilsr r Údbr rf r rJ Úobjectrh rF r r% ra r r r r Ú<module> s Nlo
|
