a `*jK-@sdZddlZddlZddlmZddlmZddZdd Zdd d Z Gd ddZ dddZ ddZ ddZ dddZddZdS)z=Utilities for monkey patching and wrapping object attributes.NFunctionWrapper)register_post_import_hookcCsnt|trt|tj|}|}|d}|d}dd}|||}|ddD]}|}|||}qP|||fS)a Resolves the dotted path supplied as `name` to an attribute on a target object. The `target` can be a module, class, or instance of a class. If the `target` argument is a string, it is assumed to be the name of a module, which will be imported if necessary and then used as the target object. Returns a tuple containing the parent object holding the attribute lookup resolved to, the attribute name (path prefix removed if present), and the original attribute value. .rcSsNt|r@t|D] }|t|vrt||Sqt||St||SdSN)inspectisclassgetmrovarsgetattr)parent attributeclsrI/var/www/html/assistant/venv/lib/python3.9/site-packages/wrapt/patches.pylookup_attribute-s    z&resolve_path..lookup_attributerN) isinstancestr __import__sysmodulessplit)targetnamer pathrroriginalrrr resolve_path s     rcCst|||dS)zv Convenience function for applying a patch to an attribute. This maps to the standard setattr() function. N)setattr)r r replacementrrr apply_patch@sr rc CsB|dur i}t||\}}}||g|Ri|}t||||S)a Wraps an object which is the attribute of a target object with a wrapper object created by the `factory` function. The `target` can be a module, class, or instance of a class. In the special case of `target` being a string, it is assumed to be the name of a module, with the module being imported if necessary and then used as the target object. The `name` is a string representing the dotted path to the attribute. The `factory` function should accept the original object and may accept additional positional and keyword arguments which will be set by unpacking input arguments using `*args` and `**kwargs` calling conventions. The factory function should return a new object that will replace the original object. N)rr ) rrfactoryargskwargsr rrwrapperrrr wrap_objectIs  r%c@s0eZdZdZddZddZddZdd Zd S) AttributeWrapperz\A descriptor that intercepts access to an instance attribute to apply a wrapper factory.cCs||_||_||_||_dSr)rr!r"r#)selfrr!r"r#rrr__init__lszAttributeWrapper.__init__cCs(|j|j}|j|g|jRi|jSr)__dict__rr!r"r#)r'instanceownervaluerrr__get__rs zAttributeWrapper.__get__cCs||j|j<dSrr)r)r'r*r,rrr__set__vszAttributeWrapper.__set__cCs|j|j=dSrr.)r'r*rrr __delete__yszAttributeWrapper.__delete__N)__name__ __module__ __qualname____doc__r(r-r/r0rrrrr&hs r&c CsH|dur i}|dd\}}t||d}t||||}t||||S)a| Wraps an object which is the attribute of a class instance with a wrapper object created by the `factory` function. It does this by patching the class, not the instance, with a descriptor that intercepts access to the instance attribute. The `module` can be a module, class, or instance of a class. In the special case of `module` being a string, it is assumed to be the name of a module, with the module being imported if necessary and then used as the target object. The `name` is a string representing the dotted path to the attribute. The `factory` function should accept the original object and may accept additional positional and keyword arguments which will be set by unpacking input arguments using `*args` and `**kwargs` calling conventions. The factory function should return a new object that will replace the original object. Nrr)rsplitrr&r ) modulerr!r"r#rrr r$rrrwrap_object_attribute}s r8csfdd}t|S)a Creates a decorator for wrapping a function with a `wrapper` function. The decorator which is returned may also be applied to any other callable objects such as lambda functions, methods, classmethods, and staticmethods, or objects which implement the `__call__()` method. The `wrapper` function should accept the `wrapped` function, `instance`, `args`, and `kwargs`, arguments and return the result of calling the wrapped function or some other appropriate value. csH|d}|dur}n(t|r.d|}n|t|}t||S)Nrrr r-typer)wrappedr*r"r#target_wrappedtarget_wrapperr$rr_wrappers z"function_wrapper.._wrapperrr$r?rr>rfunction_wrappers rAcslt|tr\|dr\|dd}|tjvr@ttj|tfSfdd}t||dSt|tfS)a Wraps a function which is the attribute of a target object with a `wrapper` function. The `target` can be a module, class, or instance of a class. In the special case of `target` being a string, it is assumed to be the name of a module, with the module being imported if necessary. If the `target` is a string with a trailing ``?``, the wrapping will be deferred until the module is imported. If the module is already imported, the wrapping will be applied immediately. The `name` is a string representing the dotted path to the attribute. The `wrapper` function should accept the `wrapped` function, `instance`, `args`, and `kwargs` arguments, and would return the result of calling the wrapped attribute or some other appropriate value. Returns the wrapped target function if the wrapping was applied immediately, or ``None`` if the wrapping was deferred. ?Ncst|tfdSrr%rr7rr$rrcallbacksz'wrap_function_wrapper..callbackrrendswithrrr%rr)rrr$rGrrFrwrap_function_wrappers   rJcsfdd}|S)a0 Creates a decorator which can be applied to a wrapper function, where the wrapper function will be used to wrap a function which is the attribute of a target object. The decorator returns the original wrapper function. The `target` can be a module, class, or instance of a class. In the special case of `target` being a string, it is assumed to be the name of a module, with the module being imported if necessary. If the `target` is a string with a trailing ``?``, the wrapping will be deferred until the module is imported. If the module is already imported, the wrapping will be applied immediately. The `name` is a string representing the dotted path to the attribute. The `enabled` argument can be a boolean or a callable that returns a boolean. When a callable is provided, it will be called each time the wrapper is invoked to determine if the wrapper function should be executed or whether the wrapped function should be called directly. If `enabled` is not provided, the wrapper is enabled by default. cszttrddrddd}|tjvrFttj|tfSfdd}t||SttfS)NrBrCcst|tfdSrrDrE)enabledrr$rrrGsz:patch_function_wrapper.._wrapper..callbackrH)r$_targetrGrKrrr>rr?s   z(patch_function_wrapper.._wrapperr)rrrKr?rrMrpatch_function_wrappersrNcsfdd}|S)aCreates a decorator that patches a target function with a wrapper function, but only for the duration of the call that the decorator was applied to. The `target` can be a module, class, or instance of a class. In the special case of `target` being a string, it is assumed to be the name of a module, with the module being imported if necessary. The `name` is a string representing the dotted path to the attribute. csfdd}t|S)NcsX|d}|durn(t|r.d|n|t|fdd}t||S)Nrc sVt\}}}t|}t|||z||i|Wt|||St|||0dSr)rrr)r;r*r"r#r rrr)rrr=rr_executes   zRtransient_function_wrapper.._decorator.._wrapper.._executer9)r;r*r"r#r<rO)rrr$)r=rr?s  z@transient_function_wrapper.._decorator.._wrapperrr@rrr>r _decoratorsz.transient_function_wrapper.._decoratorr)rrrQrrPrtransient_function_wrappers rR)rN)rN)N)r4rrZ __wrapt__rimporterrrr r%r&r8rArJrNrRrrrrs  4   (