3
.�]=
� @ sB d dl Z d dlZd dlmZ e jddd��ZG dd� de j�ZdS )� N)�EOFc # s� | j j}| j j� | j _| j|�}|d k r,|S | j jsvt� � � j| � tj� j � fdd�| j �E d H \}� � |f| j _n | j j\� }� j| � |j
� ytj � j |�E d H S tj
k
r� } z|j� | j|�S d }~X nX d S )Nc s � S )N� r )�pwr � /usr/lib/python3.6/_async.py�<lambda> s z expect_async.<locals>.<lambda>)�spawn�bufferZ string_type�new_dataZasync_pw_transport�
PatternWaiter� set_expecter�asyncioZget_event_loopZconnect_read_pipeZresume_readingZwait_for�fut� TimeoutError�
pause_reading�timeout)�expecterr Zpreviously_read�idx� transport�er )r r � expect_async s&
r c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )r
Nc C s || _ tj� | _d S )N)r r ZFuturer
)�selfr r r r r # s zPatternWaiter.set_expecterc C s$ | j j� s | j j|� | jj� d S )N)r
�doneZ
set_resultr r )r �resultr r r �found' s
zPatternWaiter.foundc C s$ | j j� s | j j|� | jj� d S )N)r
r Z
set_exceptionr r )r �excr r r �error, s
zPatternWaiter.errorc C s
|| _ d S )N)r )r r r r r �connection_made1 s z PatternWaiter.connection_madec C s� | j j}|jj|�}|j|d� | jj� r<| j|7 _d S y"| j j|�}|d k r\| j |� W n6 t
k
r� } z| j j � | j |� W Y d d }~X nX d S )N�read)
r r Z_decoder�decodeZ_logr
r r r r � ExceptionZerroredr )r �datar �s�indexr r r r �
data_received4 s
zPatternWaiter.data_receivedc C sT yd| j j_| j j� }W n, tk
rD } z| j|� W Y d d }~X n X | j|� d S )NT)r r Zflag_eof�eofr r r )r r" r r r r � eof_receivedF s
zPatternWaiter.eof_receivedc C s6 t |t�r |jtjkr | j� n|d k r2| j|� d S )N)�
isinstance�OSError�errnoZEIOr% r )r r r r r �connection_lostQ s
z PatternWaiter.connection_lost) �__name__�
__module__� __qualname__r r r r r r# r% r) r r r r r
s r
)N)r r( Zpexpectr � coroutiner ZProtocolr
r r r r �<module> s