B
|
³e°. ã @ sº d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
|
Z
|
ddl m Z m Z ddlmZ G dd dZG d d
|
d
|
ZG d d d eZed kr¶e eddh¡Zee dS )u
|
åå²K线æå¡
|
é N)Ú
|
RedisUtils)Úlogger_request_api)Ú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 Ú __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)Ú auto_freezjuejin-strategy-idz juejin-token)Ú_JueJinApi__redisManagerÚgetRedisr ÚgetÚrealse)r& ÚredisÚ
|
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ú,)r1 r2 ) rZ ÚconstantÚJUEJIN_LOCAL_APIrS ÚgmapiÚ set_tokenr0 Újoinr )r& rY r2 r1 rQ Ús_idrR 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 )rZ r\ r] rS r^ r_ r4 r ) r& r r7 r2 r1 rQ ra rR Ú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 )Nr[ )r1 Ú )r1 r2 ) rZ r\ r] rS r^ r_ r9 r` r )r& rY r1 rQ ra rR 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 )Nrc TzLsymbol,sec_type,sec_id,sec_name,listed_date,sec_level,is_suspended,pre_close)r: r; r< r= r2 )r\ r] rS r^ r_ r0 r r> )r& r: rQ ra rR 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\ r] rS r^ r_ r? r )r& rA rQ ra rR r r r r? ² s
|
|
z#JueJinApi.get_previous_trading_datec C s: t jr*| ¡ \}}}t |¡ t d|¡S t d|¡S d S )Nrh )r\ r] rS r^ r_ rB r )r& rA rQ ra rR r r r rB ¼ s
|
|
z JueJinApi.get_next_trading_datec C s> t jr,| ¡ \}}}t |¡ t d||¡S t d||¡S d S )Nrh )r\ r] rS r^ r_ rC r )r& rD rE rQ ra rR r r r rC Å s
|
|
zJueJinApi.get_trading_dates)N)N)rF rG rH Ú redis_managerÚ RedisManagerrL rI rS rZ rb re rg 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 rb )r& rY r2 r r r rb Ó s z%HistoryKDatasUtils.get_gp_latest_infoc C s t |||¡S )N)rJ re )r& r r7 r2 r r r re × s z%HistoryKDatasUtils.get_history_tick_nc C s
|
t |¡S )N)rJ rg )r& rY r r r rg Û 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_cacherN 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Úrangerm rW )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 rn Ú._HistoryKDatasUtils__latest_trading_date_cachers )r& rp Úkeyrq 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 Ú.rc Úprice)rg ÚsplitrW )r& rY r ZpricesÚitemr rx 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 )
|
NZsec_idÚ pre_closer rc éd rT c S s | d S )Nrc r )Útupr r r Ú<lambda>& ó z9HistoryKDatasUtils.get_codes_limit_rate.<locals>.<lambda>)ru ) rb r Úto_priceÚdecimalÚDecimalr r{ ÚfloatrN ÚroundrW ÚsortÚreverse) r& rY ÚdatasZpre_price_dictr r r| Z
|
now_pricesZ f_resultsrx 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 )re )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 rw rc Zsec_name)rb ry )r& rY r rd r r Z code_namer r r Úget_gp_codes_names3 s
|
|
z%HistoryKDatasUtils.get_gp_codes_names)N)N)rF rG rH rl rt rI rb re rg r? rm rs rv rB rC r{ r r r r r r r rk Ï s"
|
|
|
rk Ú__main__Z000422Z600610)Ú__doc__r r r r r r\ Zdb.redis_manager_delegater Úlog_module.logr Úutilsr r Údbr ri r rJ Úobjectrk rF r r$ rd Úprintr r r r Ú<module> s Nlo
|
