X-Ph"HUdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z mZmZddlmZddlZeeefZded<ejd Zd ed <ejd Zd ed <d?dZGddZGddeZdZd ed<dZ d ed<dZ!d ed<dZ"d ed<dZ#d ed <d!Z$d ed"<d#Z%d ed$<Gd%d&Z&d@d+Z'dAd-Z(dBd.Z)dCd/Z*dDd2Z+dEd6Z,dFd9Z-dGd=Z.dHd>Z/dS)IzParsing/inferring signatures from documentation. This module provides several functions to generate better stubs using docstrings and Sphinx docs (.rst files). ) annotationsN)MutableMappingMutableSequenceSequence)AnyFinal NamedTuple) TypeAlias _TypeAliasSigz5^[a-zA-Z_][\w\[\], .\"\'|]*(\.[a-zA-Z_][\w\[\], ]*)*$r_TYPE_REz\**[A-Za-z_][A-Za-z0-9_]*$ _ARG_NAME_REsstrreturnboolcZ|dvrdSd|vrd|vrdSt|duS)zCTry to determine whether a string might be a valid type annotation.)TrueFalseretvalF,[N)r match)rs L/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/mypy/stubdoc.py is_valid_typers@ '''u axxCqLLu >>!  D ((cFeZdZdZ ddddddZddZddZddZddZdS)ArgSigz%Signature info for a single argument.NFz...)default default_valuenamertype str | Nonerrr rNonec>||_||_||_||_dSN)r!r"rr )selfr!r"rr s r__init__zArgSig.__init__(s'   *rcl|jdo|jd S)N***r! startswithr's r is_star_argzArgSig.is_star_arg6s0y##C((K1E1Ed1K1K-KKrc6|jdS)Nr+r,r.s r is_star_kwargzArgSig.is_star_kwarg9sy##D)))rcdt|jt|jt|jS)Nz$ArgSig(name={}, type={}, default={}))formatreprr!r"rr.s r__repr__zArgSig.__repr__<s;5<< OOT$)__d4<.@.@   rotherrct|tr@|j|jko/|j|jko|j|jko|j|jkSdS)NF) isinstancerr!r"rr )r'r6s r__eq__z ArgSig.__eq__Asc eV $ $  UZ'>I+>LEM1>&%*==   urr&) r!rr"r#rrr rrr$rr)rr)r6rrr) __name__ __module__ __qualname____doc__r(r/r1r5r9rrrr%s//   + " + + + + + +LLLL****    rrcfeZdZUded<ded<ded<dZded<dd Zdd Zdd Z dddZdS) FunctionSigrr! list[ArgSig]argsr#ret_type type_argsrrct|jdo4|jdo|jo|jdjdvS)N__rr'cls)rr!r-endswithrCr.s ris_special_methodzFunctionSig.is_special_methodRsa I  & & 5 ""4(( 5  5 ! !_4    rc&|jr$|jdjdvr|jdd}n|j}t|dkoLtd|Do3|do|dS)z=Return if this signature has catchall args: (*args, **kwargs)rrINc3(K|] }|jdvVdS))Nobjectr typing.AnyN)r").0as r z0FunctionSig.has_catchall_args..bs*RRAFCCRRRRRRr)rCr!lenallr/r1)r'rCs rhas_catchall_argszFunctionSig.has_catchall_argsZs 9 1*o==9QRR=DD9D IIN (RRTRRRRR (Q##%% (Q%%''  rc<|o|jdvS)zKReturn if this signature is the catchall identity: (*args, **kwargs) -> Any)NrrR)rXrDr.s ris_catchall_signaturez!FunctionSig.is_catchall_signaturegs!%%''XDM=X,XXrFNindentis_asyncany_val docstringc Fg}|jD]}|j}|tjvrd|z}|j(|&|jdvr|jds|}n|j}|r|d|zz }|jr |d|jz }n|jr |d|jz }||d} |j r|j n|} | d| z} |rd nd} || d |j|j d d |d | d } |r&d|dtj |} nd} | | S)N_rIr*z: z = =rEz -> zasync zdef (, ): z z ...)rCr!keywordkwlistr"r-rr appendrDrFjoinmypyutilquote_docstring)r'r[r\r]r^rCargarg_defarg_typeretfieldrDprefixsigsuffixs r format_sigzFunctionSig.format_sigks9 ! !ChG'.((- 'HO33++C004(/8 34(?*;98S%6888G 32s0222 KK $(M>4==w  (H%-2^^^TY^^^4^^S[^^^  L&LLdi&?&? &J&JLLFFFvrr:)rEFNN) r[rr\rr]r#r^r#rr) r;r<r=__annotations__rFrLrXrZrur?rrrArALs IIII         YYYY " $ , , , , , , , rrArN STATE_INITrOSTATE_FUNCTION_NAMESTATE_ARGUMENT_LISTSTATE_ARGUMENT_TYPESTATE_ARGUMENT_DEFAULTSTATE_RETURN_VALUESTATE_OPEN_BRACKETc2eZdZdZddZdd Zdd Zdd Zd S)DocStringParserz+Parse function signatures in documentation. function_namerrr$c||_tg|_d|_d|_d|_d|_d|_d|_g|_ d|_ d|_ g|_ dS)NrErF) rrwstate accumulatorrparg_name arg_defaultrDfoundrCpos_only keyword_only signatures)r'rs rr(zDocStringParser.__init__s`* \ $(  '+  "$ $( (,-/rtokentokenize.TokenInfoc |jtjkrG|j|jkr7|jdt kr!|jtd S|jtj kri|jdkr^|jdtkrH|j d|_ d|_ |jtd S|jdtkr|j d S|jtj krU|jdvrL|jdt kr6|xj |jz c_ |jtd S|jtj krO|jdvrF|jdtkr0|xj |jz c_ |j d S|jtj krU|jdkrJ|jdtkr4|j |_d|_ |jt d S|jtj kr|jdkr|jdtt fvrp|jdt kr&|j |_|j n |j |_d|_ |jt$d S|jtj kr|jd vr|jdtt$t fvr|jdt$kr'|j |_|j n:|jdt kr&|j |_|j n|jdtkr|j d kr>|j|d St-|j|_d|_ n|j d rt-|jd z|_|j |_|jd kr|j dks5t4|js|d S|jd krp|jdtkrA|j:|jt-|jkr|js|d S|j |jr|jrVt9|jsB|jt;|jd t=|jnF|jt;|j|jt=|jd|_d |_d |_d|_ d S|jtj kr|jdkr|jdtkry|jdkrl|j|j|js|d St-|j|_|jt d|_ d Sd S|jtj krI|jdkr>|jdt kr(d|_ |jt@d S|jtj!tj"fvr|jdt t@fvr|jdt@krVt9|j s|d S|j |_#d|_ |j |j r@|j$tK|j|j|j#d|_ g|_d|_#d S|xj |jz c_ d S)z)Process next token from the token stream.rbrET)rrb{)]rd}rera)rrdr*NrNrd)r!r"r/z->)r!rCrDFr)&r"tokenizeNAMEstringrrrwrirxOPpoprrrzrrr|rpr~rrresetrVrCr-striprrrrrrrNEWLINE ENDMARKERrDrrA)r'rs r add_tokenzDocStringParser.add_tokensh J(- ' '  222 2*,, J  1 2 2 2 2 2 J(+ % % ## 2"555 JNN   !D DJ J  1 2 2 2 2 2 Z^2 2 2 JNN      J(+ % % // 2*,,    ,   J  0 1 1 1 1 1 J(+ % % // 2"444    ,   JNN      J(+ % % ## 2"555 ,DM!D  J  1 2 2 2 2 2 J(+ % % ## 2#68K"LLLz"~!444 $ 0      $ 0 !D  J  4 5 5 5 5 5 J(+ % % ** 2#%;=PQRRz"~!777#'#3      B#666 $ 0      B#666#s**(4 (+DID%')D$$'22377?,/ NNQ,>)$($4DM ++0@0F0F0H0HB0N0N*00??1O |s""JrN&999)5)S^^;; M<JJLLLF    } = t})E)E I$$DMd4K[F\F\]]]I$$!%T]DQUQaLbLb DM DM#D !D    J(+ % % ## 2"555|s""=,0A0MUYU^0M JJLLLF #DI  !!"5666#%   #"Z8; & &5<4+?+?DJrNV`D`D`!D  J  0 1 1 1 1 1ZH,h.@A A AdjQSn  Y G G z"~!333$T%566JJLLLF $ 0 #%     z #&&T%7diRVR_```# DI!DMMM    ,    rcJtg|_g|_d|_d|_dS)NFrE)rwrrCrrr.s rrzDocStringParser.reset]s' \   rlist[FunctionSig]cLd dd fd t|jfd  S) z:Return sorted copy of the list of signatures found so far.r!r signaturerArrcDtfd|jDS)Nc3.K|]}|jkVdSr&r!)rSxr!s rrUzBDocStringParser.get_signatures..has_arg..gs)>>!qv~>>>>>>r)anyrC)r!rs` rhas_argz/DocStringParser.get_signatures..has_argfs(>>>>y~>>>>> >rc4d|o d|S)Nz*argsz**kwargsr?)rrs r args_kwargsz3DocStringParser.get_signatures..args_kwargsis&77I..Q77:y3Q3Q Qrc"|rdndS)NrNrr?)rrs rz0DocStringParser.get_signatures..ms++a..5OQQar)key)r!rrrArr)rrArr)sortedr)r'rrs @@rget_signatureszDocStringParser.get_signaturescs] ? ? ? ? R R R R R Rdo+O+O+O+OPPPPrN)rrrr$)rrrr$)rr$)rr)r;r<r=r>r(rrrr?rrrrsu550000 d-d-d-d-L Q Q Q Q Q Qrrdocstrr#r!list[FunctionSig] | Nonect|tr|sdSt|}tjt j5 t jtj| dj }|D]}| |n#t$rYddddSwxYw dddn #1swxYwY| }d dfd|DS) a=Convert function signature to list of FunctionSig Look for function signatures of function in docstring. Signature is a string of the format () -> or perhaps without the return type. Returns empty list, when no signature is found, one signature in typical case, multiple signatures, if docstring specifies multiple signatures for overload functions. Return None if the docstring is empty. Arguments: * docstr: docstring * name: name of function for which signatures are to be found Nzutf-8rsrArrcjt|jtd|jDkS)z1return true if function argument names are uniquech|] }|j Sr?rrSrns r zCinfer_sig_from_docstring..is_unique_args..s$B$B$B#SX$B$B$Br)rVrC)rss ris_unique_argsz0infer_sig_from_docstring..is_unique_argss.38}}$B$B$B$B$B C CCCrc*g|]}| |Sr?r?)rSrsrs r z,infer_sig_from_docstring..s( 7 7 7C>>##6#6 7C 7 7 7r)rsrArr)r8rr contextlibsuppressr TokenErrorioBytesIOencodereadlinerIndentationErrorr)rr!rtokensrsigsrs @rinfer_sig_from_docstringrps vs # #t D ! !E  X0 1 1 &rz&--2H2H'I'I'RSSF ' '&&&& '       '    ! !DDDDD 8 7 7 74 7 7 77s7C  AB#"C # B<-C ;B<<C  CCrBcJtd|zd}|r |djSgS)zXConvert signature in form of "(self: TestClass, arg0: str='ada')" to List[TypedArgList].stubr)rrC)rrets r!infer_arg_sig_from_anon_docstringrs. "6F?F ; ;C 1v{ IrcDt||}|r |djSdS)zWConvert signature in form of "func(self: TestClass, arg0) -> int" to their return type.rN)rrD)rr!rs r!infer_ret_type_sig_from_docstringrs* "64 0 0C 1v 4rcd|D}td|dS)zSConvert signature in form of "(self: TestClass, arg0) -> int" to their return type.cg|]@}|d)d|zAS)rbr)rr-)rSlines rrz:infer_ret_type_sig_from_anon_docstring..sA c c ctdjjllF]F]^aFbFb cVdjjll " c c crrEr) splitlinesrrj)rliness r&infer_ret_type_sig_from_anon_docstringrs; c cv/@/@/B/B c c cE ,RWWU^^V D DDrrs'tuple[str, list[str], list[str]] | Nonecptjd|}|sdS|d}|dd}|d}|s|ggfSd|dD}g}g}d }|t |kr||d s d ||vrne|||d |dz }||dz  d rn|t |k|t |kre||}|d }|d d }|||dz }|t |ke|||fS) zSplit function signature into its name, positional an optional arguments. The expected format is "func_name(arg, opt_arg=False)". Return the name of function and lists of positional and optional argument names. z([.a-zA-Z0-9_]+)\(([^)]*)\)NrN.rrOc6g|]}|Sr?)rrs rrz#parse_signature..s 9 9 9CCIIKK 9 9 9rrrrraz[]) rergroupsplitrrVr-rirstriprK) rsmr! arg_stringrC positionaloptionalirns rparse_signaturers /55A t 771::D ::c??2 DJ     R| 9 9:#3#3C#8#8 9 9 9DJH A c$ii-- 7  c " " cT!Wnn $q'..--... Q A;   $ $   c$ii-- c$ii--1giiooiinnQ Q c$ii-- X %%rr Sequence[str]rc"g}|||D]E}|dr||-||dFdd|d}|dd}|S)zNBuild function signature from lists of positional and optional argument names.r*z=...rbrcrdz(self)rE)extendr-rirjreplace)rrrCrnrss rbuild_signaturers!#DKK && >>#   & KK     KK3 % % % % diioo C ++h # #C Jrrtuple[list[Sig], list[Sig]]cg}g}|D]}|}tjd|}|r|dd}t |}|rj|\}}} |ddkr&||t|| f||t|| ft|t|fS)ztParse all signatures in a given reST document. Return lists of found signatures for functions and classes. z,\.\. *(function|method|class) *:: *[a-zA-Z_]z::rNclass) rrrrrrrirr) rr class_sigsrrrsparsedr!fixedrs rparse_all_signaturesrs DJ P Pzz|| HDd K K  P**T""1%++--C$S))F P(.%eX771::((KKuh'G'G HIIII%%t_UH-M-M&NOOO $<< ++ ++rr Sequence[Sig] list[Sig]c8i}|D].\}}||g|/g}|D]B\}}tt |dkr|||dfCt |S)z-Remove names with duplicate found signatures.rNr) setdefaultriitemsrVsetr)rsig_mapr!rsresult name_sigss rfind_unique_signaturesrs.0G11 c4$$++C0000 F"==??00i s9~~  ! # # MM41. / / / &>>rcj|sdSd}tj||}|r|dndS)agCheck for Google/Numpy style docstring type annotation for a property. The docstring has the format ": ". In the type string, we allow the following characters: * dot: because sometimes classes are annotated using full path * brackets: to allow type hints like List[int] * comma/space: things like Tuple[int, int] Nz^([a-zA-Z0-9_, \.\[\]]*): rN)rrr)rtest_strrs rinfer_prop_type_from_docstringrs@ t,H 6""A $1771:::$r)rrrr)rr#r!rrr)rrrrB)rrr!rrr#)rrrr#)rsrrr)rrrrrr)rrrr)rrrr)rr#rr#)0r> __future__rrrrgrrcollections.abcrrrtypingrrr typing_extensionsr r mypy.utilrktuplerr rvcompiler rrrrArwrxrzr|r~rrrrrrrrrrrrr?rrrs #"""""  EEEEEEEEEE))))))))))555555S/!!!!"*UVVVVVV bj!>?? ????))))$$$$$$$$NK K K K K *K K K ^  !!!!!IQIQIQIQIQIQIQIQX"8"8"8"8JEEEE "&"&"&"&J    ,,,,.     % % % % % %r