o iC @sdZddlmZddlZddlZddlZddlmZddlZ ddl Z ddl Z ddl mZddlZddlmZddlmZddlZddlmZmZdd lmZz dd lmZmZWneyidd l mZmZYnwe j!Z"e#e"$d d d kse#e"$d ddkrddl%m&Z&nddl'm&Z&   d`ddZ(daddZ)  dbddZ*dcddZ+ddddZ,ddZ-d d!Z.ded"d#Z/ $ &dfd'd(Z0d)d*Z1d+d,Z2d-d.Z3d/d0Z4d1d2Z5 3 $ %  dgd4d5Z6 3 3 3 % 3  dhd6d7Z7   did8d9Z8ded:d;Z9dd3dggdf dAdBZdd>gdfdGdHZ?dIdJZ@dedKdLZAdedMdNZBdddOdPZCdQdRdSdTZDdedUdVZEdWdXZFdYdZZGdmd\d]ZHeId^kr^e/d_ZJeKeJdSdS)nui 交易数据接口 Created on 2014/07/31 @author: Jimmy Liu @group : waditu @contact: jimmysoa@sina.cn )divisionN)etree)cons)dateu)MA)get_apis close_apis)get_stock_basics)urlopenRequest.)StringIODMbP?c Cs$tdt|}d}|tjvr'tjtjdtjdtj||f}n|tj vr[ zget_hist_data..r turnoveraxisdateF ascending)&printct_code_to_symbolupperK_LABELS DAY_PRICE_URLP_TYPEDOMAINSK_TYPE K_MIN_LABELSDAY_PRICE_MIN_URL TypeErrorrangetimesleepr r readlen ExceptionjsonloadsPY3decode INDEX_LABELSINX_DAY_PRICE_COLUMNSDAY_PRICE_COLUMNSpd DataFrameapplymapastypefloatr0drop set_index sort_indexIOErrorNETWORK_URL_ERROR_MSG)codestartendktype retry_countpausesymbolurl_requestlinesejscolsdfcolr)r)r* get_hist_data#s^          rfcCstttjtjdtjdtjd||f}t|dd}|dkr'dSt d}| d tj r7| d n|}|d d }|d d}|dd}|dd}tj r[t|}ntj|dd}t|}tjtj|dtidtjd}|jddd}|S)u 处理当日行情分页数据，格式为json Parameters ------ pageNum:页码 return ------- DataFrame 当日所有股票交易数据(DataFrame) rvsfjvrrnullNz \,(.*?)\:z,"\1":gbkz"{symbolz{"symbolz{symbolz {"symbol"z"""z"{"{"GBK)encodingrVdtyperr\r r.)r4_write_consoler SINA_DAY_PRICE_URLr9r:PAGESr rBrecompilesubrGrHr&rEdumpsrFrLrM read_jsonobjectDAY_TRADING_COLUMNSrQ)typespager_textregrbrdr)r)r*_parsing_dayprice_jsonks,         rsnc Cs|tjvrttjdSt|}t|}|dd}tjdtjtj dtj dtj d||ftjdtj tj dtj d tj d ||ftjd tj tj dtj d |dd ||fi}t|D]c} t|zF|tjd krt||} tj| _n.t||} t| dd} | d} t| dkrWdStjt| tjdgd} W| SW| Sty} z t| WYd} ~ qgd} ~ wwttj)u 获取分笔数据 Parameters ------ code:string 股票代码 e.g. 600848 date:string 日期 format: YYYY-MM-DD retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 src : 数据源选择，可输入sn(新浪)、tt(腾讯)、nt(网易)，默认sn return ------- DataFrame 当日所有股票交易数据(DataFrame) 属性:成交时间、成交价格、价格变动，成交手、成交金额(元)，买卖类型 N-rrrsfdlr ttidx163rrrmnamesskiprows)stripr4 TICK_SRCSr3TICK_SRC_ERRORr5_code_to_symbol_dgtr&TICK_PRICE_URLr9r:rsTICK_PRICE_URL_TTTICK_PRICE_URL_NTr?r@rArL read_excel TICK_COLUMNSr r r rBrHrC read_tablerrDrTrU)rVr0rZr[srcr\ symbol_dgtdatestrr]r^rdrtr`rar)r)r* get_tick_datasL    $$&        rc Cs$|dust|dks|durdSt|}|d}t|D]o}t|zRttjtjdtj dtj d|||f}t |dd }| d }t|dkrRWdStjt|tjd gd } | durr| d d d| d <W| SW| Sty} z t| WYd} ~ qd} ~ wwttj)u 获取sina大单数据 Parameters ------ code:string 股票代码 e.g. 600848 date:string 日期 format：YYYY-MM-DD retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 return ------- DataFrame 当日所有股票交易数据(DataFrame) 属性:股票代码 股票名称 交易时间 价格 成交量 前一笔价格 类型（买、卖、中性盘） NdrrgsinaddrrrmrrrVcSs |ddS)Nrr)r'r)r)r*r+r,zget_sina_dd..)rCr4r5r?r@rAr SINA_DDr9r:rsr rBrHrLread_csvr SINA_DD_COLSmaprDr3rTrU) rVr0volrZr[r\r^rtr`rdrar)r)r* get_sina_dds6      rc CsT|dus t|dkr dSt|}t}t|D]}t|zlttj tj dtj dtj d||f}t |dd}|d}|d d }t|td tftd d d}t|}t|}t|d}t} ttd |d D]} | jt||| ||dd} qxW| Sty} z tt| WYd} ~ qd} ~ wwttj)u4 获取当日分笔明细数据 Parameters ------ code:string 股票代码 e.g. 600848 retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 return ------- DataFrame 当日所有股票交易数据(DataFrame) 属性:成交时间、成交价格、价格变动，成交手、成交金额(元)，买卖类型 Nrrrgrhrrrmr DummycSs|SNr))snr)r)r*r+ sz"get_today_tickss..) __getitem__ detailPagesT ignore_index) rCr4r5dutodayr?r@rAr TODAY_TICKS_PAGE_URLr9r:rsr rBrHevaltypedictrErwrFrLrM _write_headappend _today_ticksrDr3strrTrU) rVrZr[r\r0r^r_data_strpagesdatapNoerr)r)r*get_today_ticksssF             rc Cstt|D]z}t|z]tjtjtj dtj dtj d|||f}| d}tj r8dd|D}ndd|D}d|}d |}|d d }tjt|d d d} tj| _| ddd| d<W| Sty} z t| WYd} ~ qd} ~ wwttj)Nrrgt_ticksz//table[@id="datatbl"]/tbody/trcSg|] }t|dqSrrtostringrH.0noder)r)r* (z _today_ticks..cSg|]}t|qSr)rrrr)r)r*r*rz%s
z--0F) parse_datesrpchangecSr#)N%rr%r'r)r)r*r+0r,z_today_ticks..)r4rqr?r@rAlxmlhtmlparseTODAY_TICKS_URLr9r:rsxpathrGjoinr&rL read_htmlrTODAY_TICK_COLUMNSr rrDr3rTrU) r\tdatepageNorZr[r^rressarrrdrar)r)r*rs0      rcCspttdd}|dur,tdtjdD]}td|}|jddkr+|j|dd}q|jtdddd}|S) u= 一次性获取最近一个日交易日所有股票的交易数据 return ------- DataFrame 属性：代码，名称，涨跌幅，现价，开盘价，最高价，最低价，最日收盘价，成交量，换手率，成交额，市盈率，市净率，总市值，流通市值 hs_ar NrrTrshfxjs)r4rrr?PAGE_NUMshaper)rdinewdfr)r)r* get_today_all8s   rcCsd}t|tst|tst|tst|tjr&|D] }|t|d7}qnt|}t|dkr7|ddn|}t tj tj dtj dt |fdd d d }t|d d }|d}td}||}td}||}g} g} t|D]"\} } t| dkr| dd| dddD| || qyt| dkrdStj| tjd} | jddd} | | d<dd| jD}|D] }| |dd| |<q| S)um 获取实时交易数据 getting real time quotes data 用于跟踪交易情况（本次执行的结果-上一次执行的数据） Parameters ------ symbols : string, array-like object (list, tuple, Series). return ------- DataFrame 实时交易数据 属性:0：name，股票名字 1：open，今日开盘价 2：pre_close，昨日收盘价 3：price，当前价格 4：high，今日最高价 5：low，今日最低价 6：bid，竞买价，即“买一”报价 7：ask，竞卖价，即“卖一”报价 8：volumn，成交量 maybe you need do volumn/100 9：amount，成交金额（元 CNY） 10：b1_v，委买一（笔数 bid volume） 11：b1_p，委买一（价格 bid price） 12：b2_v，“买二” 13：b2_p，“买二” 14：b3_v，“买三” 15：b3_p，“买三” 16：b4_v，“买四” 17：b4_p，“买四” 18：b5_v，“买五” 19：b5_p，“买五” 20：a1_v，委卖一（笔数 ask volume） 21：a1_p，委卖一（价格 ask price） ... 30：date，日期； 31：time，时间； rr$Nrrsinahqz hq.sinajs.cnzhttps://finance.sina.com.cn/)hostreferer)headersrrrmz \="(.*?)\";z(?:sh|sz)(.*?)\=r cSsg|]}|qSr)r))rastrr)r)r*rz'get_realtime_quotes..!rrrr.rVcSsg|]}d|vr|qS)_vr))rclsr)r)r*rscSs |ddS)Nr)r'r)r)r*r+r,z%get_realtime_quotes..) isinstancelistsettuplerLSeriesr4r5rCr LIVE_DATA_URLr9r:_randomr rBrHrtrufindall enumeratersplitrMLIVE_DATA_COLSrQr r)symbols symbols_listrVr_r}r~rregSymsyms data_list syms_listindexrowrdlstxtr)r)r*get_realtime_quotesNsB%*         " rqfqFTcCstd|dur tn|}|durtn|}t||}|d} ttt||| |||} | dur9t } t |dkrit dt |D]"} || } t tt||| |||} | duran| j| dd} qFt | dkst | | j|k| j|k@dkrt S| d} |r| | j|k| j|k@} | d} | jdd } | S|d kr|r| jd dd } | | j|k| j|k@} d D]} | | tj| | <| | t| | <q| d} | jdd } | S|dkr|r| jd dd } t|||} | d} | jddd } | dd}| | j|dk}t|}|durt St|ddkt|ddk@r6t|d}n)ttrEt|d}ntdktdk@rYt|d}nt|d}t|d |}| | j|k| j|k@} dD]} | | || | <| | tj| | <| | t| | <qu| d} | jdd } | Sd D] } | | | d | | <q|r| jd dd } | | j|k| j|k@} d D] } | | tj| | <q| d} | jdd } | t} | S)uO 获取历史复权数据 Parameters ------ code:string 股票代码 e.g. 600848 start:string 开始日期 format：YYYY-MM-DD 为空时取当前日期 end:string 结束日期 format：YYYY-MM-DD 为空时取去年今日 autype:string 复权类型，qfq-前复权 hfq-后复权 None-不复权，默认为qfq retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 drop_factor : bool, 默认 True 是否移除复权因子，在分析过程中可能复权因子意义不大，但是如需要先储存到数据库之后再分析的话，有该项目会更加灵活 return ------- DataFrame date 交易日期 (index) open 开盘价 high 最高价 close 收盘价 low 最低价 volume 成交量 amount 成交金额 rNrr Trr0Fr1hfqfactorr.)openhighcloselowrrr  pre_closeprice rrr r)r3rtoday_last_yearr get_quartsr4r_parse_fq_data_get_index_urlrLrMrCr?rqrr0drop_duplicatesrRrSrQrFORMATrOrP_parase_fq_factor sort_valuesheadr is_holidayget_hour)rVrWrXautyperrZr[ drop_factorqsqtrdrdlabel firstDatefrowrtpreCloserater)r)r* get_h_datas  ,         "      r%cCs0t|}ttjtjdtjd|f}t|dd}|dt|d}tj r/| dn|}| dd}| d d }| d d }| d d}| dd}| dd}t |}tt|d t|d d}|dt|d<|djtjkrt|d|d<|d}|dt|d<|S)Nrrgrrr rz{_rltotalz"total"rz"data"z:"z":"z",_z","r^r)r0rr0r)r4r5r HIST_FQ_FACTOR_URLr9r:r rBrCrGrHr&rErFrLrMrkeysvaluesr _fun_exceptdtypesnpry to_datetimerrOrP)rVrWrXr\r_r}rdr)r)r*rs*         ( rcCst|dkr |ddS|S)Nri)rCr'r)r)r*r*s  r*c Csrt|D]}t|zt|}t|dd}|d}tj t |}| d}t j r5dd|D} ndd|D} d| } | dkrIWdStj| d d gd d } t| d kratWS|rlt jd d | _nt j| _| d jtjkrt| d | d <| d } W| Sty} zWYd} ~ dSd} ~ wty} z t| WYd} ~ qd} ~ wwtt j)Nrrrmz"//table[@id="FundHoldSharesTable"]cSrrrrr)r)r*r0rz"_parse_fq_data..cSrr)rrr)r)r*r2rrrr )rr0)r?r@rAr r rBrHrrrrrr4rGrrLrrCrM HIST_FQ_COLSr r+r,ryr-r ValueErrorrDr3rTrU) r]rrZr[r^r_r}rrrrdrar)r)r*r&s@        rcCs&ttjtjdtjdf}t|dd}|d}|dddd}|d dd dd d }d tj |f}t j t |d d d}|d|ddd|d<|dd|d<|d tj|d<|d tj|d<|tj}|d dd|d<|dt|d<|dt|d<|S)um 获取大盘指数行情 return ------- DataFrame code:指数代码 name:指数名称 change:涨跌幅 open:开盘价 preclose:昨日收盘价 close:收盘价 high:最高价 low:最低价 volume:成交量(手) amount:成交金额（亿元） rrrrrmz var hq_str_shrz var hq_str_szz";rk=r$%s%s)sep thousandsrprecloser rchangeamountirVcSt|dSNrrzfillr'r)r)r*r+hzget_index..)r r4 INDEX_HQ_URLr9r:r rBrHr& INDEX_HEADERrLrrrrFORMAT4 INDEX_COLSrOrP)r_r}rdr)r)r* get_indexJs$  rAcCs\|rtjtjdtjd||d|df}|Stjtjdtjd||d|df}|S)Nrrgrr )r4HIST_INDEX_URLr9r: HIST_FQ_URL)rrVrr]r)r)r*rnsrrc Csjtd|r tj|nt|}d} d} |durdn|}|du|dk@r2|dus,|dkr0tn|}|tjvr|dur?|nd} |dddvsK|rMd} |durSdnd} |dup\|dk|dupc|dk@rtjtj dtj d| | |tj |||| t d f g} n@t ||}g} |D]5}t|d }t|dd }tjtj dtj d| | t||tj |||| t d f } | | qd | tj |f} n"|tjvrtjtj dtj d|||t d fg} d|} ntdt}| D]} |jt| | ||||||dd}q|tjvr3|du|dk@|du|dk@@r3|jdkr3||j|k|j|k@}|S)u: 获取k线数据 --------- Parameters: code:string 股票代码 e.g. 600848 start:string 开始日期 format：YYYY-MM-DD 为空时取上市首日 end:string 结束日期 format：YYYY-MM-DD 为空时取最近一个交易日 autype:string 复权类型，qfq-前复权 hfq-后复权 None-不复权，默认为qfq ktype：string 数据类型，D=日k线 W=周 M=月 5=5分钟 15=15分钟 30=30分钟 60=60分钟，默认为D retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 return ------- DataFrame date 交易日期 (index) open 开盘价 high 最高价 close 收盘价 low 最低价 volume 成交量 amount 成交额 turnoverratio 换手率 code 股票代码 rrNr )15fqrrz-01-01z-12-31r2zm%srTrF)r3r4 INDEX_SYMBOLr5rrr6r7 KLINE_TT_URLr9r: TT_K_TYPErtt_datesrrr<KLINE_TT_MIN_URLr>rLrM _get_k_dataemptyr0rTrU)rVrWrXrYrrrZr[r\r]dataflagrFklineurlsyearsyear startdateenddaterr)r)r* get_k_dataxsf$        " rWc Cst|D]}t|zt|} t| dd} t| dkr"WdSWnty9} z t| WYd} ~ qd} ~ wwt j rB| dn| } | dd} t d} t | d| } t| d } |t| d |vrl|nt j|}t| d ||d krdSt| d ||d d krtj| d ||t jd }ntj| d ||t jd }|r|n||d <|t jvr|ddd|d<|jdd D] }||t||<q|SdS)Nrrrrr1r z ,{"nd.*?}rrrrrrVr0cSs:d|dd|dd|dd|dd|ddfS)Nz%s-%s-%s %s:%srrrrr r)r'r)r)r*r+s z_get_k_data..)r?r@rAr r rBrCrDr3r4rGrHrrtrusubnrErFrr(rKr6rLrMKLINE_TT_COLS_MINS KLINE_TT_COLSr<rr rOrP)r]rPr\rVrrYrZr[r^r_r`rar~rbrdrer)r)r*rNsD    * rNc Csnt}t|tst|tst|tst|tjr5|D]}t||||||d}||d<|j|dd}q|SdS)uh 批量获取历史行情数据，具体参数和返回数据类型请参考get_hist_data接口 )rWrXrYrZr[rVTrN) rLrMrrrrrrfr) rrWrXrYrZr[rdr\rr)r)r* get_histss*r\cCs|durtn|}|dd}|dkrdS|durdn |dd|ddd}tjtj||dur5d n|fd d id }|S) u 获取每日收盘行情 Parameters: ------------- date:交易日期，格式:YYYY-MM-DD Return: ------------- DataFrame code 代码, name 名称, p_change 涨幅%, price 现价, change 涨跌, open 今开, high 最高, low 最低, preprice 昨收, pe 市盈(动), volratio 量比, turnover 换手%, range 振幅%%, volume 总量, selling 内盘, buying 外盘, amount 总金额, totals 总股本(万), industry 细分行业, area 地区, floats 流通股本(万), fvalues 流通市值, abvalues AB股总市值, avgprice 均价, strength 强弱度%, activity 活跃度, avgturnover 笔换手, attack 攻击波%, interval3 近3月涨幅 ，interval 近6月涨幅 Nrr20170614rrr/hqrVryro)r last_tddater&rLrr4 ALL_DAY_FILE)r0wdatedateprerdr)r)r* get_day_all s ( rdcCs(t|dd}|dddd}|S)Nrrr:)rr&)ttstrr)r)r* get_dt_time*srhEc Cs t}t}d||||f}t|d}|dvr|dkr-|j|jjddkddf}n|dkrA|j|jjddkddf}nd}|dkrId nd}|jD]} d | } d |vrt| ||||||d } | j| d <| d t | d <| d | _| j dd gd dd| j ddidd| j ddidd| j ddidd| j ddidd| j ddidd| j ddidd| jdddf d| jdddf<| jdddf d| jdddf<| jdddf d| jdddf<| jdddf d| jdddf<| jdddf d| jdddf<| jdddf d| jdddf<| jdddfd9<| jdddfd9<| jdddfd9<| jdddfd9<d| jddd f<d| jddd!f<d| jddd"f<d| jddd#f<d| jddd$f<d| jddd%f<d| jddd&f<d| jddd'f<d| jddd(f<d| jddd)f<d| jddd*f<d| jddd+f<d| jddd,f<d| jddd-f<d| jddd.f<d| jddd/f<d| jddd0f<d| jddd1f<d| jddd2f<d| jddd3f<d| jddd4f<d| jddd5f<d| jddd6f<d| jddd7f<d| jddd8f<d| jddd9f<d| jddd:f<d| jddd;f<d| jdddf<d| jddd?f<d| jddd@f<d| jdddAf<d| jdddBf<d| jdddCf<d| jdddDf<d| jdddEf<d| jdddFf<d| jdddGf<dH| jdddIf<dH| jdddJf<dH| jdddKf<dH| jdddLf<dH| jdddMf<t | | || <qN|t|dS)NNz%s%s%sbar%s.h5a)SHSZrkr6rlrr z%s.SHmin)conn start_dateend_datefreqmarketassetTimerVT)r/inplacerOPEN)r rvrCLOSEr LOWrHIGHrVOLUMEr7TURNOVERint64i' ASKPRICE1 ASKPRICE2 ASKPRICE3 ASKPRICE4 ASKPRICE5 ASKPRICE6 ASKPRICE7 ASKPRICE8 ASKPRICE9 ASKPRICE10 BIDPRICE1 BIDPRICE2 BIDPRICE3 BIDPRICE4 BIDPRICE5 BIDPRICE6 BIDPRICE7 BIDPRICE8 BIDPRICE9 BIDPRICE10ASKVOL1ASKVOL2ASKVOL3ASKVOL4ASKVOL5ASKVOL6ASKVOL7ASKVOL8ASKVOL9ASKVOL10BIDVOL1BIDVOL2BIDVOL3BIDVOL4BIDVOL5BIDVOL6BIDVOL7BIDVOL8BIDVOL9BIDVOL10gVWAPVOL30 TOTAL_VOLUMETOTAL_TURNOVERINTEREST)rr rLHDFStorelocrrbarapplyrhrQrenamerOr3rr) rsr0rrrtfilepathrstksfnamestorestkr\rdr)r)r*bar2h50s         &&&&&&  rc std|}t| D]} z|dur ttjWdS|\} } |}|}|dkr;t||| dn|}|dvrJt| tj|}n|dkrPdn|}t| tjd}|tj vrt }td D],}|tj ||||d d }| ||d kr|nj|d d }t|d krnqg|ddd|d<|tjvrt }td D],}|tj ||||d d }| ||d kr|nj|d d }t|d krnqt |d|d<|jt|djdd ddjtjdd|dkrdn||dkrdn|}|jdd}|dvr&|tj}|tjvr%|dd |d<n!|tj}|dvr:tj|_|tj}n |jddgdd}tj|_|dkrY|durt||tj vrn|j d d d}|dj!dd|d<nfd d!|j"#fd"d|d<tjdd#D]/}|d$kr|||d||<n|||dt$dd ||<||#tj%||<q|jddd}| durYt| d krYd%| vr"t&||tj vr|j d d d}|d&j!dd|d&<nfd'd!|j"#fd(d|d&<|d|d&|d%<|d%#tj%|d%<|d%'t$|d%<|jd&dd}d)| vrYt(|dd#|d*<|d*)d+|d,<|d|d,#tj%|d)<|d)'t$|d)<|jd*d,gdd}|durt|d kr|D]+}t*|t+rt(|d-|#tj%)|d |d.|<|d.|'t$|d.|<qgd/D] }||'t$||<q|d-,d0d |d1<|d1#tj%'t$|d1<|WSYdSt-tj.)2u BAR数据 Parameters: ------------ code:证券代码，支持股票,ETF/LOF,期货/期权,港股 con:服务器连接 ，通过ts.api()或者ts.xpi()获得 start_date:开始日期 YYYY-MM-DD/YYYYMMDD end_date:结束日期 YYYY-MM-DD/YYYYMMDD freq:支持1/5/15/30/60分钟,周/月/季/年 asset:证券类型 E:股票和交易所基金，INDEX:沪深指数,X:期货/期权/港股/中概美国/中证指数/国际指数 market:市场代码，通过ts.get_markets()获取 adj:复权类型,None不复权,qfq:前复权,hfq:后复权 ma:均线,支持自定义均线频度，如：ma5/ma10/ma20/ma60/maN factors因子数据，目前支持以下两种： vr:量比,默认不返回，返回需指定：factor=['vr'] tor:换手率，默认不返回，返回需指定：factor=['tor'] 以上两种都需要：factor=['vr', 'tor'] retry_count:网络重试次数 Return ---------- DataFrame code:代码 open：开盘close/high/low/vol成交量/amount成交额/maN均价/vr量比/tor换手率 期货(asset='X') code/open/close/high/low/avg_price：均价 position：持仓量 vol：成交总量 rNrrtxapiriINDEXrXDXri rTrdatetimecSst|ddSNrrrr'r)r)r*r+rzbar..)rVF)rQrvr r.r1r/<r positionri) left_index right_index adj_factorbfill)methodcj|dS)Nrrdayrdr)r*get_valzbar..get_valct|ddSrrr'rr)r*r+rtorfloatscr)Nrrrrr)r*rrcrrrr'rr)r*r+rvrvol5meanrzma%srrp_change)/r3rr6r?r4MSG_NOT_CONNECTED _get_mkcodegetattrASSETKTYPE_LOW_COLSrLrMKTYPEto_dfrrCr KTYPE_ARRr-assignrrRrQ T_DROP_COLSrS BAR_E_COLS BAR_X_COLSBAR_X_FUTURE_COLSr BAR_X_FUTURE_RL_COLSBAR_X_OTHER_COLS factor_adjmergefillnarrrPr factor_sharesrOrshiftrint pct_changerTrU)rVrorprqrrrtrsadjmafactorsrZr^apirrYmkcodefuncrrdsrerjr))rdrr*rs"                          $     *  rcCs|d}|dkr t|}|S|dkrt|}|Stjtjr&ttj}n t |}| tj||j |kdj d}|S)Nrrirrsr) r4 _market_code_idx_market_codeospathexists INST_PLK_FrL read_pickleget_instrument to_picklerVr))rVrtrrmksr)r)r*rs    rc Cs|}t|dd}tttdd}t|D]'}z |dur0tt j WdS|\}} t } |dkrCt ||| dn|} |dvrK|n| } tdD];} ||krc| j| || ddd}n | j| ||| ddd }||}| d kr}| |n|j| d d } t|dkrnqQ|dvr|| d <| d dd| d <| d | d| d<| gd} gd| _nk| dvr||kr| jgddd} n | jgddd} d| j| jdkdf<d| j| jdkdf<| d } gd| _n6| dvr||kr| jgddd} n | jgd dd} gd!| _n| jgd"dd} W| SW| SW| SW| StyF}z t|WYd}~qd}~wwdS)#u| tick数据 Parameters: ------------ code:证券代码，支持股票,ETF/LOF,期货/期权,港股 conn:服务器连接 ，通过ts.api()或者ts.xpi()获得 date:日期 asset:证券品种，E:沪深交易所股票和基金, INDEX:沪深交易所指数， X:其他证券品种，大致如下： 支持的扩展行情包括(asset='X')： 郑州商品期权 OZ 大连商品期权 OD 上海商品期权 OS 上海个股期权 QQ 香港指数 FH 郑州商品 QZ 大连商品 QD 上海期货 QS 香港主板 KH 香港权证 KR 开放式基金 FU 货币型基金 FB 招商理财产品 LC 招商货币产品 LB 国际指数 FW 国内宏观指标 HG 中国概念股 CH 美股知名公司 MG B股转H股 HB 股份转让 SB 股指期货 CZ 香港创业板 KG 香港信托基金 KT 国债预发行 GY 主力期货合约 MA 中证指数 ZZ 港股通 GH market:市场代码，通过ts.get_markets()获取 Return ---------- DataFrame date:日期 time:时间 price:成交价 vol:成交量 type:买卖方向，0-买入 1-卖出 2-集合竞价成交 期货 0:开仓 1:多开 -1:空开 期货多一列数据oi_change:增仓数据 rrNrr,)rsrVrWcount)rsrVr0rWrrTrr0cSs2dt|ddt|ddt|ddfS)Nz %s-%s-%s rrrrrr'r)r)r*r+`s2ztick..r@r)rr r buyorsell)rr rr)G)hourminute nature_namezengcang directionsecond nature_mark nature_valuer r.)r r r r rrinature)r0r rrr)r r rrrrr)r r rr)r0r r oi_changer)r r r r rr)rr6rr&rrrr?r3r4rrLrMrget_transaction_dataget_history_transaction_datarrrCrr rQrrrrD)rVror0rtrsrZrr^rrrrconrrrdrar)r)r*tick*sr               rc Cst|D]}z|durttjWdS|\}}t}t|ts1t|ts1t|t s1t|tj rl|D]7} t | ||d} |dkrL| | | | fg} n|dkr\| | | | fg} n | || | } || }q3n0t |||d} |dkr| | | |fg}n|dkr| | | |fg}n | || |}|dvr|jgddd}n |jd gdd}W|SW|Sty} z t| WYd} ~ qd} ~ wwttj) u 获取实时快照 Parameters ------ symbols : string, array-like object (list, tuple, Series). return ------- DataFrame 实时快照，5档行情 Nrrirr) rsactive1active2reversed_bytes0reversed_bytes1reversed_bytes2reversed_bytes3reversed_bytes4reversed_bytes5reversed_bytes6reversed_bytes7reversed_bytes8reversed_bytes9r r.rs)r?r3r4rrLrMrrrrrrrget_security_quotesget_instrument_quoterrQrDrTrU) rrortrsrZr^rrrrVrrdrar)r)r*quotessH  *      r'cCsHg}tdD]}|d|d}||7}t|dkrnq||}|S)u$ 获取股票列表 rr)r?get_security_listrCr)rrprr)r)r* get_securitys   r+cCszddlmm}|dur|n|}g}tdD]}||dd}||7}t|dkr/nq||}|t j |S)u0 重新设置本地证券列表 rNrr) tushare.util.connsutilconnsxapi_xr?get_instrument_inforCrrr4rrcsrrrr)r)r*reset_instruments    r3cCstdddlmm}|dur|n|}|dur"ttjdSg}tdD]}||dd}||7}t |dkr>nq(| |}|S)u$ 获取证券列表 rrNrr) r3r,r-r.r/r4rr?r0rCrr1r)r)r*rs    rc Csd}t|D]v}t|zYt|||tf}t|dd}t|}|dd}t |}|j ddddd d }|gd }|d d d |d<|d dd |d<|d dd |d<W|St y|}z t|WYd}~qd}~wwttj)u] 获取分笔数据 Parameters ------ code:string 股票代码 e.g. 600848 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 src : 数据源选择，可输入sn(新浪)、tt(腾讯)、nt(网易)，默认sn return ------- DataFrame 当日所有股票交易数据(DataFrame) 属性:成交时间、成交价格、价格变动，成交手、成交金额(元)，买卖类型 zhttp://push2ex.eastmoney.com/getStockFenShi?pagesize=6644&ut=7eea3edcaed734bea9cbfc24409ed989&dpt=wzfscj&pageindex=0&id=%s&sort=1&ft=1&code=%s&market=1&_=%srrrr@r rr)rfr*vbsr)r@r rrcSs|dS)Nr(r)r'r)r)r*r+sz!get_today_ticks..cSs tt|Sr)bs_typerr'r)r)r*r+ r,cSr8r9r:r'r)r)r*r+ r<N)r?r@rAr rr rBrErFrLrMrrrDr3rTr4rU) rVrZr[r]r^rtr`rdrar)r)r*get_today_tickss(       r7u买入u卖出r)rD24cCs,|dur ttjdS|}||}|S)u$ 获取市场代码 N)r3r4r get_marketsr)rrr)r)r*r:s   r:cCs0ttjtjdtjd|f}|d}|S)Nrossr)rLrr4 ADJ_FAC_URLr9r:rRrVrdr)r)r*r!s   rcCs8ttjtjdtjd|fddg}|d}|S)Nrr;rr)rLrr4 SHS_FAC_URLr9r:rRr=r)r)r*r(s  r cCs2ddlm}d|d}d|d}t|||S)Nr)randintrr )randomr@r)rr@rWrXr)r)r*r0s   r__main__600000)NNNrrr)Nr )NNrrr)NNrrr)Nrrr)NNrFrrT)NrrrrFrr)rrrFrrr)NNrrr)rrrrir)rriN)Nrrirr)r?)L__doc__ __future__rr@rE lxml.htmlrrpandasrLnumpyr,r tushare.stockrr4rt tushare.utilrrtushare.util.formularrr,rrtushare.stock.fundamentalr urllib.requestr r ImportErrorurllib2 __version__r4rrior pandas.compatrfrrrrrrrr%rr*rrArrWrNr\rdrhrrrrr'r+r3rr7r6r:rrr__name__rdr3r)r)r)r*s       ,  H# 7 *. G s$$  W (   ]   V7   &