B b ,c/@sdZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZmZmZdd lmZd d lmZmZd d lmZd addddZ e!e e dddZ"e ddddZ#Gdddeeee#ZdgZ$dS)zPython bindings for 0MQ.N)Lock)AnyDictGenericListOptionalTypeTypeVar)warn)WeakSet)Context) ContextOptionErrno SocketOption)ZMQError)AttributeSetterOptValT)SocketF)returncCsdadS)NT)_exitingrr0lib/python3.7/site-packages/zmq/sugar/context.py_notice_atexitsrTr )boundSTrT)rZ covariantcseZdZUdZeeefed<dZeed<e Z dZ e eed<dZ dZeed<eZeeed<dDd eedd fd d ZddddZdZedddZeedddZeeeddddZdEeeedddZeZeeeeedddZeeeeed d!d"Z edFeeeed#d$d%Z!ddfd&d' Z"e#edd(d)Z$edd*d+d,Z%edd*d-d.Z&dGe edd/d0d1Z'eeeed2d3d4Z(eedd5d6d7Z)ee*d8d9d:Z+eee*dd;dd?d@Z-eddAdBdCZ.Z/S)Hr auCreate a zmq Context A zmq Context creates sockets via its ``ctx.socket`` method. .. versionchanged:: 24 When using a Context as a context manager (``with zmq.Context()``), or deleting a context without closing it first, ``ctx.destroy()`` is called, closing any leftover sockets, instead of `ctx.term()` which requires sockets to be closed first. This prevents hangs caused by `ctx.term()` if sockets are left open, but means that unclean destruction of contexts (with sockets left open) is not safe if sockets are managed in other threads. sockoptsN _instance _instance_pidF_sockets _socket_classrzContext[Socket])self io_threadskwargsrc sBtjfd|i||ddr*d|_nd|_i|_t|_dS)Nr#shadowFT)super__init__get_shadowrr r )r"r#r$) __class__rrr'@s  zContext.__init__)rcCsRt|jsNtsN|jsNd|_trFt|dddk rFtd|td|d|dS)a)Deleting a Context without closing it destroys it and all sockets. .. versionchanged:: 24 Switch from threadsafe `term()` which hangs in the event of open sockets to less safe `destroy()` which warns about any leftover sockets and closes them. Tr NzUnclosed context ) stacklevelsource) localsr)rclosed_warn_destroy_closer getattrResourceWarningdestroy)r"rrr__del__Is zContext.__del__z zmq.ContextcCs|j}|jdd}|dkr.|jd|j}|jr8dnd}t|ddrrt|j}|dkr^dnd}|d|}nd}d |d |d t t ||d S) N _repr_cls.z closedr rsz socket<(z) at >) r*__dict__r( __module____name__r/r1lenr hexid)r"clsr5r/Z n_socketsr8socketsrrr__repr__cs  zContext.__repr__)r"rcCs|S)Nr)r"rrr __enter__sszContext.__enter__)exc_type exc_value tracebackrcCsd|_|dS)NT)r0r3)r"rFrGrHrrr__exit__vszContext.__exit__)r"memorcCs|j|jS)z'Copying a Context creates a shadow copy)r*r% underlying)r"rJrrr__copy__{szContext.__copy__)rBaddressrcCsddlm}||}||dS)zShadow an existing libzmq context address is the integer address of the libzmq context or an FFI pointer to it. .. versionadded:: 14.1 r) cast_int_addr)r%)zmq.utils.interoprN)rBrMrNrrrr%s zContext.shadow)rBctxrcCs4ddlm}ddlm}||}||}||dS)zuShadow an existing pyczmq context ctx is the FFI `zctx_t *` pointer .. versionadded:: 14.1 r)zctx)rN)r%)ZpyczmqrQrOrNrK)rBrPrQrNrKrMrrr shadow_pyczmqs    zContext.shadow_pyczmq)rBr#rc Csn|jdks |jtks |jjrh|j<|jdksH|jtksH|jjr^||d|_t|_WdQRX|jS)aReturns a global Context instance. Most single-threaded applications have a single, global Context. Use this method instead of passing around Context instances throughout your code. A common pattern for classes that depend on Contexts is to use a default argument to enable programs with multiple Contexts but not require the argument for simpler applications:: class MyClass(object): def __init__(self, context=None): self.context = context or Context.instance() .. versionchanged:: 18.1 When called in a subprocess after forking, a new global instance is created instead of inheriting a Context that won't work from the parent process. N)r#)rrosgetpidr/_instance_lock)rBr#rrrinstances   zContext.instancecstdS)aLClose or terminate the context. Context termination is performed in the following steps: - Any blocking operations currently in progress on sockets open within context shall raise :class:`zmq.ContextTerminated`. With the exception of socket.close(), any further operations on sockets open within this context shall raise :class:`zmq.ContextTerminated`. - After interrupting all blocking calls, term shall block until the following conditions are satisfied: - All sockets open within context have been closed. - For each socket within context, all messages sent on the socket have either been physically transferred to a network peer, or the socket's linger period set with the zmq.LINGER socket option has expired. For further details regarding socket linger behaviour refer to libzmq documentation for ZMQ_LINGER. This can be called to close the context by hand. If this is not called, the context will automatically be closed when it is garbage collected. N)r&term)r")r*rrrWsz Context.termcCst|j}|tj|S)N)dirr*extendr __members__)r"keysrrr__dir__s  zContext.__dir__)socketrcCs|j|dS)zBAdd a weakref to a socket for Context.destroy / reference countingN)r add)r"r]rrr _add_socketszContext._add_socketcCs t|dddk r|j|dS)z8Remove a socket for Context.destroy / reference countingr N)r1r discard)r"r]rrr _rm_socketszContext._rm_socket)lingerrcCs|jr dStt|ddpg}xT|D]L}|r$|js$|jrRtrRtd|td|d|dk rh|tj|| q$W| dS)aClose all sockets associated with this context and then terminate the context. .. warning:: destroy involves calling ``zmq_close()``, which is **NOT** threadsafe. If there are active sockets in other threads, this must not be called. Parameters ---------- linger : int, optional If specified, set LINGER on sockets prior to closing them. Nr z(Destroying context with unclosed socket )r,r-) r/listr1r0r r2 setsockoptrZLINGERcloserW)r"rbrCr8rrrr3s    zContext.destroy)r" socket_typer$rc Ksl|jrttj|j||f|}x<|jD].\}}y|||Wq,tk rXYq,Xq,W|||S)a]Create a Socket associated with this Context. Parameters ---------- socket_type : int The socket type, which can be any of the 0MQ socket types: REQ, REP, PUB, SUB, PAIR, DEALER, ROUTER, PULL, PUSH, etc. kwargs: will be passed to the __init__ method of the socket class. ) r/rrZENOTSUPr!ritemsrer_)r"rgr$r8optvaluerrrr]s     zContext.socket)rirjrcCs||j|<dS)zkset default socket options for new sockets created by this Context .. versionadded:: 13.0 N)r)r"rirjrrrre1szContext.setsockopt)rircCs |j|S)zkget default socket options for new sockets created by this Context .. versionadded:: 13.0 )r)r"rirrr getsockopt8szContext.getsockopt)namerirjrcCs>|tjkr|||S|tjkr,||j|<ntd|dS)z"set default sockopts as attributesz"No such context or socket option: N)r rZsetrrAttributeError)r"rlrirjrrr _set_attr_opt?s     zContext._set_attr_opt)rlrircCs6|tjkr||S||jkr(t|n |j|SdS)z"get default sockopts as attributesN)r rZr(rrn)r"rlrirrr _get_attr_optHs     zContext._get_attr_opt)keyrcCsZ|}ytt|}Wn"tk r8td|YnX||jkrNt|n|j|=dS)z%delete default sockopts as attributeszNo such socket option: N)upperr1rrnr)r"rqrirrr __delattr__Rs  zContext.__delattr__)r)N)r)N)0r>r= __qualname____doc__rintr__annotations__rrrUrrr)r0r rr!rrr'r4r5strrDrrErIrL __deepcopy__ classmethodr%rRrVrWrr\r_rar3r]rerrkrorprs __classcell__rr)r*rr #sD    $ #  )%ruatexitrSZ threadingrtypingrrrrrrr warningsr weakrefr Z zmq.backendr Z ContextBaseZ zmq.constantsr rrZ zmq.errorrZ attrsettrrrr]rrrregisterrr__all__rrrrs( $       ?