Ef/(dZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZmZmZdd lmZd d lmZmZd d lmZd addZ ejBe e ddZ"e dddZ#Gddeeee#ZdgZ$y)zPython bindings for 0MQ.N)Lock)AnyDictGenericListOptionalTypeTypeVar)warn)WeakSet)Context) ContextOptionErrno SocketOption)ZMQError)AttributeSetterOptValT)SocketFcdayNT)_exiting3/usr/lib/python3/dist-packages/zmq/sugar/context.py_notice_atexitrsHrTr )boundSTrT)r covariantc 8eZdZUdZeeefed<dZeed<e Z dZ e eed<dZ dZeed<eZeeed<d1d d d ed ed dffd Zd2dZdZd efdZd ed efdZdededed dfdZd3d eded efdZeZedeeded efdZedeeded efdZ ed1deed ed efdZ!d2fd Z"d e#efd Z$d!ed dfd"Z%d!ed dfd#Z&d3d$e ed dfd%Z'd ed&ed ed efd'Z(d(ed)ed dfd*Z)d(ed e*fd+Z+d,ed(ed)e*d dfd-Z,d,ed(ed e*fd.Z-d/ed dfd0Z.xZ/S)4r 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_classselfzContext[Socket] io_threadskwargsreturnc t|dd|i||jddrd|_nd|_i|_t |_y)Nr(shadowFTr)super__init__get_shadowr"r r%)r'r(r) __class__s rr.zContext.__init__@sD 9J9&9 ::h &DL DL   rct|jsUtsN|jsAd|_t r#t |ddt d|td||jyyyy)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) localsr0rclosed_warn_destroy_closer getattrResourceWarningdestroyr's r__del__zContext.__del__Isa ||HT[['+D $j$7C'v.#   LLN6AH|rz zmq.Contextc V|j}|jjdd}||jd|j}|j rdnd}t |ddr&t|j}|dkDrdnd}|d|}nd}d |d |d tt||d S) N _repr_cls.z closedr%rsz socket<(z) at >) r1__dict__r/ __module____name__r8r:lenr%hexid)r'clsr@r8 n_socketsrCsocketss r__repr__zContext.__repr__csnnLL$$[$7  >>*!CLL>:I"kkr 4T *DMM*I 1}"A" 71#.GG9+QwiuSD]OF81EErc|SNrr=s r __enter__zContext.__enter__ss rexc_type exc_value tracebackc2d|_|jyr)r9r<)r'rTrUrVs r__exit__zContext.__exit__vs#'  rmemocL|jj|jS)z'Copying a Context creates a shadow copy)r1r, underlying)r'rYs r__copy__zContext.__copy__{s~~$$T__55rrMaddressc0ddlm}||}||S)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_addrr,)zmq.utils.interopr`)rMr]r`s rr,zContext.shadows 4('""rctxc^ddlm}ddlm}|j |}||}||S)zuShadow an existing pyczmq context ctx is the FFI `zctx_t *` pointer .. versionadded:: 14.1 r)zctxr_ra)pyczmqrerbr`r[)rMrcrer`r[r]s r shadow_pyczmqzContext.shadow_pyczmqs- 3__S)  +'""rc|j7|jtjk7s|jjr|j 5|j7|jtjk7s|jjr'|||_tj|_ddd|jS|jS#1swY|jSxYw)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()r#r$osgetpidr8_instance_lock)rMr(s rinstancezContext.instances. MM !  BIIK/}}#### 4MM)((BIIK7}}++$':$>CM(* C% 4}}s}} 4}}s A+CC/c"t|y)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'r1s rrnz Context.terms(  rcnt|j}|jtj|SrR)dirr1extendr __members__)r'keyss r__dir__zContext.__dir__s'4>>" M--. rsocketc:|jj|y)zBAdd a weakref to a socket for Context.destroy / reference countingN)r%addr'rus r _add_socketzContext._add_sockets &!rcVt|dd|jj|yy)z8Remove a socket for Context.destroy / reference countingr%N)r:r%discardrxs r _rm_socketzContext._rm_sockets* 4T * 6 MM ! !& ) 7rlingercT|jrytt|ddxsg}|D]l}|s|jr|jrtrt d|t d|| |j tj||jn|jy)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 r4) r8listr:r9r r; setsockoptrLINGERclosern)r'r}rOrCs rr<zContext.destroys ;;  z4!@!FBG A++B1#F'#$  %LL!4!4f=   r socket_typec "|jrttj|j||fi|}|j j D]\}} |j|||j||S#t$rY9wxYw)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. ) r8rrENOTSUPr&r"itemsrry)r'rr)rCoptvalues rruzContext.sockets ;;5==) )""" + !' ----/ JC  S%(      sB B Brrc"||j|<y)zkset default socket options for new sockets created by this Context .. versionadded:: 13.0 Nr")r'rrs rrzContext.setsockopt1s # crc |j|S)zkget default socket options for new sockets created by this Context .. versionadded:: 13.0 r)r'rs r getsockoptzContext.getsockopt8s }}S!!rnamec|tjvr|j||S|tjvr||j|<yt d|)z"set default sockopts as attributesz"No such context or socket option: N)rrrsetrr"AttributeError)r'rrrs r _set_attr_optzContext._set_attr_opt?sP =,, ,88C' ' \-- -!&DMM#  #EdV!LM Mrc|tjvr|j|S||jvr t ||j|S)z"get default sockopts as attributes)rrrr/r"r)r'rrs r _get_attr_optzContext._get_attr_optHsD =,, ,88C= $--'$T**}}S))rkeyc|j} tt|}||jvr t ||j|=y#t$rt d|wxYw)z%delete default sockopts as attributeszNo such socket option: N)upperr:rr"r)r'rrs r __delattr__zContext.__delattr__Rsiiik ',,C$--'$S))MM#&  D #:3'!BC C Ds A A!)rr*NrR)0rIrH __qualname____doc__rintr__annotations__r#rrkr$rr0r9r rr&r rr.r>r@strrPrrSrXr\ __deepcopy__ classmethodr,rgrlrnrrtryr|r<rurrrrrr __classcell__)r1s@rr r #sb$38nIsVN#'M8C='G$M48$"("c""QU"0IF#F a 6q66q6L #DG #c #a # # #47 # # # # "d1g"3"q""H4c"#"$"*** !hsm!t!FQSCB8#c###$#"c"g"N#NCNNDN*#*C*G* 's 't 'rr)%ratexitri threadingrtypingrrrrrr r warningsr weakrefr zmq.backendr ContextBase zmq.constantsrrr zmq.errorr attrsettrrrrurrrregisterrr__all__rrrrs  DDD.<</   Cy! TT2z'k?GBKz'z +r