o i@s^dZdZddlmZddlZGdddZGdddeZGd d d eZd d Zd dZ dS)aiDecorators for running functions with context/sockets. .. versionadded:: 15.3 Like using Contexts and Sockets as context managers, but with decorator syntax. Context and sockets are closed at the end of the function. For example:: from zmq.decorators import context, socket @context() @socket(zmq.PUSH) def work(ctx, push): ... )contextsocketwrapsNc@s2eZdZdZd ddZddZddZd d ZdS) _DecoratorzThe mini decorator factoryNcCs ||_dSN_target)selftargetr I/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/zmq/decorators.py__init__"s z_Decorator.__init__cs,ji\fdd}|S)a The main logic of decorator Here is how those arguments works:: @out_decorator(*dec_args, *dec_kwargs) def func(*wrap_args, **wrap_kwargs): ... And in the ``wrapper``, we simply create ``self.target`` instance via ``with``:: target = self.get_target(*args, **kwargs) with target(*dec_args, **dec_kwargs) as obj: ... cs tfdd}|S)Ncsj|i|}|i.}r|vr||<nr*|vr*tdj||f}|i|WdS1s@wYdS)Nz*{}() got multiple values for argument '{}') get_target TypeErrorformat__name__)argskwargsr obj)dec_args dec_kwargsfunckw_namer r r wrapper<s      $z7_Decorator.__call__..decorator..wrapperr)rrrrrr )rr decorator;sz&_Decorator.__call__..decorator)process_decorator_args)r rrrr rr __call__%s z_Decorator.__call__cOs|jS)zWReturn the target function Allows modifying args/kwargs to be passed. r)r rrr r r rRsz_Decorator.get_targetcOsXd}t|dtr|d}nt|dkr't|dtr'|d}|dd}|||fS)zProcess args passed to the decorator. args not consumed by the decorator will be passed to the target factory (Context/Socket constructor). Nnamer) isinstancegetstrpoplenr rrrr r r rYs   z!_Decorator.process_decorator_argsr)r __module__ __qualname____doc__rrrrr r r r rs  - rcs eZdZdZfddZZS)_ContextDecoratorzDecorator subclass for ContextscsttjdSr)superrzmqContext)r  __class__r r rmsz_ContextDecorator.__init__)rr'r(r)r __classcell__r r r.r r*jsr*cs0eZdZdZfddZddZddZZS)_SocketDecoratorzJDecorator subclass for sockets Gets the context from other args. cs0tj|i|\}}}|dd|_|||fS)z$Also grab context_name out of kwargs context_namer)r+rr$r2r&r.r r rws z'_SocketDecorator.process_decorator_argscOs|j|i|}|jS)z$Get context, based on call-time args) _get_contextr)r rrrr r r r}sz_SocketDecorator.get_targetcOsL|j|vr||j}t|tjr|S|D] }t|tjr |SqtjS)a Find the ``zmq.Context`` from ``args`` and ``kwargs`` at call time. First, if there is an keyword argument named ``context`` and it is a ``zmq.Context`` instance , we will take it. Second, we check all the ``args``, take the first ``zmq.Context`` instance. Finally, we will provide default Context -- ``zmq.Context.instance`` :return: a ``zmq.Context`` instance )r2r!r,r-instance)r rrctxargr r r r3s     z_SocketDecorator._get_context)rr'r(r)rrr3r0r r r.r r1qs  r1cOt|i|S)zDecorator for adding a Context to a function. Usage:: @context() def foo(ctx): ... .. versionadded:: 15.3 :param str name: the keyword argument passed to decorated function )r*rrr r r rs rcOr7)aSDecorator for adding a socket to a function. Usage:: @socket(zmq.PUSH) def foo(push): ... .. versionadded:: 15.3 :param str name: the keyword argument passed to decorated function :param str context_name: the keyword only argument to identify context object )r1r8r r r rsr) r)__all__ functoolsrr,rr*r1rrr r r r s K,