MhdZddlmZdZddlmZddlZGddZGdd eZGd d eZ d Z d Z dS)aeDecorators 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): ... ) annotations)contextsocketwrapsNc,eZdZdZddZdZdZdZdS) _DecoratorzThe mini decorator factoryNc||_dSN_target)selftargets N/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/zmq/decorators.py__init__z_Decorator.__init__$s  c@ji\fd}|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: ... cFtfd}|S)Ncj|i|}|i5}r |vr||<n&r|vrtjdd||fz}|i|cdddS#1swxYwYdS)Nz%() got multiple values for argument '') get_target TypeError__name__) argskwargsrobjdec_args dec_kwargsfunckw_namers rwrapperz7_Decorator.__call__..decorator..wrapper>s($9&99VX444 1-7#8*-w -W%6%6'#}55*1555 $sf}4000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s:AA"%A"r)rr!rrr rs` r decoratorz&_Decorator.__call__..decorator=sH 4[[ 1 1 1 1 1 1 1 1[ 1"Nr)process_decorator_args)rrrr"r s``` @r__call__z_Decorator.__call__'s_$)D(C ) #) ) %:        *rc|jS)zWReturn the target function Allows modifying args/kwargs to be passed. r )rrrs rrz_Decorator.get_targetTs |rc d}t|dtr|d}n@t |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) isinstancegetstrpoplen)rrrr s rr#z!_Decorator.process_decorator_args[s~  fjj((# . . jj((GG YY!^^ 47C 8 8^1gG8Df$$rr )r __module__ __qualname____doc__rr$rr#rrr r !s\$$+++Z%%%%%rr c"eZdZdZfdZxZS)_ContextDecoratorzDecorator subclass for Contextsc^ttjdSr )superrzmqContext)r __class__s rrz_ContextDecorator.__init__os# %%%%%r)rr.r/r0r __classcell__r8s@rr3r3ls>))&&&&&&&&&rr3c.eZdZdZfdZdZdZxZS)_SocketDecoratorzJDecorator subclass for sockets Gets the context from other args. c~tj|i|\}}}|dd|_|||fS)z$Also grab context_name out of kwargs context_namer)r5r#r,r>)rrrr r8s rr#z'_SocketDecorator.process_decorator_argsysH > > O O Ov"JJ~yAAf$$rc*|j|i|}|jS)z$Get context, based on call-time args) _get_contextr)rrrrs rrz_SocketDecorator.get_targets!#$#T4V44~rc|j|vr)||j}t|tjr|S|D] }t|tjr|cS!tjS)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 )r>r)r6r7instance)rrrctxargs rr@z_SocketDecorator._get_contexts   & &*+C#s{++    C#s{++   {##%%%r)rr.r/r0r#rr@r9r:s@rr<r<ss` %%%%%  &&&&&&&rr<c*t|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 )r3rrs rrrs      / / //rc*t|i|S)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 )r<rFs rrrs     t .v . ..r) r0 __future__r__all__ functoolsrr6r r3r<rrr1rrrKs"#"""""    H%H%H%H%H%H%H%H%V&&&&& &&&)&)&)&)&)&z)&)&)&X 0 0 0 /////r