o i!2@sdZddlZejZeedddkseedddkr&ddlmZnddl mZddl m Z ddl ZddlZddlZddlZddlZddlmZdd lmZdd l mZz dd lmZmZWneysdd lmZmZYnwd#ddZd$ddZ ddd d e!fddZ"d$ddZ#ddd d e!fddZ$d$ddZ%ddd d e!fddZ&d%ddZ'dd d e!fdd Z(d!d"Z)dS)&uk 龙虎榜数据 Created on 2015年6月10日 @author: Jimmy Liu @group : waditu @contact: jimmysoa@sina.cn N.)StringIO)cons)etree)dateu)ref_vars)urlopenRequestMbP?c Cs.|durtdkrt}n t}nt|rdSt|D]}t|ztt j t j dt j d||f}t|dd}|d}|dd }t|td tftd d d }t|}t|}tj|dt jd}t j|_|d}|dd}|dt |d<|dt |d<|dt |d<|dt |d<|d|d|d<|d|d|d<|d!t j"|d<|d!t j"|d<||d<dD]}||t ||<||d||<||!t j"||<q|j#dd d}W|St$y}z t%|WYd}~qd}~wwt&t j')u 获取每日龙虎榜列表 Parameters -------- date:string 明细数据日期 format：YYYY-MM-DD 如果为空，返回最近一个交易日的数据 retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 Return ------ DataFrame code：代码 name ：名称 pchange：涨跌幅 amount：龙虎榜成交额(万) buy：买入额(万) bratio：占总成交比例 sell：卖出额(万) sratio ：占总成交比例 reason：上榜原因 date ：日期 Nhttpem timeoutGBKz_1=rDummycSs|S)N)snrrJ/opt/alphahud/venv/lib/python3.10/site-packages/tushare/stock/billboard.pyJsztop_list..) __getitem__data)columnsrbuysellamountTurnoverbratiosratiodate)r!rr i'axis)(duget_hour last_tddatetoday is_holidayrangetimesleepr rvLHB_URLctP_TYPEDOMAINSr readdecodesplitevaltypedictjsondumpsloadspd DataFrame LHB_TMP_COLSLHB_COLSrfillnareplaceastypefloatmapFORMATdrop ExceptionprintIOErrorNETWORK_URL_ERROR_MSG) r% retry_countpause_requesttextdfcolerrrtop_list sT      "        rUcCsVt|dur)tt|d||d}|dur'|ddd|d<|d}|SdS)u 获取个股上榜统计数据 Parameters -------- days:int 天数，统计n天以来上榜次数，默认为5天，其余是10、30、60 retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 Return ------ DataFrame code：代码 name：名称 count：上榜次数 bamount：累积购买额(万) samount：累积卖出额(万) net：净额(万) bcount：买入席位数 scount：卖出席位数 TrpageNorMrNNcodecSt|dSNstrzfillxrrrrzcap_tops..)r2_check_lhb_input _write_head _cap_topsrFdrop_duplicatesdaysrMrNrRrrrcap_topsfs ric CsZtt|D]}t|zttjtjdtj dtj dtj d||f}t |dd }|d}tjt|}|d} tjrNd d | D} nd d | D} d | } d | } t| d} tj| _|j| dd}|d} t| dkrtd| dd}t|||||WS|WSty} z t | WYd} ~ qd} ~ wwdS)Nrvsfrfdrrr//table[@id="dataTable"]/trcSg|] }t|dqSzutf-8rtostringr6.0noderrr z_cap_tops..cSg|]}t|qSrrrprqrrrrtr%s
T ignore_index(//div[@class="pages"]/a[last()]/@onclick\d+)!r2_write_consoler-r.r/r r0 LHB_SINA_URLr3r4 LHB_KINDSPAGESr r5r6lxmlhtmlparserxpathPY3joinr> read_html LHB_GGTJ_COLSrappendlenrefindallrerIrJlastrXrMrNdataArrrOrPrQrressarrrRnextPagerTrrrre:         recCs.t|durtt|d||d}|SdS)u 获取营业部上榜统计数据 Parameters -------- days:int 天数，统计n天以来上榜次数，默认为5天，其余是10、30、60 retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 Return --------- broker：营业部名称 count：上榜次数 bamount：累积购买额(万) bcount：买入席位数 samount：累积卖出额(万) scount：卖出席位数 top3：买入前三股票 TrrWN)r2rcrd _broker_topsrgrrr broker_topssrc CsZtt|D]}t|zttjtjdtj dtj dtj d||f}t |dd }|d}tjt|}|d} tjrNd d | D} nd d | D} d | } d | } t| d} tj| _|j| dd}|d} t| dkrtd| dd}t|||||WS|WSty} z t | WYd} ~ qd} ~ wwdS)NrrjrrkrrrrlcSrmrnrorqrrrrtruz _broker_tops..cSrvrrwrqrrrrtrxrryrTrzr|r})!r2r~r-r.r/r r0rr3r4rrr r5r6rrrrrrrr>r LHB_YYTJ_COLSrrrrrrrIrJrrrrrrrcCsDt|dur tt|d||d}|ddd|d<|SdS)ud 获取机构席位追踪统计数据 Parameters -------- days:int 天数，统计n天以来上榜次数，默认为5天，其余是10、30、60 retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 Return -------- code:代码 name:名称 bamount:累积买入额(万) bcount:买入次数 samount:累积卖出额(万) scount:卖出次数 net:净额(万) TrrWrYcSrZr[r]r`rrrrrbzinst_tops..N)r2rcrd _inst_topsrFrgrrr inst_topssrc Csltt|D]}t|zttjtjdtj dtj dtj d||f}t |dd }|d}tjt|}|d} tjrNd d | D} nd d | D} d | } d | } t| d} | jddgdd} tj| _|j| dd}|d} t| dkrtd| dd}t|||||WS|WSt y} z t!| WYd} ~ qd} ~ wwdS)NrrjrkrrrrlcSrmrnrorqrrrrt ruz_inst_tops..cSrvrrwrqrrrrt rxrryrr rr&Trzr|r})"r2r~r-r.r/r r0rr3r4rrr r5r6rrrrrrrr>rrH LHB_JGZZ_COLSrrrrrrrIrJrrrrrs<         rcCs<ttd||d}t|dkr|ddd|d<|S)u& 获取最近一个交易日机构席位成交明细统计数据 Parameters -------- retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 Return ---------- code:股票代码 name:股票名称 date:交易日期 bamount:机构席位买入额(万) samount:机构席位卖出额(万) type:类型 rrWrrYcSrZr[r]r`rrrr4rbzinst_detail..)r2rd _inst_detailrrF)rMrNrRrrr inst_details rc CsXtt|D]}t|zttjtjdtj dtj dtj dd|f}t |dd }|d}tjt|}|d }tjrNd d |D} nd d |D} d| } d | } t| d} tj| _|j| dd}|d} t| dkrtd| dd}t||||WS|WSty} z t | WYd} ~ qd} ~ wwdS)Nrrjr rkrrrrrlcSrmrnrorqrrrrtDruz _inst_detail..cSrvrrwrqrrrrtFrxryrTrzr|r})!r2r~r-r.r/r r0rr3r4rrr r5r6rrrrrrrr>r LHB_JGMX_COLSrrrrrrrIrJ) rXrMrNrrOrPrQrrrrRrrTrrrr8s:         rcCsVd|dvr)|d|d<tddD] }||d||<qtddD]}tj||<q!|S)N%r r\ rVr)r-npNaN)rairrr_f_rowsVs   r)Nr r )rVr r )r r )*__doc__pandasr> __version__vintr7ior pandas.compat tushare.stockrr2numpyrr.r;r lxml.htmlrr tushare.utilrr(r r0urllib.requestr r ImportErrorurllib2rUrir?rerrrrrrrrrrrs<,        F"