o

�N�g]�
@szdZd
dl
mZmZmZmZmZ
d
dlmZ
d
dl	m
Z
mZmZ
Gdd�d�Z
dded	ed
edeefdd
�Zdd
gZdS)z*0MQ polling related functions and classes.�)�Any�Dict�List�Optional�Tuple)
�zmq_poll)�POLLERR�POLLIN�POLLOUTc@s�eZ
dZUd
Zeeeefed<e	ed<ddd�Z
dedefd	d
�Ze
eBf
dedefdd
�Ze
eBf
dd�
Zdefdd�Zddeedeeeeffdd�ZdS)�Pollerz>A stateful poll interface that mirrors Python's built-in poll.�sockets�_map�returnNc

Csg|_i|_
dS�
N)rr
)
�self�r�B/opt/saltstack/salt/lib/python3.10/site-packages/zmq/sugar/poll.py�__init__s


zPoller.__init__�socketcCs
|
|jvSr)
r
)rrrrr�__contains__s

zPoller.__contains__�flagscCsp|r)|
|jvr|j|
}|
|f|j
|<d
St|j
�
}|j
�|
|f�

||j|
<d
S|
|jvr5|�|
�

d
S	d
S)ap.register(socket, flags=POLLIN|POLLOUT)

        Register a 0MQ socket or native fd for I/O monitoring.

        register(s,0) is equivalent to unregister(s).

        Parameters
        ----------
        socket : zmq.Socket or native socket
            A zmq.Socket or any Python object having a ``fileno()``
            method that returns a valid file descriptor.
        flags : int
            The events to watch for.  Can be POLLIN, POLLOUT or POLLIN|POLLOUT.
            If `flags=0`, socket will be unregistered.
        N)r
r�len�append�
unregister)rrr�idxrrr�registers









zPoller.registercCs|�|
|�
d
S)zCModify the flags for an already registered 0MQ socket or native fd.N)
r)rrrrrr�modify<sz
Poller.modifycCsF|j�
|
�
}|j�
|�

|j|d
�D]
\}
}|j|
d8<qd
S)z�Remove a 0MQ socket or native fd for I/O monitoring.

        Parameters
        ----------
        socket : Socket
            The socket instance to stop polling.
        N�
)r
�popr)rrrrrrrr@s


�zPoller.unregister�timeoutcCs6|
d
us|
dkrd}
n	t|
t
�rt|
�
}
t|j|
d�S)a�Poll the registered 0MQ or native fds for I/O.

        If there are currently events ready to be processed, this function will return immediately.
        Otherwise, this function will return as soon the first event is available or after timeout
        milliseconds have elapsed.

        Parameters
        ----------
        timeout : int
            The timeout in milliseconds. If None, no `timeout` (infinite). This
            is in milliseconds to be compatible with ``select.poll()``.

        Returns
        -------
        events : list of tuples
            The list of events that are ready to be processed.
            This is a list of tuples of the form ``(socket, event_mask)``, where the 0MQ Socket
            or integer fd is the first element, and the poll event mask (POLLIN, POLLOUT) is the second.
            It is common to call ``events = dict(poller.poll())``,
            which turns the list of tuples into a mapping of ``socket : event_mask``.
        Nr
���)
r)�
isinstance�float�intrr)rrrrr�pollNs





zPoller.poll)rNr)�__name__�
__module__�__qualname__�__doc__rrrr#�__annotations__rr�boolrr	r
rrrrr$rrrrrs



(rN�rlist�wlist�xlistrcCs�|d
urd}t|d�
}|dkrd}g}t
||
|�
D]#}d}||vr(|tO}||
vr0|tO}||vr8|tO}|�||f�

qt||�}ggg}}
}|D]\}}|t@r\|�|�

|t@re|
�|�

|t@rn|�|�

qO||
|fS)aXselect(rlist, wlist, xlist, timeout=None) -> (rlist, wlist, xlist)

    Return the result of poll as a lists of sockets ready for r/w/exception.

    This has the same interface as Python's built-in ``select.select()`` function.

    Parameters
    ----------
    timeout : float, int, optional
        The timeout in seconds. If None, no timeout (infinite). This is in seconds to be
        compatible with ``select.select()``.
    rlist : list of sockets/FDs
        sockets/FDs to be polled for read events
    wlist : list of sockets/FDs
        sockets/FDs to be polled for write events
    xlist : list of sockets/FDs
        sockets/FDs to be polled for error events

    Returns
    -------
    (rlist, wlist, xlist) : tuple of lists of sockets (length 3)
        Lists correspond to sockets available for read/write/error events respectively.
    Nr g@�@r
)r#�setr	r
rrr)r+r,r-rr�
srZreturn_socketsrrr�selectks4

























�

r0r)r(�typingrrrrrZzmq.backendrZ
zmq.constantsrr	r
rr"r0�__all__rrrr�<module>
s
 [9