j?dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl mZdZejdeZd Zd Zd Zdd ZdZdZdefdZdZdZddZdS)z> My own variation on function-specific inspect-like features. N)islice)open)pformatzAargs varargs varkw defaults kwonlyargs kwonlydefaults annotations FullArgSpecc|j|jt|ddg|j|j|j|jtt|j |j |j |j f}|jD]a}t|t jr#|t'|?|t|bt)jdd}|d|dd |S) aDigest the contents of a code object. ``hash(code)`` would be simpler, but it hashes the strings it is built from and those are salted by PYTHONHASHSEED. This value is written to func_code.py and compared by *other* processes, which then conclude the function changed and wipe its cache directory (#1694). The fields below mirror CPython's own ``code_hash()``: https://github.com/python/cpython/blob/5918085bb6f4a3a48193cacb9bb99b044d4e0452/Objects/codeobject.c#L2608-L2624 co_localsplusnames is spelled out as varnames/cellvars/freevars. Only the position information, co_firstlineno and co_linetable, is deliberately left out: joblib compares source *text* and tracks the line number separately, so moving a function without editing it must not invalidate its cache. For the same reason co_filename is not included, unlike ``marshal.dumps``, which would key a notebook function to the cell number it last ran in. co_exceptiontablemd5F)usedforsecurity zutf-8replace)errors)co_nameco_codehexgetattrco_names co_varnames co_cellvars co_freevarsmaprepr co_argcountco_posonlyargcountco_kwonlyargcountco_flags co_consts isinstancetypesCodeTypeappend_code_fingerprinthashlibnewupdatejoinencode hexdigest)codepartsconstdigests v/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/joblib/func_inspect.pyr#r#sZ$  )3//3355                '&     E(&& eU^ , , & LL*511 2 2 2 2 LLe % % % % [ 6 6 6F MM$))E""))'))DDEEE     r cd} |j}|j}tj|sdt j|d}d}|dr>tj d| \}}t|}d|z}|||fSt|5}|j}tt!||dz d}dddn #1swxYwYdt j|||fS#t%|drt'|j|d fcYSt)||d fcYSxYw) a/Attempts to retrieve a reliable function code hash. The reason we don't use inspect.getsource is that it caches the source, whereas we want this to be modified on the fly when the function is modified. Returns ------- func_code: string The function code source_file: string The path to the file in which the function is defined. first_line: int The first line of the code in the source file. Notes ------ This function does a bit more magic than inspect, and is thus more robust. Nrrz z __code__)r1 co_filenameospathexistsr'inspectgetsourcelines startswithrematchgroupsintopen_py_sourceco_firstlinenolistrgetblockhasattrr#r)func source_filer* source_codeline_nosource_file_obj first_line source_liness r. get_func_coderJKs*K /}& w~~k** 5'''"8">">q"ABBKG%%k22 ;')x4k((&((% Wg,,,{:  W4 4 K ( ( OO,J Q M MNNL O O O O O O O O O O O O O O Owww' 5566 ZOO / 4 $ $ /$T]33["D D D D ::{B. . . .s<B3D18D1)C<0 D1<DD1D,D11)E0E0cddl}t|dr|j}nddl}|jj}dD]!}||||}"|S)z2Windows cannot encode some characters in filename.rNquote)<>!:\)urllibrBrL urllib.parseparser)stringrRrLchars r._clean_win_charsrWsoMMMvw#   "*33eeDkk22 Mr Tct|dr|j}nE tj|}n/#t$r"t|dr |jj}nd}YnwxYw|d}|dkr. t jtj |}n #d}YnxYw|| t j }|d drG|d d }d |dd |d dz|d<n3t|d kr |d  d rd|d <d |}|dr |dd}|d z|z}| d}t|dr|j}nt|dr|j}nd}|r-t|dr||jvr|j||urd|z}t|dr@|j|kr5||j dddtj|r1t|dr!|j}||jt jdkr|rt3|}d|D}||fS)aReturn the function import path (as a list of module names), and a name for the function. Parameters ---------- func: callable The func to inspect resolv_alias: boolean, optional If true, possible local aliases are indicated. win_characters: boolean, optional If true, substitute special characters using urllib.quote This is useful in Windows, as it cannot encode some filenames __module__ __class__unknownNr0__main__r2zz!get_func_name..s!666!"1%%666r )rBrYr7 getmodule TypeErrorrZr4r5abspath getsourcefilesplitsepr9r'lenendswithrerfrgrhextendismethodrir"namerW) rC resolv_aliaswin_charactersmodulefilenamer+rtrzklasss r. get_func_namerst\"" # #&t,,FF # # #t[)) #2"  # ~  ww'%%66v666 4<s/)AA*1BB"c.d||S)z.Helper function to output a function signaturez{}{})format) function_namearg_sigs r._signature_strrs == 0 00r cd}t|dd}dd|D}||||S)z)Helper function to output a function callz {0}({1}, {2})rr2, c3*K|]\}}|d|VdS)=Nrl)rmkvs r. z'_function_called_str..s1GG1aaa+GGGGGGr )rr'itemsr)rargskwargs template_strargs_str kwargs_strs r._function_called_strrs["LDzz!B$HGG GGGGGJ   }h C CCr rlc t|}t|tr#td|dt |dt j|s5t j|s!|rtj d|dd||dSt j |}g}g}d }d }t|j D]} | j| jur|| jnT| j| jur|| jn+| j| jur| j}n| j| jur| j}| j| jur| j| j<t j|rK|jg|z}t j |j} t3t5| j } | g|z}t7|d \} } t9|D]\}}|t;|krF|||<||vr6td |d t=| |dt?| ||d^||vr |||<n|vr3tdt=| |dt?| ||dt;|t;|krP|3tdt=| |dt?| ||d|t;|d d<n|gd<|D]I}||vr |||<|vr3tdt=| |dt?| ||dJfd| D}||d<n5|r3tdt=| |dt?| ||d|D]>}|vr!|td|dt=| |S)aXFilters the given args and kwargs using a list of arguments to ignore, and a function specification. Parameters ---------- func: callable Function giving the argument specification ignore_lst: list of strings List of arguments to ignore (either a name of an argument in the function spec, or '*', or '**') *args: list Positional arguments passed to the function. **kwargs: dict Keyword arguments passed to the function Returns ------- filtered_args: list List of filtered positional and keyword arguments. z2ignore_lst must be a list of parameters to ignore z (type z ) was givenzCannot inspect object z, ignore list will not work.r^) stacklevel)***NF)r{z Argument z1 was given both as positional and as keyword for z: z was called.zWrong number of arguments for zToo many arguments for rc$i|] \}}|v || Srlrl)rmrrarg_dicts r. zfilter_args..qs)FFF$!QAX4E4EA4E4E4Er rzToo many keyword arguments for zIgnore list: argument 'z' is not defined for function )"r@rstr ValueErrortyper7ry isfunctionwarningswarn signaturedict parametersvalueskindPOSITIONAL_OR_KEYWORDr"rz KEYWORD_ONLYVAR_POSITIONAL VAR_KEYWORDdefaultempty__self____func__nextiterr enumeratervrrrpop)rC ignore_lstrrr arg_namesarg_kwonlyargs arg_varargs arg_varkwparamclass_method_sig self_name_rz arg_positionarg_name varkwargsitemrs @r. filter_argsrs* ::D*c""  @ @ @"&z"2"2 @ @ @    D ! !)'*>!!  N.w*G*GNN+D$??NNN S^^--.   #   v  !'!1HX   X % %Ng1N1NNN+D$??NNN &GFFF&,,..FFFI"   JnT7.K.K J J'dF;; J J J    8   LL    <$<<*499<<  Or cht|d}t|dkr d|ddz}|S)Nr^)indentiz%s...i)rrv)arg formatted_args r. _format_argrs@C***M =D  -"55 r ct|\}}d|D}|r+||d|}n|}t}d}|D]@}t |} |dkrd| z} t | }|| A|d|Dd|}|d|d } || fS) Ncg|]}||Srlrl)rmms r.roz$format_signature..s % % %A1 %a % % %r rdrPz %sc<g|]\}}|dt|S)r)r)rmris r.roz$format_signature..s-MMMdaqqq+a...1MMMr r())rr"r'r@rrvrxr) rCrrr}rz module_patharg_strprevious_lengthrrrs r.format_signaturers &&LFD % % % % %F  dhhv&&  ffGO&&#C(( R  "]2Mm,,}%%%% NNMMfllnnMMMNNNii  G DD'''*I  !!r MemorycFt|g|Ri|\}}dd|d|d|}|S)zdReturns a nicely formatted statement displaying the function call with the given arguments. P________________________________________________________________________________z [z ] Calling z... )r)rCrr object_namer5rmsgs r. format_callrsF't=d===f==OD)*2((KKKyy QC Jr )TT)r)__doc__ collectionsr$r7r4r:r r itertoolsrtokenizerr>loggerrfull_argspec_fields namedtuplefull_argspec_typer#rJrWrrrrrrrrrlr r.rsw ++++++H+K*=:MNN---`6/6/6/r    WWWWt111 DDD(*$$&&IIIIX"""0r