File: //opt/saltstack/salt/lib/python3.10/site-packages/zmq/__pycache__/asyncio.cpython-310.pyc
o
;jv � @ s0 U d Z ddlmZ ddlZddlZddlZddlZddlmZmZ ddl m
Z
ddl Z ddl m
Z
e
� Zded< G d d
� d
e�Zd"d
d�Zd"dd�ZejdkrReZneZG dd� d�ZG dd� dee
j�ZG dd� dee
j�Zee_G dd� de j e �Z G dd� de�Z dZ d d � Z d e _ d d � Z!g d!�Z"dS )#z8AsyncIO support for zmq
Requires asyncio and Python 3.
� )� annotationsN)�Future�SelectorEventLoop)�WeakKeyDictionary)�_futurer �
_selectorsc @ s e Zd ZdZdS )� ProactorSelectorThreadWarningz�Warning class for notifying about the extra thread spawned by tornado
We automatically support proactor via tornado's AddThreadSelectorEventLoopN)�__name__�
__module__� __qualname__�__doc__� r
r
�?/opt/saltstack/salt/lib/python3.10/site-packages/zmq/asyncio.pyr s r �return�asyncio.AbstractEventLoopc s� � t v rt � S ttd�rGt� tj�rGzddlm} W n ty& td��w t j
dt dd� |� � �t � < � j �� ��fdd �}|� _ �S � S )
z�Get selector-compatible loop
Returns an object with ``add_reader`` family of methods,
either the loop itself or a SelectorThread instance.
Workaround Windows proactor removal of
*reader methods, which we need for zmq sockets.
�ProactorEventLoopr )�AddThreadSelectorEventLoopa Proactor event loop does not implement add_reader family of methods required for zmq. zmq will work with proactor if tornado >= 6.1 can be found. Use `asyncio.set_event_loop_policy(WindowsSelectorEventLoopPolicy())` or install 'tornado>=6.1' to avoid this error.a Proactor event loop does not implement add_reader family of methods required for zmq. Registering an additional selector thread for add_reader support via tornado. Use `asyncio.set_event_loop_policy(WindowsSelectorEventLoopPolicy())` to avoid this warning.� ��
stacklevelc s �� _ t�� d � �� � d S �N)�closer �popr
�� asyncio_loopZ
loop_closeZ
selector_loopr
r �_close_selector_and_loopM s z7_get_selector_windows.<locals>._close_selector_and_loop)
r �hasattr�asyncio�
isinstancer Ztornado.platform.asyncior � ImportError� RuntimeError�warnings�warn�RuntimeWarningr )r r r r
r r �_get_selector_windows s4 � ���
�r$ c C s | S )zno-op on non-Windowsr
)�loopr
r
r �_get_selector_noopZ s r& �win32c @ s$ e Zd ZeZejZejZ dd� Z
dS )�_AsyncIOc C s4 zt �� W S ty tjdtdd� Y t �� S w )NuO No running event loop. zmq.asyncio should be used from within an asyncio loop.� r )r Zget_running_loopr r! r" r# Zget_event_loop��selfr
r
r �
_default_loopj s
��z_AsyncIO._default_loopN) r r
r r Z_Future� selectors� EVENT_WRITE�_WRITE�
EVENT_READ�_READr, r
r
r
r r( e s
r( c @ s e Zd ZdZdd� Zdd� ZdS )�Pollerz1Poller returning asyncio.Future for poll results.c sL t |�}|| j@ r|�|� fdd�� || j@ r$|�|� fdd�� dS dS )z"Schedule callback for a raw socketc � � � S r r
��args��fr
r �<lambda>~ � z*Poller._watch_raw_socket.<locals>.<lambda>c r3 r r
r4 r6 r
r r8 � r9 N)�
_get_selectorr1 �
add_readerr/ Z
add_writer)r+ r% �socketZevtr7 �selectorr
r6 r �_watch_raw_socketz s
�zPoller._watch_raw_socketc G s* t |�}|D ] }|�|� |�|� qdS )z$Unschedule callback for a raw socketN)r: �
remove_readerZ
remove_writer)r+ r% Zsocketsr= r< r
r
r �_unwatch_raw_sockets� s
�zPoller._unwatch_raw_socketsN)r r
r r r> r@ r
r
r
r r2 w s r2 c @ s0 e Zd ZdZeZd dd�Zd dd�Zdd� ZdS )
�Socketz<Socket returning asyncio Futures for send/recv/poll methods.Nc C s |d u r| � � }t|�S r )Z _get_loopr: �r+ Zio_loopr
r
r r: � s zSocket._get_selectorc s � � |��� j� fdd�� dS )z#initialize the ioloop event handlerc s � � dd�S )Nr )Z_handle_eventsr
r* r
r r8 � s z'Socket._init_io_state.<locals>.<lambda>N)r: r; �_fdrB r
r* r �_init_io_state� s
�zSocket._init_io_statec C s> | j }|r|�� s| jdkr | �|��| j� dS dS dS dS )zEclear any ioloop event handler
called once at close
���N)Z
_current_loopZ is_closedrC r: r? )r+ r% r
r
r �_clear_io_state� s �zSocket._clear_io_stater ) r r
r r r2 Z
_poller_classr: rD rF r
r
r
r rA � s
rA c s0 e Zd ZdZeZdZ dd� fd d
�
Z� ZS )�Contextz/Context for creating asyncio-compatible SocketsN� r r+ �
io_threads�int | _zmq.Context�shadow�_zmq.Context | intr �Nonec s t � �||� d S r )�super�__init__)r+ rI rK �� __class__r
r rO � s zContext.__init__)rH r )r+ rG rI rJ rK rL r rM ) r r
r r rA �
_socket_classZ _instancerO �
__classcell__r
r
rP r rG � s �rG c s" e Zd ZdZd� fdd� Z� ZS )� ZMQEventLoopz}DEPRECATED: AsyncIO eventloop using zmq_poll.
pyzmq sockets should work with any asyncio event loop as of pyzmq 17.
Nc s t � t� �|�S r )� _deprecatedrN rO )r+ r= rP r
r rO � s zZMQEventLoop.__init__r )r r
r r rO rS r
r
rP r rT � s rT c C s$ t jrd S dt _tjdtdd� d S )NTzsZMQEventLoop and zmq.asyncio.install are deprecated in pyzmq 17. Special eventloop integration is no longer needed.� r )rU �calledr! r" �DeprecationWarningr
r
r
r rU � s
�rU Fc C s
t � dS )z(DEPRECATED: No longer needed in pyzmq 17N)rU r
r
r
r �install� s
rY )rG rA r2 rT rY )r r )#r �
__future__r r r- �sysr! r r �weakrefr ZzmqZ_zmqr r �__annotations__r# r r$ r& �platformr: r( Z _AsyncPollerr2 Z _AsyncSocketrA rR rG rT Z_looprU rW rY �__all__r
r
r
r �<module> s8
<