File: //opt/saltstack/salt/lib/python3.10/site-packages/zmq/sugar/__pycache__/context.cpython-310.pyc
o
�N�gM* � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ ddlmZmZmZ ddlmZ d d
lmZmZ d d l m Z d a ddd�Z e� e � e
ddd�Z!e
dddd�Z"G dd� deee e" �ZdgZ#dS )zPython bindings for 0MQ.� N)�Lock)�Any�Dict�Generic�List�Optional�Type�TypeVar)�WeakSet)�Context)�
ContextOption�Errno� SocketOption)�ZMQError� )�AttributeSetter�OptValT)�SocketF�returnc C s da d S )NT)�_exiting� r r �E/opt/saltstack/salt/lib/python3.10/site-packages/zmq/sugar/context.py�_notice_atexit s r �Tr )�bound�STr T)r Z covariantc s$ e Zd ZU dZeeef ed< dZeed< e � Z
dZ e e ed< dZ
eed< eZee ed< dFd
d d ed
eddf� fdd�
ZdGdd�ZdZdefdd�Zd
edefdd�Zdedededdfdd �ZdHd
ed edefd d �ZeZ e d ee d!edefd"d#��Z e d ee d$edefd%d&��Z e dFd ee d edefd'd(��Z dG� fd)d*� Z!de"e fd+d,�Z#d-eddfd.d/�Z$d-eddfd0d1�Z%dHd2e e& ddfd3d4�Z'd
ed5ed
edefd6d7�Z(d8ed9eddfd:d;�Z)d8ede*fd<d=�Z+d>ed8ed9e*ddfd?d@�Z,d>ed8ede*fdAdB�Z-dCeddfdDdE�Z.� Z/S )Ir z[Create a zmq Context
A zmq Context creates sockets via its ``ctx.socket`` method.
�sockoptsN� _instance�
_instance_pidF�_sockets�
_socket_classr �selfzContext[Socket]�
io_threads�kwargsr c sB t � jdd|i|�� |�dd�rd| _nd| _i | _t� | _d S )Nr" �shadowFTr )�super�__init__�get�_shadowr r
r )r! r" r# �� __class__r r r& 2 s zContext.__init__c C sF t � | js ts | js!tjd| � �td| d� | �� dS dS dS dS )zMdeleting a Context should terminate it, without trying non-threadsafe destroyzunclosed context � )�
stacklevel�sourceN)�localsr( r �closed�warnings�warn�ResourceWarning�term�r! r r r �__del__; s � �zContext.__del__z zmq.Contextc C s� | j }|j�dd �}|d u r|j� d|j� �}| jr dnd}t| dd �r9t| j�}|dkr/dnd}|� d|� �}nd}d |� d
|� d t t
| ��� |� d �S )
N� _repr_cls�.z closed� r r �sz socket�<�(z) at �>) r* �__dict__r' �
__module__�__name__r/ �getattr�lenr �hex�id)r! �clsr6 r/ Z n_socketsr9 �socketsr r r �__repr__L s
$zContext.__repr__c C s | S �Nr r4 r r r � __enter__\ s zContext.__enter__�exc_type� exc_value� tracebackc C s | � � d S rG )r3 )r! rI rJ rK r r r �__exit___ s zContext.__exit__�memoc C s | j �| j�S )z'Copying a Context creates a shadow copy)r* r$ �
underlying)r! rM r r r �__copy__b s zContext.__copy__rD �addressc C s ddl m} ||�}| |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.interoprR )rD rP rR r r r r$ h s
zContext.shadow�ctxc C s4 ddl m} ddlm} |�|�}||�}| |d�S )zuShadow an existing pyczmq context
ctx is the FFI `zctx_t *` pointer
.. versionadded:: 14.1
r )�zctxrQ rS )ZpyczmqrV rT rR rN )rD rU rV rR rN rP r r r �
shadow_pyczmqv s
zContext.shadow_pyczmqc C s� | j du s| jt�� ks| j jrI| j�. | j du s$| jt�� ks$| j jr8| |d�| _ t�� | _W d � | j S W d � | j S 1 sDw Y | j S )a Returns 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 �os�getpidr/ �_instance_lock)rD r" r r r �instance� s$
�
�
��
��zContext.instancec s t � �� dS )aL Close 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% r3 r4 r) r r r3 � s z Context.termc C s t | j�}|�tj� |S rG )�dirr* �extendr � __members__)r! �keysr r r �__dir__� s
zContext.__dir__�socketc C s | j �|� dS )zBAdd a weakref to a socket for Context.destroy / reference countingN)r �add�r! ra r r r � _add_socket� s zContext._add_socketc C s$ t | dd�dur| j�|� dS dS )z8Remove a socket for Context.destroy / reference countingr N)r@ r �discardrc r r r �
_rm_socket� s �zContext._rm_socket�lingerc C sV | j rdS | j}t� | _|D ]}|r$|j s$|dur |�tj|� |�� q| �� dS )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/ r r
�
setsockoptr ZLINGER�closer3 )r! rg rE r9 r r r �destroy� s
� zContext.destroy� socket_typec K sh | j rttj��| j| |fi |��}| j�� D ]\}}z|�||� W q ty, Y qw | �|� |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 r
�ENOTSUPr r �itemsrh rd )r! rk r# r9 �opt�valuer r r ra � s
�� �
zContext.socketrn ro c C s || j |< dS )zkset default socket options for new sockets created by this Context
.. versionadded:: 13.0
N�r )r! rn ro r r r rh s zContext.setsockoptc C s
| j | S )zkget default socket options for new sockets created by this Context
.. versionadded:: 13.0
rp )r! rn r r r �
getsockopt s
zContext.getsockopt�namec C s< |t jv r | �||�S |tjv r|| j|< dS td|� ���)z"set default sockopts as attributesz"No such context or socket option: N)r r^ �setr r �AttributeError)r! rr rn ro r r r �
_set_attr_opt s
zContext._set_attr_optc C s0 |t jv r
| �|�S || jvrt|��| j| S )z"get default sockopts as attributes)r r^ r' r rt )r! rr rn r r r �
_get_attr_opt( s
zContext._get_attr_opt�keyc C sR |� � }ztt|�}W n ty td|����w || jvr#t|��| j|= dS )z%delete default sockopts as attributeszNo such socket option: N)�upperr@ r rt r )r! rw rn r r r � __delattr__2 s �
zContext.__delattr__)r �r NrG )0r? r> � __qualname__�__doc__r �intr �__annotations__r r rZ r r r( r
r r r r r& r5 r6 �strrF r rH rL rO � __deepcopy__� classmethodr$ rW r[ r3 r r` rd rf �floatrj ra rh r rq ru rv ry �
__classcell__r r r) r r # sD
$
rz )$r| �atexitrX r0 � threadingr �typingr r r r r r r �weakrefr
Z zmq.backendr Z ContextBaseZ
zmq.constantsr r
r Z zmq.errorr Z attrsettrr r ra r r r �registerr r �__all__r r r r �<module> s* $