$60 GRAYBYTE WORDPRESS FILE MANAGER $46

�
h f1.�@sqdZd
dl
Z
d
dlmZ
Gdd�de
j�ZGdd�d�ZGdd	�d	�Zd
d�ZdS)a�A class supporting chat-style (command/response) protocols.

This class adds support for 'chat' style protocols - where one side
sends a 'command', and the other sends a response (examples would be
the common internet protocols - smtp, nntp, ftp, etc..).

The handle_read() method looks at the input stream for the current
'terminator' (usually '\r\n' for single-line responses, '\r\n.\r\n'
for multi-line output), calling self.found_terminator() on its
receipt.

for example:
Say you build an async nntp client using this class.  At the start
of the connection, you'll have self.terminator set to '\r\n', in
order to process the single-line greeting.  Just before issuing a
'LIST' command you'll set it to '\r\n.\r\n'.  The output of the LIST
command will be accumulated (using your own 'collect_incoming_data'
method) up to the terminator, and then control will be returned to
you - by calling your self.found_terminator() method.
�N)
�dequec@s
eZ
dZd
ZdZdZdZdZdddd�Zdd	�Z	d
d�Z
dd
�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�ZdS)(�
async_chatz�This is an abstract class.  You must derive from this class, and add
    the two methods collect_incoming_data() and found_terminator()i
r
zlatin-1NcCs8d
|_g|_
t�|_tjj||
|�
dS)N�)�ac_in_buffer�incomingr�
producer_fifo�asyncore�
dispatcher�__init__)�selfZsock�map�r
�-/opt/alt/python34/lib64/python3.4/asynchat.pyr
Cs		
zasync_chat.__init__cCstd
�
�
dS)Nzmust be implemented in subclass)
�NotImplementedError)r�datar
r
r�collect_incoming_dataQs
z async_chat.collect_incoming_datacCs|jj
|
�

dS)
N)r�append)rrr
r
r�_collect_incoming_dataTs
z!async_chat._collect_incoming_datac
Cs&d
j|j
�
}
|j
dd�=|
S)Nr)�joinr)r�
dr
r
r�	_get_dataWs


zasync_chat._get_datac

Cstd
�
�
dS)Nzmust be implemented in subclass)
r)
rr
r
r�found_terminator\s
zasync_chat.found_terminatorcCsdt|
t
�r-|jr-t|
|j�}
n*t|
t�rW|
d
krWtd�
�
n|
|_dS)zdSet the input delimiter.

        Can be a fixed string of any length, an integer, or None.
        r
z-the number of received bytes must be positiveN)�
isinstance�str�use_encoding�bytes�encoding�int�
ValueError�
terminator)rZtermr
r
r�set_terminator_s




zasync_chat.set_terminatorc


Cs|jS)
N)
r)
rr
r
r�get_terminatorjs
zasync_chat.get_terminatorc
Csry|j|j
�
}
WnHtk
r.


dSYn3tk
r`
}
z|j�
dSWYdd}~Xn
Xt|
t�r�|jr�tt|j	�}
n|j
|
|_
x�
|j
rmt|j
�
}|j�}|s�|j
|j
�

d
|_
q�t|t�rz
|}||kr4
|j
|j
�

d
|_
|j||_qj|j
|j
d|��

|j
|d�|_
d|_|j�
q�t|�
}|j
j|�
}|dkr�
|dkr�
|j
|j
d|��

n|j
||d�|_
|j�
q�t|j
|�}|rQ||krM|j
|j
d|��

|j
|d�|_
nPq�|j
|j
�

d
|_
q�WdS)Nrr
�
���)Zrecv�ac_in_buffer_size�BlockingIOError�OSError�handle_errorrrrrrr�lenr!rrrr�find�find_prefix_at_end)rrZwhyZlbr�
nZterminator_len�indexr
r
r�handle_readrsR

















	


	

	








zasync_chat.handle_readc


Cs|j�
dS)
N)
�
initiate_send)
rr
r
r�handle_write�s
zasync_chat.handle_writec


Cs|j�
dS)
N)
�close)
rr
r
r�handle_close�s
zasync_chat.handle_closecCs�t|
t
ttf�s0td
t|
�
��
n|j}t|
�
|kr�xQtdt|
�
|�D]$}|j	j
|
|||��

qdWn|j	j
|
�

|j�
dS)Nz#data argument must be byte-ish (%r)r
)rr�	bytearray�
memoryview�	TypeError�type�ac_out_buffer_sizer(�rangerrr.)rrZsabs�
ir
r
r�push�s



	


%
zasync_chat.pushcCs|jj
|
�

|j�
dS)
N)rrr.)rZproducerr
r
r�push_with_producer�s

zasync_chat.push_with_producerc


Csd
S)z4predicate for inclusion in the readable for select()r"r
)
rr
r
r�readable�szasync_chat.readablec


Cs|jp|j
S)
z4predicate for inclusion in the writable for select())r�	connected)
rr
r
r�writable�szasync_chat.writablec

Cs|jj
d
�

d
S)zAautomatically close this channel once the outgoing queue is emptyN)rr)
rr
r
r�close_when_done�szasync_chat.close_when_donec
Csp
xi
|jrk
|j
rk
|jd
}
|
sO|jd
=|
dkrO|j�
dSn|j}y|
d|�}WnDtk
r�


|
j�}|r�|jj|�

n
|jd
=wYn
Xt|t�r�|j	r�t
||j�}ny|j|�
}Wn t
k
r



|j�
dSYn
X|rg
|t|�
ks@
|t|
�
krZ
|
|d�|jd
<qg
|jd
=ndSWdS)Nr
)rr<r1r6r4�more�
appendleftrrrrr�sendr&r'r()r�firstZobsrZnum_sentr
r
rr.�s8








	















	
$

zasync_chat.initiate_sendc

Cs*d
|_|j
dd�=|jj�
dS)Nr)rrr�clear)
rr
r
r�discard_buffers
s	

zasync_chat.discard_buffers)�__name__�
__module__�__qualname__�__doc__r$r6rrr
rrrrr r!r-r/r1r9r:r;r=r>r.rDr
r
r
rr4s,

H(rc@s+eZ
dZd
dd�
Zdd�ZdS)�simple_producericCs|
|_||_
dS)
N)r�buffer_size)rrrJr
r
rr

s
	
zsimple_producer.__init__c
Cset|j
�
|jkrK|j
d|j�}
|j
|jd�|_
|
S|j
}
d
|_
|
SdS)Nr)r(rrJ)r�resultr
r
rr?
s



	
	
zsimple_producer.moreN)rErFrGr
r?r
r
r
rrI
srIc@s[eZ
dZd
dd�
Zdd�Zdd�Zdd	�Zd
d�Zdd
�Zd
S)�fifoNcCs(|
st�|_
nt|
�
|_
dS)
N)r�list)rrMr
r
rr
!
s

z
fifo.__init__c

Cs
t|j
�
S)
N)r(rM)
rr
r
r�__len__'
s
zfifo.__len__c


Cs|jS)
N)
rM)
rr
r
r�is_empty*
s
z
fifo.is_emptyc

Cs|jd
S)Nr
)
rM)
rr
r
rrB-
s
z
fifo.firstcCs|jj
|
�

dS)
N)rMr)rrr
r
rr90
s
z	fifo.pushc

Cs$|jrd
|jj
�fSdSdS)Nr"r
)r
N)rM�popleft)
rr
r
r�pop3
s
	
zfifo.pop)	rErFrGr
rNrOrBr9rQr
r
r
rrL 
s
rLcCsEt|
�
d
}x.|r@|j
|
d|��
r@|d
8}qW|S)Nr")r(�endswith)ZhaystackZneedle�
lr
r
rr*I
s

#

r*)	rHr�collectionsrr	rrIrLr*r
r
r
r�<module>/s

�)