i*ddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z mZddlZddlZddlmZddlmZmZmZmZddlmZdd lmZmZe d Zd ed efd ZedgdZ edd d0de!deedfd e!fdZ" d1de!deedfde#de$d ee de e ff dZ% d2de!deedfd ee dffdZ&de!de e fdZ'gdfde(d e e!d!e$fd"Z)e(e*fd#e e!e fd$e e!fd%Z+e d&Z,d2d'e e!ge,fd(e d)e,d e,fd*Z- d3d+e!d,e!d-e!d.e$d df d/Z.gZ/dS)4N) OrderedDict namedtuplewraps)AnyCallableDictListTypeVarUnion)retry)FS_DICTMARKET_NUMBER_DICT MagicConfig MarketType)SEARCH_RESULT_CACHE_PATH)SEARCH_RESULT_DICTsessionFfuncreturncgdtfd}dttttfdttttffd|S)u 将 DataFrame 或者 Series 尽可能地转为数字的装饰器 Parameters ---------- func : Callable 返回结果为 DataFrame 或者 Series 的函数 Returns ------- Union[DataFrame, Series] )u 股票代码u 基金代码u代码u 市场类型u 市场编号u 债券代码u行情IDu 正股代码c|i|}t|tjr-|jD]$}|vr||||<%n.run.st&v&& fbl + + ; . C C''%+F^%9%9'%B%BF6N C * * ; ; ;&&$+GF5M$:$:F5M orctjdt|s|S t|rt |}nt |}n#YnxYw|S)Nz\d)refindallstrisalnumintfloat)r,s r)r'zto_numeric..convert=sjz%Q(( H 1vv~~ FF!HH  Ds AA''A+)rr r0r2r3)rr*r'r(s` @@r) to_numericr4s   F 4[[      [  5c5) eCO.D     Jr+Quote)codenamepinyinidjysclassify market_typesecurity_typeName security_typemkt_numtype_usquote_id unified_code inner_code)triesdelayTF stock_coder<c $tt|dkr|rdStdt |f||d|}t |t r|jS||stj d|ddSdS)u 生成东方财富股票专用的行情ID Parameters ---------- stock_code : str 证券代码或者证券名称 market_type : MarketType, optional 市场类型，目前可筛选A股，港股，美股和英股。默认不筛选 use_local : bool, optional 是否使用本地缓存 suppress_error : bool, optional 遇到错误的股票代码，是否不报错，返回空字符串 Returns ------- str 东方财富股票专用的 secid ru#证券代码应为长度不应为 0)r< use_localNu证券代码 "u" 可能有误) lenr0strip Exception search_quoterr5rArichprint)rHr<rKsuppress_errorr$quotes r) get_quote_idrTbs6 3z?? " "##q((  2=>>>   +y  DJ  E%~ } D JB BBB C C Cr}r+keywordcountrKc |r|dkrt}|r|Sd}dfdddt|dff} tj|| }|d d } n%#tj$rt d Yd SwxYw| W| rUfd| D} t| d d|dkrt| dkr| dnd S| d |Sd S)u 根据关键词搜索以获取证券信息 Parameters ---------- keyword : str 搜索词(股票代码、债券代码甚至证券名称都可以) market_type : MarketType, optional 市场类型，目前可筛选A股，港股，美股和英股。默认不筛选 count : int, optional 最多搜索结果数, 默认为 `1` use_local : bool, optional 是否使用本地缓存 Returns ------- Union[Quote, None, List[Quote]] rE)r<z/https://searchapi.eastmoney.com/api/suggest/getinput)type14)token D43BF722C8E33BDC906FB84D85E326E8rV)paramsQuotationCodeTableDataz@unable to parse search quote result, consider if you are blockedNcg|]\}tjdr |dkr.j|dkAt |]S)FCodeNClassify)getrQUOTE_SYMBOL_MODEvaluer5r%).0itemrUr$r<s r) z search_quote..sy    JJ{N)N)N 4;;== !*O)N)Nr+r) search_quote_locallymaxrrdjsonJSONDecodeErrorRuntimeWarningsave_search_resultrL) rUr<rVrKr$rSurlr^ json_responseitemsquotess `` ` r)rOrOsk6UaZZ$W+FFF  L ;C W,5 S]]$% F  C777<<>> 23F;   N   tt   U            7F2A2J/// A:: #F q 0 06!99d :fuf~ 4s6A55BBc\tj|}|H|dr3t|tr |d|jkrdS|d}d}t j}||z |krdS|}|d=tdi|}|S)u2 在本地里面使用搜索记录进行关键词搜索 Parameters ---------- keyword : str 搜索词 market_type : MarketType, optional 市场类型，目前可筛选A股，港股，美股和英股。默认不筛选 Returns ------- Union[Quote,None] N last_timer;i)rrdrrrftimecopyr5)rUr<qrumax_tsnow_qrSs r)rjrjs$ w''A uu[!!  {J / / z"" (9::t~I F )++C i6!!t B ; KKBKKE Lr+rscBttdd5}|D]E}tj}t|}||d<|t |<t jt j|ddddS#1swxYwYdS)u 存储搜索结果到文件中 Parameters ---------- keyword : str 搜索词 quotes : List[Quote] 搜索结果 wzutf-8)encodingruN) openrrwdict_asdictrrldumprx)rUrsfrSr{ds r)roros &g > > >0!  E)++CU]]__%%A AkN*+ w '  $)++Q///000000000000000000sA/BBBfields to_be_removedkeep_allcfd}|S)u: 重命名 DataFrame 和 Series 的列名的装饰器 Parameters ---------- fields : dict 新的表头 to_be_removed : List[str], optional 要移除的列, by default [] keep_all : bool, optional 是否保存全部列(包含未重命名的列), by default True cDtfd}|S)Nc|i|}t|tjrt}r@|jD]}||vr||||}n||}|D] }|vr||= n/t|tjr|}|S)N)r) rrrlistr%rappendrenamer!) r#r$r%rr&rrrrs r)wrapperz?rename_dataframe_and_series..decorator..wrapper%sT4*6**F&",// /v}}//D"(.33!00#NN6222#]]6]::7CFF#]]6]::7CF$++F.."6N+FBI.. /v..Mr+r)rrrrrs` r) decoratorz.rename_dataframe_and_series..decorator$sC t         &r+rv)rrrrs``` r)rename_dataframe_and_seriesrs0 . r+function_fieldsremove_columns_and_indexescfd}|S)u 对 DataFrame 和 Series 进一步操作 Parameters ---------- function_fields : Dict[str, Callable], optional 函数字典 remove_columns_and_indexes : List[str], optional 需要删除的行或者列, by default list() cBtfd}|S)Nc\|i|}t|tjrTD]-\}}||jvr|||||<.D]}||jvr||=n4t|tjrD]}||}|Sr)rrrrrrr r!drop) r#r$r%r& function_namer"rrrs r)rz@process_dataframe_and_series..decorator..wrapperNsT4*6**F&",// 0-<-B-B-D-DII)FMV^33 %+F^%9%9-%H%HF6NN8++F//"6N+FBI.. 0700E#[[//FFMr+r)rrrrs` r)rz/process_dataframe_and_series..decoratorMs> t        r+rv)rrrs`` r)process_dataframe_and_seriesr>s*& r+Trrfdefaultc< ||}|S#||cYS|cYSxYw)u( 类型转换 Parameters ---------- f : Callable[[str], T] 转换函数 value : Any 待转换的值 default : T, optional 转换失败时的返回值, 默认为 ``None`` 表示原样返回 Returns ------- T 转换结果 rv)rrfrs r)to_typerfs9&%  ?LLLs category market_number market_namedrop_duplicatec|t|<tj|d}|d|}|rDdt j|dt|<dS|t|<dS)u 添加市场 Parameters ---------- category : str 市场类别 market_number : str 市场编号 market_name : str 市场名称 drop_duplicate : bool, optional 是否去重, 默认为 ``True`` rJz,m:,N)rrrdjoinrfromkeyssplit)rrrroldnews r) add_marketrs|")4}% +h # #C  $ $] $ $C HH[%9#))C..%I%IJJr+)NTF)NrETr)T)0rlr.rw collectionsrr functoolsrtypingrrr r r r pandasrrP retry.apir common.configrrrrconfigrsharedrrrr4r5r0rTr2boolrOrjrorrrrrrr__all__rvr+r)rs ////////<<<<<<<<<<<<<<<< PPPPPPPPPPPP------00000000 GCLL5Q515555r     *Qa,0 &&&z4'(&  &&&&V,0 BB Bz4'(B B B  5$U #$ BBBBL:>** *$Z%56* 5$;****Z00T%[0000..0$'' '!%c';?''''V,0466,0DFF""#x-(" $S """"J GCLLxq!#Q:QU   "% 47 IM      4 r+