3
.�]q) � @ sL d dl Z ddlmZmZ G dd� de�ZG dd� de�ZG dd � d e�ZdS )
� N� )�EOF�TIMEOUTc @ sD e Zd Zddd�Zdd� Zddd�Zdd d
�Zd d � Zdd
d�ZdS )�Expecterr c C s$ || _ || _|dkr|j}|| _d S )Nr ���)�spawn�searcher�searchwindowsize)�selfr r r � r � /usr/lib/python3.6/expect.py�__init__ s
zExpecter.__init__c C s� | j }| j}|j| }t|�}|j||| j�}|dkrz||jd � |_|d |j� |_||j|j� |_ |j
|_
||_ |S ||_d S )Nr ) r r �buffer�len�searchr �end�start�before�after�match� match_index)r
�datar r �incoming�freshlen�indexr r r �new_data
s
zExpecter.new_dataNc C s� | j }|j|_|j� |_t|_| jj}|dkr>t|_||_ |S d |_d |_ t
|�}|d| j 7 }|d k rxt
|�d | }t|��d S )Nr z
searcher: %s�
) r r r � string_typer r r � eof_indexr r �str)r
�errr r �msgr r r �eof s
z Expecter.eofc C sz | j }|j|_t|_| jj}|dkr4t|_||_|S d |_d |_t |�}|d| j 7 }|d k rnt |�d | }t|��d S )Nr z
searcher: %sr )
r r r r r r �
timeout_indexr r r )r
r r r r! r r r �timeout3 s zExpecter.timeoutc C s$ | j }|j|_d |_d |_d |_d S )N)r r r r r r )r
r r r r �erroredF s
zExpecter.erroredc C s | j }|dk rtj� | }y�|j}|j� |_xn| j|�}|dk rD|S |dk r\|dk r\| j� S |j|j|�}| j jdk r�tj | j j� |dk r.|tj� }q.W W n^ t
k
r� } z
| j |�S d}~X n: t k
r� } z
| j|�S d}~X n | j
� � Y nX dS )zBlocking expectNr )r �timer r r r$ Zread_nonblockingZmaxreadZdelayafterreadZsleepr r" r r% )r
r$ r Zend_timer �idx�er r r � expect_loopM s0
zExpecter.expect_loopr )r )N)Nr )r ) �__name__�
__module__� __qualname__r
r r" r$ r% r) r r r r r s
r c @ s* e Zd ZdZdd� Zdd� Zd dd�ZdS )
�searcher_stringa- This is a plain string search helper for the spawn.expect_any() method.
This helper class is for speed. For more powerful regex patterns
see the helper class, searcher_re.
Attributes:
eof_index - index of EOF, or -1
timeout_index - index of TIMEOUT, or -1
After a successful match by the search() method the following attributes
are available:
start - index into the buffer, first byte of match
end - index into the buffer, first byte after match
match - the matching string itself
c C s\ d| _ d| _g | _xDt|�D ]8\}}|tkr4|| _ q |tkrD|| _q | jj||f� q W dS )z�This creates an instance of searcher_string. This argument 'strings'
may be a list; a sequence of strings; or the EOF or TIMEOUT types. r Nr r )r r# �_strings� enumerater r �append)r
Zstrings�n�sr r r r
� s zsearcher_string.__init__c C s| dd� | j D �}|jd
� | jdkr:|j| jd| j f� | jdkrZ|j| jd| j f� |j� tt|� �d }dj|�S ) zUThis returns a human-readable string that represents the state of
the object.c S s g | ]}|d d| f�qS )r z %d: "%s"r )�.0�nsr r r �
<listcomp>� s z+searcher_string.__str__.<locals>.<listcomp>r �searcher_string:r z %d: EOFz %d: TIMEOUTr r )r r6 )r. r0 r r# �sort�list�zip�join)r
�ssr r r �__str__� s
zsearcher_string.__str__Nc C s� d}xb| j D ]X\}}|dkr,|t|� }n| }|j||�}|dkr |dksV||k r |}|| } }
q W |dkrtdS |
| _|| _| jt| j� | _| S )a� This searches 'buffer' for the first occurrence of one of the search
strings. 'freshlen' must indicate the number of bytes at the end of
'buffer' which have not been searched before. It helps to avoid
searching the same, possibly big, buffer over and over again.
See class spawn for the 'searchwindowsize' argument.
If there is a match this returns the index of that string, and sets
'start', 'end' and 'match'. Otherwise, this returns -1. Nr r r )r. r �findr r r ) r
r r r � first_matchr r2 �offsetr1 �
best_indexZ
best_matchr r r r � s zsearcher_string.search)N)r* r+ r, �__doc__r
r< r r r r r r- n s r- c @ s* e Zd ZdZdd� Zdd� Zd dd�ZdS )
� searcher_reaV This is regular expression string search helper for the
spawn.expect_any() method. This helper class is for powerful
pattern matching. For speed, see the helper class, searcher_string.
Attributes:
eof_index - index of EOF, or -1
timeout_index - index of TIMEOUT, or -1
After a successful match by the search() method the following attributes
are available:
start - index into the buffer, first byte of match
end - index into the buffer, first byte after match
match - the re.match object returned by a successful re.search
c C sj d| _ d| _g | _xRtttt|���|�D ]8\}}|tkrB|| _ q*|tkrR|| _q*| jj ||f� q*W dS )z�This creates an instance that searches for 'patterns' Where
'patterns' may be a list or other sequence of compiled regular
expressions, or the EOF or TIMEOUT types.r Nr r )
r r# � _searchesr9 r8 �ranger r r r0 )r
Zpatternsr1 r2 r r r r
� s zsearcher_re.__init__c C s� t � }x\| jD ]R\}}y |j|d||jf f� W q tk
r^ |j|d||jf f� Y qX qW |jd
� | jdkr�|j| jd| j f� | jdkr�|j| jd| j f� |j� t t|� �d }dj |�S ) zUThis returns a human-readable string that represents the state of
the object.z %d: re.compile("%s")z %d: re.compile(%r)r � searcher_re:r z %d: EOFz %d: TIMEOUTr r )r rE )
r8 rC r0 �pattern�UnicodeEncodeErrorr r# r7 r9 r: )r
r; r1 r2 r r r r< � s "
zsearcher_re.__str__Nc C s� d}|dkrd}nt dt|�| �}xL| jD ]B\}}|j||�}|dkrJq,|j� } |dksb| |k r,| }|}
|} q,W |dkr~dS || _|
| _| jj� | _| S )a� This searches 'buffer' for the first occurrence of one of the regular
expressions. 'freshlen' must indicate the number of bytes at the end of
'buffer' which have not been searched before.
See class spawn for the 'searchwindowsize' argument.
If there is a match this returns the index of that string, and sets
'start', 'end' and 'match'. Otherwise, returns -1.Nr r r )�maxr rC r r r r ) r
r r r r> Z searchstartr r2 r r1 Z the_matchr@ r r r r
s&
zsearcher_re.search)N)r* r+ r, rA r
r< r r r r r rB � s rB )r& �
exceptionsr r �objectr r- rB r r r r �<module> s i_