File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/zmq/__pycache__/_future.cpython-310.pyc
Back
o ;j_ � @ s� d Z ddlmZ ddlZddlmZ ddlmZ ddlm Z ddl m Z ddl m Z mZmZmZmZmZ ddlZdd lmZmZmZ G d d � d e�ZG d d � d �ZG dd� deej�ZG dd� d�Z eddd�Z G dd� deej e �Z dS )z%Future-returning APIs for coroutines.� )� annotationsN)�Future)�deque)�partial)�chain)�Any� Awaitable�Callable� NamedTuple�TypeVar�cast)�EVENTS�POLLIN�POLLOUTc @ s> e Zd ZU ded< ded< ded< ded< d ed < d ed < d S ) � _FutureEventr �future�str�kind�tuple�args�dict�kwargsr �msg�timerN)�__name__� __module__� __qualname__�__annotations__� r r �?/opt/saltstack/salt/lib/python3.10/site-packages/zmq/_future.pyr s r c @ sF e Zd ZU dZdZded< ded< ddd �Zdd d �Zddd d�ZdS )�_Asyncz Mixin for common async logicNr � _current_loopz type[Future]�_Future�returnc C sL | j du r| �� | _ | �| j � | j S | �� }|| j ur$|| _ | �|� |S )z�Get event loop Notice if event loop has changed, and register init_io_state on activation of a new event loop N)r! � _default_loop�_init_io_state)�selfZ current_loopr r r � _get_loop1 s z_Async._get_loopc C s t d��)Nz!Must be implemented in a subclass��NotImplementedError)r& r r r r$ B s z_Async._default_loop�Nonec C � d S �Nr �r& �loopr r r r% E s z_Async._init_io_state)r# r r, )r# r* ) r r r �__doc__r! r r' r$ r% r r r r r + s r c sZ e Zd ZU dZded< ded< ded< ded< ddd�Zddd�Zd d � fdd� Z� ZS ) � _AsyncPollerz:Poller that returns a Future on poll, instead of blocking.ztype[_AsyncSocket]� _socket_class�int�_READ�_WRITEz list[Any]� raw_socketsr. r �socket�evt�fr r# r* c C � t � �)z"Schedule callback for a raw socketr( )r& r. r6 r7 r8 r r r �_watch_raw_socketQ � z _AsyncPoller._watch_raw_socket�socketsc G r9 )z$Unschedule callback for a raw socketr( )r& r. r< r r r �_unwatch_raw_socketsU r; z!_AsyncPoller._unwatch_raw_sockets���� Awaitable[list[tuple[Any, int]]]c s� �� � � |dkr0zt� �d�}W n ty( } z � �|� W Y d}~� S d}~ww � �|� � S ��� ��� � �g ��fdd�}�����fdd�� g ��fdd�}�jD ]W\}}t |t j �r�t |�j �sq�j � |�}��|� |t j@ r}|jd �d � |t j@ r�|jd �d � qV��|� d}|t j@ r�|�jO }|t j@ r�|�jO }���|||� qV�� ��fd d �} ��| � �r�|� |dur�|dkr�fd d�} ��d| | ����fdd�} � �| � �fdd�} � �| � � S )z Return a Future for a poll eventr Nc � � � � s � �d � d S d S r, ��done� set_result)r ��watcherr r �wake_rawm � �z#_AsyncPoller.poll.<locals>.wake_rawc s �j � g��R � S r, )r= �r8 )r. r5 r& r r �<lambda>r s z#_AsyncPoller.poll.<locals>.<lambda>c s � D ]}|� � qd S r, )�_clear_io_state)r8 �s)�wrapped_socketsr r �_clear_wrapper_iow s �z,_AsyncPoller.poll.<locals>._clear_wrapper_io�poll�r c s� �� � rd S ��� r z��� W d S ty Y d S w ��� r)����� � d S z tt���d�}W n t yK } z ��|� W Y d }~d S d }~ww �� |� d S �Nr ) rB � cancelled�cancel� RuntimeError� exception� set_exception�superr0 rN � ExceptionrC )r8 �result�e)� __class__r r& rE r r � on_poll_ready� s& ����z(_AsyncPoller.poll.<locals>.on_poll_readyc r@ r, rA r rD r r �trigger_timeout� rG z*_AsyncPoller.poll.<locals>.trigger_timeout���MbP?c s$ t �d�r ��� d S � ��� d S )NrR )�hasattrrR Zremove_timeoutrH )r. �timeout_handler r �cancel_timeout� s z)_AsyncPoller.poll.<locals>.cancel_timeoutc s � � � s � �� d S d S r, )rB rR rH rD r r �cancel_watcher� s �z)_AsyncPoller.poll.<locals>.cancel_watcher)r" rV rN rW rU rC r' �add_done_callbackr< � isinstance�_zmq�Socketr1 � from_socket�appendr �_add_recv_eventr �_add_send_eventr3 r4 r: � call_later) r& �timeoutrX rY rF rM r6 �maskr7 r[ r\ r` ra �rZ )r r. r5 r& r_ rE rL r rN Y sb �� � � z_AsyncPoller.poll) r. r r6 r r7 r2 r8 r r# r* )r. r r< r r# r* )r> )r# r? ) r r r r/ r r: r= rN � __classcell__r r rm r r0 I s r0 c @ s e Zd Zedd� �ZdS )�_NoTimerc C r+ r, r r r r r rR � s z_NoTimer.cancelN)r r r � staticmethodrR r r r r ro � s ro �T� _AsyncSocket)�boundc s| e Zd ZU dZdZdZded< eZdZ d]d^� fd d � Z e d_d`dd��Z d_da� fdd� Z ejj je _� fdd�Zejjje_ dbdcd"d#�Z dbddd%d&�Zddd'�ded*d+�Z dbdfd.d/�Z dbdgd2d3�Zd4d5� Zdejfdhd6d7�Zd8d9� Zd:d;� Ze d_d<d=��Z dddd>�didGdH�Z djdIdJ�Z dKdL� Z dMdN� Z!dkdOdP�Z"d_dQdR�Z#dSdT� Z$dUdV� Z%dWdX� Z&d_dYdZ�Z'd[d\� Z(� Z)S )lrr Nr � _zmq.Socket� _shadow_sockr> � _from_socket�_zmq.Socket | Noner# r* c s� t |tj�r d |}}|d urt� j|jd� || _nt� j||fi |�� tj�| j�| _|d ur?tj | j j � d�t dd� t � | _t � | _d| _| jj| _d S )N)�shadowz^(io_loop) argument is deprecated in pyzmq 22.2. The currently active loop will always be used.� )� stacklevelr )rc rd re rV �__init__Z underlyingru rx �warnings�warnrZ r �DeprecationWarningr � _recv_futures� _send_futures�_stateZFD�_fd)r& �contextZ socket_type�io_looprv r rm r r r{ � s" �z_AsyncSocket.__init__�cls�type[T]r6 r� r rq c C s | ||d�S )z.Create an async socket from an existing Socket)rv r� r )r� r6 r� r r r rf � s z_AsyncSocket.from_socket�linger� int | Nonec sz | j s4| jd ur4tt| jpg | jpg ��}|D ]}|j�� s/z|j�� W q t y. Y qw q| � � t � j |d� d S )N)r� ) �closedr� �listr r r� r rB rR rS rJ rV �close)r& r� � event_list�eventrm r r r� � s � ��z_AsyncSocket.closec s" t � �|�}|tkr| �|� |S r, )rV �getr �_schedule_remaining_events)r& �keyrX rm r r r� s z_AsyncSocket.getTF�flagsr2 �copy�bool�track�)Awaitable[list[bytes] | list[_zmq.Frame]]c C � | j dt|||d�d�S )zvReceive a complete multipart zmq message. Returns a Future whose result will be a multipart message. �recv_multipart�r� r� r� �r �rh r �r& r� r� r� r r r r� s �z_AsyncSocket.recv_multipart� Awaitable[bytes | _zmq.Frame]c C r� )z�Receive a single zmq frame. Returns a Future, whose result will be the received frame. Recommend using recv_multipart instead. �recvr� r� r� r� r r r r� s �z_AsyncSocket.recv��nbytesr� r� �Awaitable[int]c C s | j d|ft||d�d�S )z�Receive a single zmq frame into a pre-allocated buffer. Returns a Future, whose result will be the number of bytes received. � recv_intor� )r r r� )r& �bufr� r� r r r r� & s �z_AsyncSocket.recv_into� msg_parts�%Awaitable[_zmq.MessageTracker | None]c K s( ||d< ||d< ||d< | j d||d�S )zqSend a complete multipart zmq message. Returns a Future that resolves when sending is complete. r� r� r� �send_multipart�r r )ri )r& r� r� r� r� r r r r r� 1 s z_AsyncSocket.send_multipart�datar c K s< ||d< ||d< ||d< |� t|||d�� | jd||d�S )z�Send a single zmq frame. Returns a Future that resolves when sending is complete. Recommend using send_multipart instead. r� r� r� r� �sendr� )�updater ri )r& r� r� r� r� r r r r r� = s z_AsyncSocket.sendc s> | � � � � ��fdd�}��|� � �fdd�}� �|� � S )zDeserialize with Futuresc s� � � � r ��� s��� du rt�d� � d�� d�t� dS ��� r*� ���� � dS ��� }z�|�}W n tyL } z � �|� W Y d}~dS d}~ww � � |� dS )z+Chain result through serialization to recvdNzFuture z completed while awaiting z . A message has been dropped!) rB rQ rT r| r} �RuntimeWarningrU rX rW rC )�_r� ZloadedrY �r8 �load�recvdr r �_chainU s" � ��z)_AsyncSocket._deserialize.<locals>._chainc s$ �� � rdS � �� r��� dS dS )z"Chain cancellation from f to recvdN)rB rQ rR )r� )r8 r� r r � _chain_cancelp s �z0_AsyncSocket._deserialize.<locals>._chain_cancel)r" rb )r& r� r� r� r� r r� r � _deserializeQ s z_AsyncSocket._deserializec s� �j r t�tj����� }|��|� tt|�|����� � � � ��fdd�}�� � r0|�� n�� |� �fdd�}� � |� � S )zSpoll the socket for events returns a Future for the poll results. c st � � � rd S ��� r z� �� W d S ty Y d S w | �� r)� ���� � d S t��� �}� �|� �d�� d S rP ) rB rQ rR rS rT rU r rX rC r� )r8 Zevts�r � poll_futurer& r r � unwrap_result� s �� z(_AsyncSocket.poll.<locals>.unwrap_resultc s0 � � � sz� �� W dS ty Y dS w dS )z4Cancel underlying poll if request has been cancelledN)rB rR rS rO )r� r r � cancel_poll� s ��z&_AsyncSocket.poll.<locals>.cancel_poll) r� rd ZZMQError�ENOTSUP� _poller_class�registerr r rN r" rB rb )r& rk r� �pr� r� r r� r rN { s z_AsyncSocket.pollc s � fdd�}| � ||�S )z'Add a timeout for a send or recv Futurec s � � � rd S � �t�� � d S r, )rB rU rd �Againr rO r r �future_timeout� s z1_AsyncSocket._add_timeout.<locals>.future_timeout)� _call_later)r& r rk r� r rO r � _add_timeout� s z_AsyncSocket._add_timeoutc C s | � � �||�S )z�Schedule a function to be called later Override for different IOLoop implementations Tornado and asyncio happen to both have ioloop.call_later with the same signature. )r' rj )r& �delay�callbackr r r r� � s z_AsyncSocket._call_laterc C s. |sdS z|� |� W dS ty Y dS w )z�Make sure that futures are removed from the event list when they resolve Avoids delaying cleanup until the next send/recv event, which may never come. N)�remove� ValueError)r r� r� r r r �_remove_finished_future� s �z$_AsyncSocket._remove_finished_future)r r r r r r � tuple | None�dict[str, Any] | Noner � Future | Noner c C s: |p| � � }|du r d}|du ri }|�d�rO|�dd�tj@ rOt| j|�}z ||i |��}W n tyG } z |�|� W Y d}~|S d}~ww |� |� |S t } t td�rf| jj } | dkrf| � || d �} t||||d| d�} | j�| � | j�t�t@ r�| �� | jr�| | jv r�|�t| j| j| d �� | �t� |S ) z4Add a recv event, returning the corresponding FutureNr r� r� r ZRCVTIMEOr] �r r r r �r� r� )r" � startswithr� rd �DONTWAIT�getattrru rW rU rC ro r^ Zrcvtimeor� r r rg r r � _handle_recvrb r r� � _add_io_state) r& r r r r r8 r� �rrY r � timeout_ms� _future_eventr r r rh � sJ �� � �� z _AsyncSocket._add_recv_eventc C sh |p| � � }|dv rw| jsw|�dd�}|�� }|tjB |d< t| j|�}d} z ||fi |��} W n6 tjyP } z|tj@ rD|� | � nd} W Y d} ~ n d} ~ w t yf } z |� | � W Y d} ~ n d} ~ ww |� | � | rw| j ru| � � |S t} ttd�r�| j�tj�} | dkr�| �|| d �} t||d ||| d �}| j�|� |�t| j| j|d �� | �t� |S ) z4Add a send event, returning the corresponding Future)r� r� r� r TFN�SNDTIMEOr] r r� r� )r" r� r� r� rd r� r� ru r� rU rW rC r r� ro r^ r� r� r rg rb r r� r� r )r& r r r r r8 r� Z nowait_kwargsr� Z finish_earlyr� rY r r� r� r r r ri sR ��� � �� z _AsyncSocket._add_send_eventc C s, | j �t�t@ s dS d}| jr%| j�� \}}}}}}|�� r!d}nn| js| js-| �t� |du r3dS |�� |dkrB|� d� dS |dkrK| j j }n|dkrT| j j }n|dkr]| j j }nt d|����|d tjO < z ||i |��}W n ty� } z |�| � W Y d} ~ dS d} ~ ww |� |� dS )zHandle recv eventsNrN r� r� r� zUnhandled recv event type: r� )ru r� r r r �popleftrB �_drop_io_staterR rC r� r� r� r� rd r� rW rU ) r& r8 r r r r� r r� rX rY r r r r� J s@ � ��z_AsyncSocket._handle_recvc C s | j �t�t@ s d S d }| jr%| j�� \}}}}}}|�� r!d }nn| js| js-| �t� |d u r3d S |�� |dkrB|� d � d S |dkrK| j j }n|dkrT| j j }nt d|����|d t jO < z ||fi |��}W n ty� } z |�| � W Y d } ~ d S d } ~ ww |� |� d S )NrN r� r� zUnhandled send event type: r� )ru r� r r r� r� rB r� rR rC r� r� r� rd r� rW rU ) r& r8 r r r r r r� rX rY r r r � _handle_sendu s< � ��z_AsyncSocket._handle_sendc C sH | j jrdS | j �t�}|tj@ r| �� |tj@ r | �� | � � dS )z(Dispatch IO events to _handle_recv, etc.N) ru r� r� r rd r r� r r� r� )r& �fd�eventsZ zmq_eventsr r r �_handle_events� s z_AsyncSocket._handle_eventsc C sB | j dkrdS |du r| j�t�}|| j @ r | �d| j� dS dS )zkSchedule a call to handle_events next loop iteration If there are still events to handle. r N)r� ru r� r r� r� )r& r� r r r r� � s �z'_AsyncSocket._schedule_remaining_eventsc C s* | j |kr | j |B }| _ | �| j � dS )zAdd io_state to poller.N�r� �_update_handler�r& �stater r r r� � s z_AsyncSocket._add_io_statec C s( | j |@ r | j | @ | _ | �| j � dS )z&Stop poller from watching an io_state.Nr� r� r r r r� � s z_AsyncSocket._drop_io_statec C s |r| � � | �� dS )zOUpdate IOLoop handler with state. zmq FD is always read-only. N)r' r� r� r r r r� � s z _AsyncSocket._update_handlerc C s6 |du r| � � }|�| j| j| j� | �d| j� dS )z#initialize the ioloop event handlerNr )r' Z add_handlerru r� r3 r� r- r r r r% � s z_AsyncSocket._init_io_statec C s2 | j }| j jr | j}| jdur| j�|� dS dS )zNunregister the ioloop event handler called once during close N)ru r� r� r! Zremove_handler)r& r� r r r rJ � s �z _AsyncSocket._clear_io_state)Nr> NN)rv rw r# r* r, )r� r� r6 rt r� r r# rq )r� r� r# r* )r TF)r� r2 r� r� r� r� r# r� )r� r2 r� r� r� r� r# r� )r� r2 r� r2 r# r� )r� r r� r2 r� r� r# r� ) r� r r� r2 r� r� r� r� r r r# r� )r# r� ) r r r r� r r� r r� r# r )NNN)r r )*r r r r r� r� r r0 r� r� r{ � classmethodrf r� rd re r/ r� r� r� r� r� r� r� r rN r� r� rp r� rh ri r� r� r� r� r� r� r� r% rJ rn r r rm r rr � sd � � �� � �*2 � 7=+ ) ) r/ � __future__r r| Zasyncior � collectionsr � functoolsr � itertoolsr �typingr r r r r r Zzmqrd r r r r r ZPollerr0 ro rq re rr r r r r �<module> s w
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings