B
|
L KfØ. ã @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl
|
m Z m Z ddlm Z mZ ddlmZ G dd d ZG d
|
d d ZG d d d eZedkr¾e eddh¡Ze e dS )u
|
åå²K线æå¡
|
é N)Úconstant)Ú
|
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_cb\code_attribute\history_k_data_util.pyÚ deformat_date s ,
|
z.JueJinHttpApi.__request.<locals>.deformat_dateTé u æéAPI请æ±æ¶é´ï¼ú-)ÚjsonÚcoder Údata)r Z load_juejinÚtimeÚrequestr ÚinfoÚintÚ_JueJinHttpApi__BASE_URLÚrequestsÚpostÚtextr Úloadsr ÚlistÚdict)ÚclsZpath_strÚ data_jsonr ZDELEGATEÚfdataZ_JueJinHttpApi__start_timeZ_JueJinHttpApi__use_timeÚurlZresponseÚresultZ
|
resultJsonÚdÚkr r r Z __request s:
|
|
|
|
zJueJinHttpApi.__requestc C s | d||d¡S )NÚget_instruments)ÚsymbolsÚfields)Ú_JueJinHttpApi__request)r' r/ r0 r r r r. B s z JueJinHttpApi.get_instrumentsc C s | d|||||d¡S )NÚ history_n)ÚsymbolÚ frequencyÚcountÚadjustr0 )r1 )r' r3 r4 r5 r6 r0 r r r r2 F s zJueJinHttpApi.history_nc C s | d||d¡S )NÚcurrent)r/ r0 )r1 )r' r/ r0 r r r r7 K s zJueJinHttpApi.currentc C s | d|||||d¡S )Nr. )Ú exchangesÚ sec_typesÚskip_suspendedÚskip_str0 )r1 )r' r8 r9 r: r; r0 r r r Úget_exchanges_codesO s z!JueJinHttpApi.get_exchanges_codesc C s | d||d¡S )NÚget_previous_trading_date)ÚexchangeÚdate)r1 )r' r> r? r r r r= U s z'JueJinHttpApi.get_previous_trading_datec C s | d||d¡S )NÚget_next_trading_date)r> r? )r1 )r' r> r? r r r r@ Y s z#JueJinHttpApi.get_next_trading_datec C s | d|||d¡S )NÚget_trading_dates)r> Ú
|
start_dateÚend_date)r1 )r' r> rB rC r r r rA ] s z JueJinHttpApi.get_trading_datesN) Ú__name__Ú
|
__module__Ú __qualname__r Ú classmethodr1 r. r2 r7 r< r= r@ rA r r r r r s * r c @ s e Zd Ze ej¡Zedd Z edd Z
|
edddZ eddd Z ed
|
d Z ed d Zedd Zedd Zedd ZdS )Ú JueJinApic 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__redisManagerZgetRedisr ÚgetZrealse)r' ZredisÚ
|
account_idZ strategy_idÚtokenr r r ÚgetJueJinAccountInfog s
|
|
z JueJinApi.getJueJinAccountInfoc C sX g }xN|D ]F}|dd
dkr0| d |¡¡ q
|
|dd
dkr
|
| d |¡¡ q
|
W |S )Nr é Ú00zSZSE.{}Ú60zSHSE.{})ÚappendÚformat)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ú,)r/ r0 ) rT r ÚJUEJIN_LOCAL_APIrM ÚgmapiÚ set_tokenr. Újoinr )r' rS r0 r/ rK Ús_idrL 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é )r3 r4 r5 r6 r0 )rT r rV rM rW rX r2 r ) r' r r5 r0 r/ rK rZ rL Ú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 )NrU )r/ Ú )r/ r0 ) rT r rV rM rW rX r7 rY r )r' rS r/ rK rZ rL 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)r8 r9 r: r; r0 )r rV rM rW rX r. r r< )r' r8 rK rZ rL 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)r rV rM rW rX r= r )r' r? rK rZ rL r r r r= ² s
|
|
z#JueJinApi.get_previous_trading_datec C s: t jr*| ¡ \}}}t |¡ t d|¡S t d|¡S d S )Nra )r rV rM rW rX r@ r )r' r? rK rZ rL r r r r@ ¼ s
|
|
z JueJinApi.get_next_trading_datec C s> t jr,| ¡ \}}}t |¡ t d||¡S t d||¡S d S )Nra )r rV rM rW rX rA r )r' rB rC rK rZ rL r r r rA Å s
|
|
zJueJinApi.get_trading_dates)N)N)rD rE rF Ú redis_managerZ RedisManagerr ZREDIS_DBrI rG rM rT r[ r^ r` r< r= r@ rA r r r r rH c s
|
|
rH 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)rH r[ )r' rS r0 r r r r[ Ó s z%HistoryKDatasUtils.get_gp_latest_infoc C s t |||¡S )N)rH r^ )r' r r5 r0 r r r r^ × s z%HistoryKDatasUtils.get_history_tick_nc C s
|
t |¡S )N)rH r` )r' rS r r r r` Û s z&HistoryKDatasUtils.get_gp_current_infoc C s
|
t |¡S )N)rH r= )r' r? 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_cacherJ r= )r' r? 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Úrangere rQ )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 rf Ú._HistoryKDatasUtils__latest_trading_date_cacherk )r' rh Úkeyri r r r Ú get_latest_trading_date_cacheø s
|
|
|
|
z0HistoryKDatasUtils.get_latest_trading_date_cachec C s
|
t |¡S )N)rH r@ )r' r? r r r r@ s z(HistoryKDatasUtils.get_next_trading_datec C s t ||¡S )N)rH rA )r' rB rC r r r rA s z$HistoryKDatasUtils.get_trading_datesc C sH | |¡}g }x4|D ],}|d d¡d }|d }| ||f¡ qW |S )Nr3 Ú.r\ Úprice)r` ÚsplitrQ )r' rS r ZpricesÚitemr rp 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 rN c S s | d S )Nr\ r )Útupr r r Ú<lambda>& ó z9HistoryKDatasUtils.get_codes_limit_rate.<locals>.<lambda>)rm ) r[ r Úto_priceÚdecimalÚDecimalr rs ÚfloatrJ ÚroundrQ ÚsortÚreverse) r' rS ÚdatasÚpre_price_dictr r ru Z
|
now_pricesZ f_resultsrp 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 )r^ )r' r r5 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 )Nr3 ro r\ Úsec_name)r[ rq )r' rS r r] r r Z code_namer r r Úget_gp_codes_names3 s
|
|
z%HistoryKDatasUtils.get_gp_codes_names)N)N)rD rE rF rd rl rG r[ r^ r` r= re rk rn r@ rA rs r r r r r r r rc Ï s"
|
|
|
rc Ú__main__Z000422Z600610)Ú__doc__r r{ r r r! Úutilsr Zdb.redis_manager_delegater Úlog_module.logr r r r Údbr rb r rH Úobjectrc rD r r% r] r r r r Ú<module> s Nlo
|
