o ,wi2 @s$dZddlZddlZddlZddlmZmZmZmZm Z m Z m Z m Z m Z mZddlZddlZeZGdddZeZeZeZdejfddZdejfd d Zdefd d Zd e ededeffddZdeeeffddZdddededeeeffddZ ej!GdddZ"dS)z;Module for getting (and caching) inspect.Signature objects.N) AnyCallableDictGenericListMappingOptionalTupleTypeUnionc@s(eZdZdZddZddZddZdS) NoValuez+Sentinel class for arguments with no value.cCsdS)Nz fdl.NO_VALUEselfr r S/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/fiddle/_src/signatures.py__repr__"szNoValue.__repr__cCs~|S)z>Override for deepcopy that does not copy this sentinel object.r )rmemor r r __deepcopy__%szNoValue.__deepcopy__cCs|S)zCOverride for `copy.copy()` that does not copy this sentinel object.r rr r r__copy__*szNoValue.__copy__N)__name__ __module__ __qualname____doc__rrrr r r rr s  r returnc Cst|}|dur |n|}tjdkrt|trt|tr|j}zt |WSt yJt|trIt |drIz t |j WYSt yHYww)zReturns the signature for a function or class. This mostly calls inspect.signature, but handles generics and some builtin types better. Args: fn_or_cls: Function or class callable. N) __call__)typing_extensions get_originsys version_info isinstancetype issubclassr__init__inspect signature ValueErrorhasattrr) fn_or_clsoriginr r r_get_signature_uncached5s"     r+cCsJzt|WStyt|YSty$t|}|t|<|YSw)aReturns (and maybe caches) the signature for `fn_or_cls`. If a signature is not available for `fn_or_cls` and `fn_or_cls` is a type, this function will instead try to get a signature for `fn_or_cls.__call__`. This behavior allows a reasonable signature to be inferred for certain builtin types such as `dict`. Args: fn_or_cls: The function or class to return a signature for. )_signature_cache TypeErrorr+KeyError)r)r&r r r get_signatureSs    r/c Cs(zt|WdSttfyYdSw)a4Returns whether `fn_or_cls` has an available signature. Note that this will return `True` even if a signature for `fn_or_cls` is not yet in the cache. The signature for `fn_or_cls` will be cached if it is available. Args: fn_or_cls: The function or class to check signature availability for. FT)r/r'r-)r)r r r has_signaturehs  r0cls.cCs@t|D]}d|jvr|jSd|jvr|jSqtd)zEFind the first ``__init__`` or ``__new__`` method in the class's MRO.r$__new__z#Could not find a class constructor.)r%getmro__dict__r$r2 RuntimeError)r1baser r r_find_class_construction_fnys    r7c CsHzt|trt|st|}tj||dWSttfy#iYSw)zFReturns a dictionary corresponding to the annotations for `fn_or_cls`.include_extras) r!r" dataclasses is_dataclassr7rget_type_hintsr- NameError)r)r9r r r_get_type_hints_uncachedsr>Fr8r)r9cCsZzt||fWStyt||dYSty,t||d}|t||f<|YSw)zJReturns (and maybe caches) type hints for `fn_or_cls`, suppressing errors.r8)_type_hints_cacher-r>r.)r)r9hintsr r rr<s    r<c @sXeZdZUdZejed<dZee ed<ddZ e de e eefeffdd Zd e eefd edefd d Z  d$de eefde de deeee eefffddZd%ddZde eefdefddZdede eefde eeffddZedeeejffddZedeefddZedeefd d!Zedeefd"d#Z dS)& SignatureInfoz:To store signature related information about the callable.r&Nhas_var_keywordcCsTd|_t|jjD]\}}|j|jkr||_q |j|jkr'|jdur'd|_q dS)NT) _var_positional_start enumerater& parametersvalueskindVAR_POSITIONAL VAR_KEYWORDrB)rindexparamr r r __post_init__s zSignatureInfo.__post_init__rc Ost|}z |j|i|j}Wnty&}z td|d||d}~wwtt|D]C\}}|j|}|j|j krH| |j } | ||<|j|j krc| |j } t| D] \} } | ||| <qX|j|j krr|| |j q/|S)axBind arguments and return an args dict in canonical storage format. Args: fn_or_cls: A function or a class object. *args: A list of positional arguments. **kwargs: A dict of keyword arguments. Returns: An arguments dict in canonical storage format. Please see `Buildable` docstring for the definition of canonical storage format. zCannot bind arguments to z: N)r/ bind_partial argumentsr-rDlistkeysrErGPOSITIONAL_ONLYpopnamerHrIupdate) r)argskwargsr&rNerJrSrKvaluerFir r rsignature_bindings(      zSignatureInfo.signature_bindingargumentmissingcCsz|}d}t|tr||jvr|j|}nt|tsJ|jdur0||jkr0t|j}||}|r;|j|jur;|j}|S)aoGet default value for the argument, return missing if not found. Args: argument: Specify the argument either by its name or the index for positional arguments. missing: A value to return if the argument does not have defaults. Returns: The default value for the argument if exists, return the missing object otherwise. N) r!strrEintvar_positional_startrOrFdefaultempty)rr[r\rXrKparamsr r r get_defaults      zSignatureInfo.get_defaultFrNinclude_pos_or_kw_in_argsinclude_no_valuecCs|}t|j}g}t|D]N\}}|j|jkr5||vr*|||||=n |r5|||t |j|j kr_|sB|j |vr_|j |vrT|||j ||j =q|r_|||t q|j dur~|j }||vr~|||||=|d7}||vsl||fS)a(Transform arguments in dict to *args and **kwargs. Args: arguments: An arguments dict in canonical storage format (please see `Buildable` docstring for the precise definition). include_pos_or_kw_in_args: Whether to always put positional-or-keyword args in the *args list as output. If set to False, such args will be added to **kwargs output dict if the variadic positional args list is empty in `arguments`. include_no_value: Whether to include `NO_VALUE` in the output. If set as True, a value is set to `NO_VALUE` if no user specific value exists and its default value is empty. Returns: A positional args list that can be used to build the Buildable. A keyword args dict that can be used to build the Buildable. N) copyrOrErFrDrGrQappendrcNO_VALUEPOSITIONAL_OR_KEYWORDr_rS)rrNrdrerEpositional_valuesrJrKr r rtransform_to_args_kwargss4     z&SignatureInfo.transform_to_args_kwargscCs|jj|}|dur3|j|jkrtd|d||j|jkr+td|d||j|jkr3d}|dur^|js`||jjvrJd|d}t|d|d|d d |j d }t|dSdS) z:Raises an error if ``name`` is not a valid parameter name.Nz(Cannot access POSITIONAL_ONLY parameter z on z'Cannot access VAR_POSITIONAL parameter zVariadic arguments (e.g. *z) are not supported.zNo parameter named 'z ' exists for z; valid parameter names: z, .) r&rEgetrGrQAttributeErrorrHrIrBjoinvalid_param_names)rrSr)rKerr_msgr r rvalidate_param_name%s0      z!SignatureInfo.validate_param_namekeycs\fdd}t|trt||j||j|j}n||}t|ttfs,Jd|d|S)z.Replace VARARGS handle in index key if exists.cs|turjS|SN)VARARGSr_)xrr rDsz6SignatureInfo.replace_varargs_handle..z!Key must be an int or slice, got rm)r!slicestartstopstepr^)rrt replace_fnr rrreplace_varargs_handleBs   z$SignatureInfo.replace_varargs_handlerJcCs`|dkr||dd\}}|t|7}t|jj}|t|kr.||}|j|jkr.|jS|S)zAMap index of positional arguments to the key for `__arguments__`.rT) rllenrOr&rErFrGrjrS)rrJrNrU_rbrKr r r index_to_keyNs   zSignatureInfo.index_to_keycCs|jjSru)r&rErr r rrE\szSignatureInfo.parameterscCstdd|jjDS)Ncss*|]\}}|j|j|jfvr|VqdSru)rGrj KEYWORD_ONLY).0rSrKr r r bsz2SignatureInfo.valid_param_names..)tupler&rEitemsrr r rrq`s zSignatureInfo.valid_param_namescCs,|jjD] }|j|jkr|jSqdSru)r&rErFrGrIrS)rrKr r rvar_keyword_namehs   zSignatureInfo.var_keyword_namecCs|jSru)rCrr r rr_osz"SignatureInfo.var_positional_start)FF)rN)!rrrrr% Signature__annotations__rBrboolrL staticmethodrr r^r]rrZrcr rrlrsryr~rpropertyr ParameterrErqrr_r r r rrAsL   !   6    rA)#rr:r%rtypingrrrrrrrr r r weakrefrobjectrvr riWeakKeyDictionaryr,r?rr+r/rr0r7r]r>r< dataclassrAr r r rs:0