3

VS�_t#�@s�dZd
dl
Z
d
dlZd
dlmZ
d
dlmZ
d
dlmZ	
ddl
mZ
ddlm
Z

dd	lmZmZmZ
dd
lmZ
d
dlmZ
dad
d�Ze
je�

Gdd�de	e
�Zdg
ZdS)zPython bindings for 0MQ.�N)
�Lock)
�WeakSet)
�Context�
)
�	constants)
�AttributeSetter)�ENOTSUP�LINGER�
ctx_opt_names)
�Socket)
�ZMQErrorFc
Csd
adS)NT)
�_exiting�rr�/usr/lib64/python3.6/context.py�_notice_atexitsrcs�eZ
dZd
ZdZdZe�ZdZdZ	dZ
d-�f
dd�	Zdd�Zd	d
�Z
dd�Zd.d
d�
ZeZedd��
Zedd��
Zed/dd�
�
Z�f
dd�Zdd�Zdd�Zdd�Zd0dd�
Zedd ��
Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Z d+d,�Z!�Z"S)1rz[Create a zmq Context

    A zmq Context creates sockets via its ``ctx.socket`` method.
    NFrcsFtt
|�jfd
|
i
|��

|jdd�r.d|_nd|_i|_t�|_dS)N�
io_threads�shadowFT)�superr�__init__�get�_shadow�sockoptsr�_sockets)�selfr�kwargs)
�	__class__rrr&s




zContext.__init__c


Cs t�
|j
rtr|j�
d
S)zMdeleting a Context should terminate it, without trying non-threadsafe destroyN)�localsrr
�term)
rrrr�__del__/s
zContext.__del__c


Cs|S)
Nr)
rrrr�	__enter__8s
zContext.__enter__c

Os|j�
dS)
N)
r)r�argsrrrr�__exit__;s
zContext.__exit__cCs|jj
|j�
S)
z'Copying a Context creates a shadow copy)rr�
underlying)r�memorrr�__copy__>szContext.__copy__cCsd
dlm
}
||
�
}
||
d�
S)z�Shadow 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.interopr%)�cls�addressr%rrrrDs	

zContext.shadowcCs4d
dlm
}
d
dlm}
|j|
�
}||�
}||d�
S)z�Shadow an existing pyczmq context
        
        ctx is the FFI `zctx_t *` pointer
        
        .. versionadded:: 14.1
        r
)
�zctx)
r%)
r)Zpyczmqr)r&r%r")r'Zctxr)r%r"r(rrr�
shadow_pyczmqQs




zContext.shadow_pyczmqc
Csn|jd
ks |j
tj�ks |jjrh|j�<
|jd
ksH|j
tj�ksH|jjr^||
d�
|_tj�|_
Wd
QRX|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)�	_instance�
_instance_pid�os�getpid�closed�_instance_lock)r'rrrr�instanceas







zContext.instancec

stt
|�j�S)
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.
        )rrr)
r)
rrrr�szContext.termc
Cs(t|j
�
}
xtf
D]}|
j|�

qW|
S)
N)�dirrr
�extend)r�keysZ
collectionrrr�__dir__�s





zContext.__dir__cCs|jj
|
�

dS)
N)r�add)r�socketrrr�_add_socket�s
zContext._add_socketcCs|jr|jj
|
�

dS)
N)r�discard)rr7rrr�
_rm_socket�s

zContext._rm_socketcCsZ|jr
d
S|j
}t�|_
x4|D],}|r|jr|
d
k	rB|jt|
�
|j�
qW|j�
d
S)a�
Close 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.
        N)r/rr�
setsockoptr	�closer)rZlingerZsockets�
srrr�destroy�s







zContext.destroyc


CstS)
N)
r)
rrrr�
_socket_class�szContext._socket_classcKsj|jrt
t�
�
|j||
f|�
}x<|jj�D].\}}y|j||�
Wq*t
k
rV


Yq*Xq*W|j|�

|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/rrr?r�itemsr;r8)rZsocket_typerr=�opt�valuerrrr7�s









zContext.socketcCs||j|
<d
S)zkset default socket options for new sockets created by this Context

        .. versionadded:: 13.0
        N)
r)rrArBrrrr;�szContext.setsockoptcCs
|j|
S)
zkget default socket options for new sockets created by this Context

        .. versionadded:: 13.0
        )
r)rrArrr�
getsockopt�szContext.getsockoptcCs$|
tj
kr|j||�S||j|<d
S)z"set default sockopts as attributesN)rr
�setr)r�namerArBrrr�
_set_attr_opt�s

zContext._set_attr_optcCs6|
tj
kr|j|�
S||jkr(t|
�
�
n
|j|Sd
S)z"get default sockopts as attributesN)rr
rr�AttributeError)rrErArrr�
_get_attr_opt
s






zContext._get_attr_optcCsX|
j�}
yt
t|
�}Wn tk
r6


td
|
�
�
YnX||jkrLt|
�
�
n|j|=dS)z%delete default sockopts as attributeszno such socket option: %sN)�upper�getattrrrGr)r�keyrArrr�__delattr__

s






zContext.__delattr__)
r)
N)
r)
N)#�__name__�
__module__�__qualname__�__doc__rr+rr0r,rrrrrr!r$�__deepcopy__�classmethodrr*r1rr5r8r:r>�propertyr?r7r;rCrFrHrL�
__classcell__rr)
rrrs:





		


$


r)rP�atexitr-Z	threadingr�weakrefrZzmq.backendrZContextBase�rZ	attrsettrrrr	r
r7rZ	zmq.errorrr
r�register�__all__rrrr�<module>s"