o jN!@sdZddlmZmZmZmZddlmZddlm Z m Z m Z m Z e ZddZdd Zd d Zd d ZddZGddde ZGdddZGdddee ZGddde ZGdddee ZdddddddZdS)aThe `with_signature` decorator: override the signature (and related introspection attributes) of a wrapped callable without mutating the wrapped function itself. Accepts a prototype callable, a prebuilt `inspect.Signature`, or a factory that derives a signature from the wrapped function at decoration time. This replaces the need for the `adapter=` argument of `wrapt.decorator`, which is planned for deprecation. ) CO_VARARGSCO_VARKEYWORDS Parameter Signature) signature)BaseObjectProxyBoundFunctionWrapperCallableObjectProxyFunctionWrappercs&ddfdd}t|S)aQRead-only property that memoizes its derivation in a `_self_*` slot. `functools.cached_property` is unusable on ObjectProxy subclasses: wrapt overrides `__dict__` with a property returning the wrapped object's dict, so cached_property's `instance.__dict__[name] = value` would mutate the wrapped function. Storing under a `_self_*` name routes through wrapt's own setattr handling and lands in the proxy's real instance dict. Using a `@property` (data descriptor) also ensures we win over any value copied into the instance dict by `FunctionWrapper.__init__`. _self_cached__cs.t|t}|tur|}|||SN)getattr_MISSING_derive__self_setattr__)selfvalueattrslotD/var/www/ai-app/venv/lib/python3.10/site-packages/wrapt/signature.pygetter%s   z_cached..getter)lstripproperty)rrrrr_cacheds rcCs.dd|jD}|jtjur|j|d<|S)NcSs"i|] }|jtjur|j|jqSr) annotationremptyname.0prrr 0s  z'_derive_annotations..return) parametersvaluesreturn_annotationrr)sigannrrr_derive_annotations/s   r+cCs tdd|jD}|pdS)Ncss4|]}|jtjtjfvr|jtjur|jVqdSr)kindrPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORDdefaultrr!rrr ;s z#_derive_defaults..)tupler&r')r)defaultsrrr_derive_defaults:sr3cCsdd|jD}|p dS)NcSs.i|]}|jtjur|jtjur|j|jqSr)r,r KEYWORD_ONLYr/rr r!rrrr$Es z&_derive_kwdefaults..)r&r')r) kwdefaultsrrr_derive_kwdefaultsDsr6cCsggg}}}d}}|jD]<}|jtjur ||jq|jtjur-||jq|jtjur7|j}q|jtj urD||jq|jtj urM|j}q|||}|r[|||rb||t |Sr) r&r'r,rr-appendr r.VAR_POSITIONALr4 VAR_KEYWORDr1)r)pos_only pos_or_kwkw_onlyvar_posvar_kwr#namesrrr_derive_varnamesMs(        r@csPeZdZdZfddZddZedZedZedZ ed Z ed Z Z S) _SignatureCodezCode-object proxy deriving argument-related co_* attrs from a Signature. Non-argument bits (co_flags for coroutine/generator, co_filename, etc.) fall through to the wrapped function's real __code__. ct|||_dSrsuper__init___self_signaturerwrappedr __class__rrrEj  z_SignatureCode.__init__cs|j}|dkrtjtjftfdd|jDS|dkr+tdd|jDS|dkr;tdd|jDS|dkrCt|S|d krnd d |jD|jj t t B@}tj vrc|t O}tj vrl|t O}|St|) N co_argcountc3s|]}|jvVqdSrr,r!kindsrrr0rsz)_SignatureCode._derive..co_posonlyargcountcs|] }|jtjuVqdSr)r,rr-r!rrrr0t  co_kwonlyargcountcsrQr)r,rr4r!rrrr0xrR co_varnamesco_flagscSsh|]}|jqSrrMr!rrr ~sz)_SignatureCode._derive..)rFrr-r.sumr&r'r@ __wrapped__rUrrr8r9AttributeError)rrr)flagsrrNrrns.   z_SignatureCode._deriverLrPrSrTrU) __name__ __module__ __qualname____doc__rErrrLrPrSrTrU __classcell__rrrIrrAcs rAc@sDeZdZdZddZeddZedZedZ edZ ed Z d S) _SignatureMixinzEShared Signature-derived introspection attrs for wrapper + surrogate.cCsT|j}|dkr t|S|dkrt|S|dkrt|S|dkr&t|jj|St|)N__annotations__ __defaults____kwdefaults____code__)rFr+r3r6rArXrdrY)rrr)rrrrsz_SignatureMixin._derivecCs|jSr)rFrrrr __signature__sz_SignatureMixin.__signature__rarbrcrdN) r[r\r]r^rrrfrrarbrcrdrrrrr`s  r`cs eZdZdZfddZZS)_SignatureFunctionSurrogateaFunction surrogate exposing Signature-derived introspection attrs. Used as __func__ of a bound wrapper so that inspect.signature -- which treats the bound wrapper as a MethodType and consults __func__ -- sees our override and strips self/cls correctly. crBrrCrGrIrrrErKz$_SignatureFunctionSurrogate.__init__)r[r\r]r^rEr_rrrIrrgsrgc@s$eZdZeddZeddZdS)_BoundSignatureFunctionWrappercCst|jj|jjSr)rgrX__func__ _self_parentrFrerrrris z'_BoundSignatureFunctionWrapper.__func__cCs|jjSr)rjrFrerrrrfsz,_BoundSignatureFunctionWrapper.__signature__N)r[r\r]rrirfrrrrrhs  rhcs,eZdZeZfddZeddZZS)_SignatureFunctionWrappercst||||_dSrrC)rrHwrapperrrIrrrEs z"_SignatureFunctionWrapper.__init__cCst|j|jSr)rgrXrFrerrrrisz"_SignatureFunctionWrapper.__func__) r[r\r]rh__bound_function_wrapper__rErrir_rrrIrrks  rkN) prototyperfactorycs\tddfD}|dkrtd|dkrtdfdd}|d ur*|S||S) aOverride the signature of a wrapped callable. Exactly one of `prototype`, `signature`, or `factory` must be supplied: - `prototype`: a callable whose signature will be used. - `signature`: a prebuilt `inspect.Signature` object. - `factory`: a callable `factory(wrapped)` invoked at decoration time that returns either a `Signature` or a prototype callable. The resulting wrapper exposes the override via `__signature__`, and derives `__annotations__`, `__defaults__`, `__kwdefaults__`, and the argument-related attributes of `__code__` from the same source. The wrapped function is not mutated. Calling behaviour is unchanged. css|]}|duVqdSrr)r"xrrrr0sz!with_signature..rzBwith_signature requires one of prototype=, signature=, or factory=rzFwith_signature accepts only one of prototype=, signature=, or factory=csRdd}dur }ndurt}n|}t|tr|nt|}t|||S)NcSs||i|Srr)rHinstanceargskwargsrrr_wrappersz4with_signature.._decorator.._wrapper)_inspect_signature isinstancerrk)rHrtresolvedproducedrornrrr _decorators  z"with_signature.._decoratorN)rW TypeError)rHrnrro specifiedrzrryrwith_signaturesr}r)r^inspectrrrrrru __wrapt__rr r r objectrrr+r3r6r@rAr`rgrhrkr}rrrrs     +