iFddlmZddlmZmZmZddlZddlZddlZ ddl m Z ddl m Z ddl m Z ddlZddlmZdd lmZmZdd lmZmZmZdd lmZmZd d lmZmZmZmZm Z m!Z!m"Z"ede#de j$fdZ%e d/de#de#de#de&de&deedfde'de'de j$fdZ( d0d ee#de#de#de&de&d!e&deedfde'de'dee#e j$ffd"Z) d/d ee#ee#fde#de#de&de&deedfde'de'dee j$ee#e j$fffd#Z*ede#de j$fd$Z+ede#de j$fd%Z,ed&e#de j-fd'Z.ed1d&e#d)e&de j$fd*Z/ed+ee#ee#fde j$fd,Z0ed2de#d-e&de j$fd.Z1dS)3)datetime)DictListUnionN)jsonpath)retry)tqdm)ThreadPoolExecutor)MARKET_NUMBER_DICT MarketType)BASE_INFO_CACHEsessionMAX_CONNECTIONS) get_quote_id to_numeric)EASTMONEY_BASE_INFO_FIELDSEASTMONEY_HISTORY_BILL_FIELDSEASTMONEY_KLINE_FIELDSEASTMONEY_KLINE_NDAYS_FIELDSEASTMONEY_QUOTE_FIELDSEASTMONEY_REQUEST_HEADERS MagicConfigfsreturnc it|tjiddt dt ffd }dtjfd}|dd }|d d }t|d d }t||\}}|r|dzn|} t5} | |td| dz|g| z} t| } d d d n #1swxYwYfd| D} tj| ddddd|}|dt(dz|dt(z|d<|dt(d|d<|dd|d<tj|dd !t(|d<|d}|d=||d<|d=|S)"u 获取沪深市场最新行情总体情况 Returns ------- DataFrame 沪深市场最新行情信息（涨跌幅、换手率等信息） ,pnpzc d|fd|fddddddfd ff }d }tj|t| }|S) Nrr )po1)npr#fltt2invtr')fidf12rfieldsz+http://push2.eastmoney.com/api/qt/clist/getheadersparams)rgetrjson)rr r/url json_responser,rs A/usr/local/lib/python3.11/dist-packages/efinance/common/getter.py get_by_pagez.get_realtime_quotes_by_fs..get_by_page)so 2J 2J      2J v   < 26   $&& seriesctj|jt}|D]/\}}t |t rtj||<*|||<0|S)Nindexdtype) pdSeriesr:floatitems isinstancestrr$nan)r7 sort_keysidxvals r4mixed_sort_keyz1get_realtime_quotes_by_fs..mixed_sort_key<scIFL>>>   % %HC#s## %!# #!$ #r6r)r datatotaldiffNcg|]G}tj|ddtHS)rHrJ)r< DataFramelistkeys).0responsecolumnss r4 z-get_realtime_quotes_by_fs..QsR     Xf%f-..tGLLNN/C/CD   r6rTaxis ignore_indexrQu 涨跌幅F)by ascendingrUkey 市场编号.代码行情IDc*tj|SN)r r0xs r4z+get_realtime_quotes_by_fs..bs7I7Ma7P7Pr6 市场类型更新时间戳cDttj|Sr_rAr fromtimestampr`s r4rbz+get_realtime_quotes_by_fs..ds8CYZ[C\C\?]?]r6 更新时间最新交易日%Y%m%dformat)rr0r EXTRA_FIELDSjoinrNintr<r=lendivmodr maprangerMconcatrenamevalues sort_valuesastyperAapply to_datetime)rkwargsr5rFr3rIr divmodpagesexecutortasks responsesdfsdftmprQr,s` @@r4get_realtime_quotes_by_fsrsU'T6::k6NPR+S+STG XXgllnn % %F& ry     Kc***M & !' *E ]6 "6 * + +BeR  HC #C!GGE     [%519*=*=te|LLKK                    !   C #AD111  !1!1 3    '..s33c9BxL!!#&&,,-P-PQQ~-.445]5]^^B~N2.?+@RRRYYZ]^^B  C B  Is.$ 1 1 1EKK   1 1 1r6rHnamer[)rMrrNrwror0r QUOTE_ID_MODErrrr1rinsertr<rLr)rrrrrrrrr|r,rQrquote_idr/r2r3klinesrowsrrs r4get_quote_history_singlerms"(-// 0 0F)02233GhhvG zz++,,   #")       B G    ( 3 3 F BCK .ve dff!??F -q(###q(###|G,,,, 1 1& 1 1 1D   (D >>#  r "D dG , , ,BIIa4   IIa4   Ir6codestriesc . i t|} tjt|ddtf  f d } t | |D]J} ttjtkrtj d| | Ktj   tjrtj dd S S) u8 获取多只股票、债券历史行情信息 r)rdelayrc  t|f d}||< d d|dS)NrrrrrrrrzProcessing => )rupdateset_description_str) r_dfrrrrrr|rpbarrrs r4startz&get_quote_history_multi..starts}'   #)%      D  A   !8$!8!899999r6)rIrrTrS)rq multitaskingtaskrrAr get_active_tasksrtimesleepwait_for_taskscloser0r RETURN_DFr<ru)rrrrrrrrrr|rIrrrrs ```` ```` @@r4get_quote_history_multirs>"$&C JJE a   :C::::::::::::::! : e   D |,.. / // A A JqMMM d !!!JJLLL zz+'((9y148888 Jr6c t|trt|f|||||||d|St|dr$t |}t |f|||||||d|St d)u9 获取股票、ETF、债券的 K 线数据 Parameters ---------- codes : Union[str,List[str]] 股票、债券代码 或者 代码构成的列表 beg : str, optional 开始日期，默认为 ``'19000101'`` ，表示 1900年1月1日 end : str, optional 结束日期，默认为 ``'20500101'`` ，表示 2050年1月1日 klt : int, optional 行情之间的时间间隔，默认为 ``101`` ，可选示例如下 - ``1`` : 分钟 - ``5`` : 5 分钟 - ``15`` : 15 分钟 - ``30`` : 30 分钟 - ``60`` : 60 分钟 - ``101`` : 日 - ``102`` : 周 - ``103`` : 月 fqt : int, optional 复权方式，默认为 ``1`` ，可选示例如下 - ``0`` : 不复权 - ``1`` : 前复权 - ``2`` : 后复权 market_type : MarketType, optional 市场类型，目前可筛选A股，港股，美股和英股。默认不筛选，可选示例如下 - ``A_stock`` : A股 - ``Hongkong`` : 香港 - ``London_stock_exchange`` : 英股 - ``US_stock`` : 美股 suppress_error : bool, optional 遇到错误的股票代码，是否不报错，返回空的DataFrame use_id_cache : bool, optional 是否使用本地缓存的东方财富股票行情ID Returns ------- Union[DataFrame, Dict[str, DataFrame]] 股票、债券的 K 线数据 - ``DataFrame`` : 当 ``codes`` 是 ``str`` 时 - ``Dict[str, DataFrame]`` : 当 ``codes`` 是 ``List[str]`` 时 r__iter__u$代码数据类型输入不正确！)r@rArhasattrrMr TypeError) rrrrrrrrr|s r4get_quote_historyrs@% '   #)%        # #  U &   #)%       : ; ;;r6cttj}ttj}d|}t |}ddd|fdd|ff}d}t j|t| }t|d }|sA| d d | d d tj | Sd|D} t|dd } |dd}tj | | } | d d || d d | | S)u 获取单支股票、债券的历史单子流入流出数据 Parameters ---------- code : str 股票、债券代码 Returns ------- DataFrame 沪深市场单只股票、债券历史单子流入流出数据 r)lmt100000)r101rrz f1,f2,f3,f7rz=http://push2his.eastmoney.com/api/qt/stock/fflow/daykline/getr-rrr\rrVc8g|]}|dSrrrs r4rRz$get_history_bill..brr6$..namer[r)rMrrNrwrorrr0rr1rrr<rLr) rr,rQrrr/r2r3rrrrs r4get_history_billr<sx"/466 7 7F0799::GhhvGD!!H (" G F JCK .v dff!??F -q(###q(###|G,,,, 1 1& 1 1 1D M9 - -a 0D >>#  r "D dG , , ,BIIa4   IIa4   Ir6cFt|}ddd|fddf}d}tj|t|}gd}t |d d }|d d }t |d }|sA|d d|d dtj |Sd|D}tj ||} | d d|| d d|| S)u  获取单只股票最新交易日的日内分钟级单子流入流出数据 Parameters ---------- code : str 股票、债券代码 Returns ------- DataFrame 单只股票、债券最新交易日的日内分钟级单子流入流出数据 )r0)rr#rr)rz3f51,f52,f53,f54,f55,f56,f57,f58,f59,f60,f61,f62,f63z7http://push2.eastmoney.com/api/qt/stock/fflow/kline/getr-)时间u主力净流入u小单净流入u中单净流入u大单净流入u超大单净流入rrr[rrr\rrVc8g|]}|dSrrrs r4rRz"get_today_bill..rr6) rrr0rr1rrrr<rL) rrr/r2r3rQrrrrs r4get_today_billrlsE D!!H ("J F DCK .v dffG M9 - -a 0D >>#  r "D ??F -q(###q(###|G,,,, 1 1& 1 1 1D dG , , ,BIIa4   IIa4   Ir6rcdtj}dddd|fd|ff}d}tj|t |}|d }|s'tjtj d Stj|d  t }|S)Nr)ut fa5fd1943c7b386f172d6893dbfba10br(r%r,rz+http://push2.eastmoney.com/api/qt/stock/getr-rHobjectr9)r;)r:) rorrNrr0rr1r<r=rwrv)rr,r/r2r3r?ss r4 get_base_inforsXX0577 8 8F2 6 ( F 8CK .v dff & !E Ty9@BB(SSSS %x(((//6P/QQA Hr6@B max_countctj|t|}|t|<gd}t|ddkrt j|S|d}|d}d|fddd d t|ff}tjd | }| }|d d} d| D} t j|tt| } || j dddf<|| j dddf<t j| gd} | dd|d d| j| j dd| jf<| S)u 获取股票、期货、债券的最新交易日成交情况 Parameters ---------- quote_id : str 包含市场编号的股票、期货、债券代码 max_count : int, optional 最大数据条数, 默认为 ``1000000`` Returns ------- DataFrame 股票、期货、债券的最新交易日成交情况 Notes ------ 返回的数据表头: ``['名称', '代码', '时间', '昨收', '成交价', '成交量', '单数']`` )rr\r昨收 成交价 成交量单数r\rBrVrr)rzf1,f2,f3,f4,f5)rzf51,f52,f53,f54,f55pos-z4https://push2.eastmoney.com/api/qt/stock/details/getr/rHdetailscHg|]}|ddd S)rNr)rOlines r4rRz#get_deal_detail..s+ 2 2 2DDJJsOOBQB  2 2 2r6)rQr:N)rrrrrrprePrice)rr0rrAlowerr<rLrprr1rtrqlocrrwrQ) rr base_inforQrrr/rPjslinesrr detail_dfs r4get_deal_detailrs* #HmH.E.EFFI )OHZZZG 9X %%''500|G,,,, X D X D (%* $C NN$$% F{>vH}}B&z),E 2 2E 2 2 2D gU3t99-=-= > > >BBF111h;BF111h; T+Y+Y+YZZZI Q"V*Z"8999#,#3BF111i  Ir6 quote_id_listc 0t|tr|g}|}it|tji}d|}ddd|fddddd|fd d f }d }tj|t|  }t|d }|s(tj |} nMtj |t||} | dd| d<| dtdz| dtz| d<| d=| dd| d<tj| ddt| d<| d} | d=| | d<| d=| S)u 获取股票、期货、债券的最新行情 Parameters ---------- quote_id_list : List[str] 带市场编号的行情ID或者行情ID组成的列表格式如下 - ``'1.600159'`` - ``['1.600159','0.300750']`` 市场编号参见文件 ``efinance/common/config.py`` 中的 ``MARKET_NUMBER_DICT`` Returns ------- DataFrame 股票、期货、债券的最新行情 Notes ------ 返回的数据表头: ``['代码', '名称', '涨跌幅', '最新价', '最高', '最低', '今开', '涨跌额', '换手率', '量比', '动态市盈率','成交量', '成交额', '昨日收盘', '总市值', '流通市值', '市场类型', '行情ID']`` r) OSVersionz14.3) appVersion6.3.8r,r%)platIphone)productEFundsecids) serverVersionz6.3.6)versionrz/https://push2.eastmoney.com/api/qt/ulist.np/getr-z $..diff[:]rVrZcDtjt|Sr_)r r0rAr`s r4rbz"get_latest_quote..s.rhr6rirjrkrl)r@rArr0rrnrorNrrr1rr<rLrwrMrvrzryr{) rr|rrQr,r/r2r3rrrs r4get_latest_quoters.-%%(& %FT'T6::k6NPR+S+STG XXgllnn % %F 6 388F##$" F -.445]5]^^B~N2.?+@RRRYYZ]^^B  C B  Ir6ndaysc ttj}ttj}d|}|t jr|}nt|}dd|fd|fddd|ff}tjd|  }t|d } | sA| d d | d d tj|Sd| D} |dd} |dd}tj| |} | d d || d d | | S)u 获取股票、期货、债券的最近 ``ndays`` 天的1分钟K线行情 Parameters ---------- code : str 代码、名称或者行情ID 如果是行情ID则需传入 ``quote_id_mode=True`` ndays : int, optional 天数 默认为 ``1`` 最大为 ``5`` Returns ------- DataFrame 股票、期货、债券的最近 ndays 天的1分钟K线行情 rrrr)iscrr)isccarrz6http://push2his.eastmoney.com/api/qt/stock/trends2/getrz $..trends[:]rr\rrVc8g|]}|dSrrrs r4rRz*get_latest_ndays_quote..Qrr6rHrr[r)rMrrNrwror0rrrrr1rrr<rLr) rrr|r,rQrrr/r3rrrrs r4get_latest_ndays_quoter%s$.355 6 6F/68899GhhvG zz++,,&%%A G % ( FK@ dff!??F -q(###q(###|G,,,, 1 1& 1 1 1D   (D >>#  r "D dG , , ,BIIa4   IIa4   Ir6)rrrrNFT)rrrrrNFT)r)r)2rtypingrrrnumpyr$rpandasr<rr tqdm.autor rconcurrent.futuresr common.configr r sharedrrrutilsrrconfigrrrrrrrrArLrrpboolrrrrrr=rrrrrr6r4rs$$$$$$$$$$ 111111::::::::>>>>>>>>>>,,,,,,,, O#OBLOOO Od +/ :: : : : : : z4'( :::\::: :~+/ 11 91 1 1 1 1  1z4'(111 #r| 1111l+/ Z<Z< d3i Z< Z< Z< Z< Z< z4'( Z<Z<Z< 2<c2</0 01Z<Z<Z<Z