o ¯µjÒIã@sÒdZddlmZddlmZmZddlmZmZm Z m Z ddl m Z Gdd„de ƒZ Gd d „d e ƒZGd d „d eƒZGd d„de ƒZGdd„dƒZGdd„deƒZeZddde dœdd„ZGdd„deƒZeZdS)zoThis module implements decorators for implementing other decorators as well as some commonly used decorators. é)Úpartial)ÚisclassÚ signatureé)ÚBaseObjectProxyÚBoundFunctionWrapperÚCallableObjectProxyÚFunctionWrapper)Ú formatargspeccsXeZdZ‡fdd„Zedd„ƒZedd„ƒZedd„ƒZed d „ƒZed d „ƒZ ‡Z S) Ú_AdapterFunctionCodecótt|ƒ |¡||_dS©N)Úsuperr Ú__init__Ú_self_adapter_code)ÚselfÚ wrapped_codeÚ adapter_code©Ú __class__©úE/var/www/ai-app/venv/lib/python3.10/site-packages/wrapt/decorators.pyró z_AdapterFunctionCode.__init__cCó|jjSr )rÚ co_argcount©rrrrr óz _AdapterFunctionCode.co_argcountcCrr )rÚco_coderrrrr$rz_AdapterFunctionCode.co_codecCrr )rÚco_flagsrrrrr(rz_AdapterFunctionCode.co_flagscCrr )rÚco_kwonlyargcountrrrrr,rz&_AdapterFunctionCode.co_kwonlyargcountcCrr )rÚ co_varnamesrrrrr 0rz _AdapterFunctionCode.co_varnames) Ú__name__Ú __module__Ú __qualname__rÚpropertyrrrrr Ú __classcell__rrrrr s     r csLeZdZ‡fdd„Zedd„ƒZedd„ƒZedd„ƒZed d „ƒZ‡Z S) Ú_AdapterFunctionSurrogatecr r )rr&rÚ _self_adapter)rÚwrappedÚadapterrrrr7rz"_AdapterFunctionSurrogate.__init__cCst|jj|jjƒSr )r Ú __wrapped__Ú__code__r'rrrrr+>ó ÿz"_AdapterFunctionSurrogate.__code__cCrr )r'Ú __defaults__rrrrr-Drz&_AdapterFunctionSurrogate.__defaults__cCrr )r'Ú__kwdefaults__rrrrr.Hrz(_AdapterFunctionSurrogate.__kwdefaults__cCs t|jƒSr )rr'rrrrÚ __signature__Ls z'_AdapterFunctionSurrogate.__signature__) r!r"r#rr$r+r-r.r/r%rrrrr&5s    r&c@s$eZdZedd„ƒZedd„ƒZdS)Ú_BoundAdapterFunctionWrappercCst|jj|jjƒSr )r&r*Ú__func__Ú _self_parentr'rrrrr1Sr,z%_BoundAdapterFunctionWrapper.__func__cCs t|jjƒSr )rr2r'rrrrr/Ys z*_BoundAdapterFunctionWrapper.__signature__N)r!r"r#r$r1r/rrrrr0Qs  r0csPeZdZeZ‡fdd„Zedd„ƒZedd„ƒZedd„ƒZ ed d „ƒZ ‡Z S) Ú_AdapterFunctionWrappercs8| d¡}tt|ƒj|i|¤Žt|j|ƒ|_||_dS)Nr))Úpoprr3rr&r*Ú_self_surrogater')rÚargsÚkwargsr)rrrrbs  z _AdapterFunctionWrapper.__init__cCrr )r5r+rrrrr+krz _AdapterFunctionWrapper.__code__cCrr )r5r-rrrrr-orz$_AdapterFunctionWrapper.__defaults__cCrr )r5r.rrrrr.srz&_AdapterFunctionWrapper.__kwdefaults__cCrr )r5r/rrrrr/wrz%_AdapterFunctionWrapper.__signature__) r!r"r#r0Ú__bound_function_wrapper__rr$r+r-r.r/r%rrrrr3^s    r3c@seZdZdd„ZdS)ÚAdapterFactorycCstƒ‚r )ÚNotImplementedError©rr(rrrÚ__call__}szAdapterFactory.__call__N)r!r"r#r<rrrrr9|s r9cs$eZdZ‡fdd„Zdd„Z‡ZS)Ú_DelegatedAdapterFactorycstt|ƒ ¡||_dSr )rr=rÚfactory)rr>rrrr‚s z!_DelegatedAdapterFactory.__init__cCs | |¡Sr )r>r;rrrr<‡s z!_DelegatedAdapterFactory.__call__)r!r"r#rr<r%rrrrr=s r=N©Úenabledr)ÚproxycsFˆdurd‡fdd„ ‰‡‡‡‡fdd„}ˆˆ|tdSttˆˆˆdS) aÓ The decorator should be supplied with a single positional argument which is the `wrapper` function to be used to implement the decorator. This may be preceded by a step whereby the keyword arguments are supplied to customise the behaviour of the decorator. The `adapter` argument is used to optionally denote a separate function which is notionally used by an adapter decorator. In that case parts of the function `__code__` and `__defaults__` attributes are used from the adapter function rather than those of the wrapped function. This allows for the argument specification from `inspect.getfullargspec()` and similar functions to be overridden with a prototype for a different function than what was wrapped. The `enabled` argument provides a way to enable/disable the use of the decorator. If the type of `enabled` is a boolean, then it is evaluated immediately and the wrapper not even applied if it is `False`. If not a boolean, it will be evaluated when the wrapper is called for an unbound wrapper, and when binding occurs for a bound wrapper. When being evaluated, if `enabled` is callable it will be called to obtain the value to be checked. If `False`, the wrapper will not be called and instead the original wrapped function will be called directly instead. The `proxy` argument provides a way of passing a custom version of the `FunctionWrapper` class used in decorating the function. Ncsœ|rGt|tƒr ||ƒ}t|ƒs?i}i}t|tƒs,t|ƒdkr(|d}|dd…}t|Ž}td|›d||ƒ|d}|r?||_t||||dSˆ|||dS)Nééÿÿÿÿz def adapterz: passr))r(Úwrapperr@r))r(rDr@) Ú isinstancer9ÚcallableÚstrÚlenr ÚexecÚ__annotations__r3)r(rDr@r)ÚnsÚ annotations)rArrÚ_build³s&    ÿzdecorator.._buildcs¢|durtˆƒr|s‡‡‡‡‡fdd„}|S|d}ˆ}t|ƒtur(|s&|Sd}|dur7tˆƒr4ˆƒ}nˆ}nt|ƒrBˆ d|¡}nˆ |t|ƒ¡}ˆ|||ˆƒS)Ncs8ˆ}t|ƒtur|s |Sd}ˆdiˆ¤Ž}ˆ|||ˆƒS)Nr)ÚtypeÚbool)Útarget_wrappedÚ_enabledÚtarget_wrapper)rMr)r@r7r(rrÚ_captureüs z-decorator.._wrapper.._capturer)rrNrOÚ__get__)r(Úinstancer6r7rSrPrQrR)rMr)r@rD)r7r(rÚ_wrapperâs" zdecorator.._wrapper)r)r?)NN)Ú decoratorr)rDr@r)rArVr)rMr)r@rArDrrW•s /0rWcs6eZdZdZddœ‡fdd„ Zdd„Zdd „Z‡ZS) Ú_StateBindingWrappera<A descriptor decorator that binds the owner instance to wrappers produced by a wrapper factory method. When applied on top of a method decorated with ``function_wrapper`` or ``decorator``, it intercepts the descriptor ``__get__`` so that when the method is accessed through an instance, the owner instance is automatically attached as an attribute on the resulting wrapper via ``__self_setattr__``. The *name* keyword argument controls the attribute name under which the owner instance is stored on the wrapper (default ``"state"``). Ústate)Únamecstƒ d¡||_dSr )rrÚ _self_name)rrZrrrr´s  z_StateBindingWrapper.__init__cCs ||_|Sr )r*)rÚwrapper_factoryrrrr<¼sz_StateBindingWrapper.__call__cs4ˆdur|S|j ˆ|¡‰|j‰‡‡‡fdd„}|S)Ncsˆ|ƒ}| ˆˆ¡|Sr )Ú__self_setattr__)ÚfuncrD©rUrZr\rrÚ_bindÓs z+_StateBindingWrapper.__get__.._bind)r*rTr[)rrUÚownerr`rr_rrTÄs  z_StateBindingWrapper.__get__)r!r"r#Ú__doc__rr<rTr%rrrrrX¦s  rXr )rbÚ functoolsrÚinspectrrÚ __wrapt__rrrr Ú argumentsr r r&r0r3r9r=Úadapter_factoryrWrXÚbind_state_to_wrapperrrrrÚs"     :