File: //opt/saltstack/salt/lib/python3.10/site-packages/zmq/__pycache__/asyncio.cpython-310.pyc
o
�N�g � @ s0 U d Z ddlZddlZddlZddlZddlmZmZ ddlmZ ddl Z
ddl m Z e� Z ee
d< G dd� de�Zd ejfd
d �Zd ejfd d
�ZejdkrReZneZG dd� d�ZG dd� dee j�ZG dd� dee j�Zee_G dd� de
je �ZG dd� de�Z dZ dd� Z de _ d d � Z g d �Z!dS ) z8AsyncIO support for zmq
Requires asyncio and Python 3.
� N)�Future�SelectorEventLoop)�WeakKeyDictionary)�_future�
_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 �returnc 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_loopI 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 s. � ���
r" c C s | S )zno-op on non-Windowsr )�loopr r r
�_get_selector_noopV s r$ �win32c @ s$ e Zd ZeZejZejZ dd� Z
dS )�_AsyncIOc C sF t jdkr zt�� W S ty tjdtdd� Y t�� S w t�� S )N)� � uO No running event loop. zmq.asyncio should be used from within an asyncio loop.� r ) �sys� version_infor Zget_running_loopr r r r! Zget_event_loop��selfr r r
�
_default_loopf s
��z_AsyncIO._default_loopN) r r r
r Z_Future� selectors� EVENT_WRITE�_WRITE�
EVENT_READ�_READr. r r r r
r& a 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 r5 r r r6 r8 r r
r: } r; N)�
_get_selectorr3 �
add_readerr1 Z
add_writer)r- r# �socketZevtr9 �selectorr r8 r
�_watch_raw_socketw 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@ rB r r r r
r4 t s r4 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
r: � s z'Socket._init_io_state.<locals>.<lambda>N)r< r= �_fdrD 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_closedrE r< rA )r- r# r r r
�_clear_io_state� s �zSocket._clear_io_stater ) r r r
r r4 Z
_poller_classr< rF rH r r r r
rC � s
rC c @ s e Zd ZdZeZdZdS )�Contextz/Context for creating asyncio-compatible SocketsN)r r r
r rC �
_socket_classZ _instancer r r r
rI � s rI 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 )� _deprecated�super�__init__)r- r? �� __class__r r
rN � s zZMQEventLoop.__init__r )r r r
r rN �
__classcell__r r rO r
rK � s rK 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' r )rL �calledr r �DeprecationWarningr r r r
rL � s
�rL Fc C s
t � dS )z(DEPRECATED: No longer needed in pyzmq 17N)rL r r r r
�install� s
rT )rI rC r4 rK rT )"r r r/ r* r r r �weakrefr ZzmqZ_zmqr r �__annotations__r! r ZAbstractEventLoopr" r$ �platformr< r& Z _AsyncPollerr4 Z _AsyncSocketrC rJ rI rK Z_looprL rR rT �__all__r r r r
�<module> s:
�9