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Ú s z expect_async..)Ú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 Ú s