EfldZdZddlmZddlZGddZGddeZGd d eZd Zd Z y) 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*eZdZdZddZdZdZdZy) _DecoratorzThe mini decorator factoryNc||_yN_target)selftargets 0/usr/lib/python3/dist-packages/zmq/decorators.py__init__z_Decorator.__init__"s  cJji\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: ... c8tfd}|S)Ncj|i|}|i5}r |vr||<n1r)|vr%tdjj||fz}|i|cdddS#1swYyxYw)Nz*{}() got multiple values for argument '{}') get_target TypeErrorformat__name__) argskwargsrobjdec_args dec_kwargsfunckw_namer s rwrapperz7_Decorator.__call__..decorator..wrapper<s($9&9X44 17*-w W%6'--3VDMM7-K $sf}00 1 1 1s AA++A4r)rr rrrr s` r decoratorz&_Decorator.__call__..decorator;s 4[ 1 1"Nr)process_decorator_args)r rrr!rs``` @r__call__z_Decorator.__call__%s7$)D(C(C ) #) %: *rc|jS)zWReturn the target function Allows modifying args/kwargs to be passed. r )r rrs rrz_Decorator.get_targetRs ||rcd}t|jdtr|jd}n+t |dk\rt|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)r rrrs rr"z!_Decorator.process_decorator_argsYsc  fjj(# .jj(G Y!^ 47C 81gG8Df$$rr )r __module__ __qualname____doc__rr#rr"rrrrs$+Z%rrc"eZdZdZfdZxZS)_ContextDecoratorzDecorator subclass for Contextsc@t|tjyr )superrzmqContext)r __class__s rrz_ContextDecorator.__init__ms %r)rr-r.r/r __classcell__r7s@rr2r2js)&&rr2c.eZdZdZfdZdZdZxZS)_SocketDecoratorzJDecorator subclass for sockets Gets the context from other args. cbt||i|\}}}|jdd|_|||fS)z$Also grab context_name out of kwargs context_namer)r4r"r+r=)r rrrr7s rr"z'_SocketDecorator.process_decorator_argsws> % > O Ov"JJ~yAf$$rc>|j|i|}|jS)z$Get context, based on call-time args) _get_contextr)r rrrs rrz_SocketDecorator.get_target}s##$##T4V4~~rc|j|vr+||j}t|tjr|S|D] }t|tjs|cStjj S)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(r5r6instance)r rrctxargs rr?z_SocketDecorator._get_contextsl    &**+C#s{{+  C#s{{+  {{##%%r)rr-r.r/r"rr?r8r9s@rr;r;qs %  &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 )r2rrs 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;rEs rrrs   t .v ..r) r/__all__ functoolsrr5rr2r;rrr0rrrIsI(   H%H%V& &)&z)&X 0 /r