U md|7@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z dZ ede Zd d Zd d ZdddZddZddZdefddZddZddZdddZdS) z> My own variation on function-specific inspect-like features. N)islice)open)pformatzAargs varargs varkw defaults kwonlyargs kwonlydefaults annotations FullArgSpecc Csd}z|j}|j}tj|spdt|d}d}|drdt d| \}}t |}d|}|||fWSt | }|j}tt||dd}W5QRXdt|||fWSt|drt|j|d fYSt||d fYSYnXdS) al 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__)r co_filenameospathexistsjoininspectgetsourcelines startswithrematchgroupsintopen_py_sourceco_firstlinenolistrgetblockhasattrstr__hash__repr)funcZ source_filecodeZ source_codeline_noZsource_file_obj first_lineZ source_linesr"L/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/joblib/func_inspect.py get_func_codes.      r$cCsHddl}t|dr|j}nddl}|jj}dD]}||||}q.|S)z2Windows cannot encode some characters in filename.rNquote)<>!:\)urllibrr% urllib.parseparsereplace)stringr+r%charr"r"r#_clean_win_charsQs r1Tc Cs.t|dr|j}nsz!get_func_name..)rr2r getmodule TypeErrorr3r r abspath getsourcefilesplitseprrlenendswithr<r=r>r?extendismethodr@appendnamer1) r resolv_aliasZwin_charactersmodulefilenamepartsZsplittedrPklassr"r"r# get_func_name_s\         $          rVcCs d||S)z.Helper function to output a function signaturez{}{})format) function_namearg_sigr"r"r#_signature_strsrZcCs:d}t|dd}ddd|D}||||S)z)Helper function to output a function callz {0}({1}, {2})rr , css|]\}}d||fVqdS)%s=%sNr")rBkvr"r"r# sz'_function_called_str..)rritemsrW)rXargskwargsZ template_strargs_strZ kwargs_strr"r"r#_function_called_strs rdr"c Cst|}t|tr&td|t|ft|sZt|sZ|rPtj d|dd||dSt |}g}g}g}d}d} |j D]} | j | jkr|| jnL| j | jkr|| j|| jn&| j | jkr| j}n| j | jkr| j} | j| jk r|| jqt|rB|jg|}t |j} tt| j } | g|}t|dd\} }t}d }t|D]\}}|t|kr||kr||||<n td |t||t|||fn~|t|}||kr||||<nZz||||<WnHt t!fk r0}z$td t||t|||f|W5d}~XYnXqdt}t"|#D]B\}}||krd|||<n$| dk rx|||<nt$d ||fqH| dk r||d <|dk r||dd}||d<|D]2}||kr|%|ntd|t||fq|S)a Filters 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. zHignore_lst must be a list of parameters to ignore %s (type %s) was givenz4Cannot inspect object %s, ignore list will not work.r7) stacklevel)***NF)rQr zZKeyword-only parameter '%s' was passed as positional parameter for %s: %s was called.z5Wrong number of arguments for %s: %s was called.zAIgnore list for %s() contains an unexpected keyword argument '%s'rgrrfz9Ignore list: argument '%s' is not defined for function %s)&r isinstancer ValueErrortyperrN isfunctionwarningswarn signature parametersvalueskindPOSITIONAL_OR_KEYWORDrOrP KEYWORD_ONLYVAR_POSITIONAL VAR_KEYWORDdefaultempty__self____func__nextiterrVdict enumeraterKrZrd IndexErrorKeyErrorsortedr`rFpop)rZ ignore_lstrarbrY arg_namesZ arg_defaultsZarg_kwonlyargsZ arg_varargsZ arg_varkwparamZclass_method_sig self_name_rPZarg_dictZ arg_positionZarg_namepositioneZ varkwargs arg_valuevarargsitemr"r"r# filter_argss                             rcCs,t|dd}t|dkr(d|dd}|S)Nr7)indentiz%s...i)rrK)arg formatted_argr"r"r# _format_argIs  rc Ost|\}}dd|D}|r4||d|}n|}t}d}|D].}t|} |dkrbd| } t| }|| qF|dd|Dd|}d ||f} || fS) NcSsg|] }|r|qSr"r")rBmr"r"r#rDSsz$format_signature..r;rPz %scSs g|]\}}d|t|fqS)r\)r)rBr^ir"r"r#rDasr[z%s(%s))rVrOrrrrKrMr`) rrarbrRrP module_pathZarg_strZprevious_lengthrrrnr"r"r#format_signaturePs$      rMemorycCs(t|f||\}}dd|||f}|S)zi Returns a nicely formatted statement displaying the function call with the given arguments. z%s [%s] Calling %s... %sZP________________________________________________________________________________)r)rrarbZ object_namer rnmsgr"r"r# format_callhs r)TT)r)__doc__rrlrr collections itertoolsrtokenizerrloggerrZfull_argspec_fields namedtupleZfull_argspec_typer$r1rVrZrdr|rrrrr"r"r"r#s(    8 Z