3
.]q) @ sL d dl Z ddlmZmZ G dd deZG dd deZG dd d eZdS )
N )EOFTIMEOUTc @ sD e Zd ZdddZdd ZdddZdd d
Zd d Zdd
dZdS )Expecterr c C s$ || _ || _|dkr|j}|| _d S )Nr )spawnsearchersearchwindowsize)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 bufferlensearchr endstartbeforeaftermatch match_index)r
datar r incomingfreshlenindexr 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 rtj | 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 idxer 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 ddZdS )
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 )zThis 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
Zstringsnsr 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| fqS )r z %d: "%s"r ).0nsr r r
s z+searcher_string.__str__..r searcher_string:r z %d: EOFz %d: TIMEOUTr r )r r6 )r. r0 r r# sortlistzipjoin)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 ddZdS )
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 )zThis 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 patternUnicodeEncodeErrorr 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 s i_