h f1. @ sq d Z d d l Z d d l m Z Gd d d e j Z Gd d d Z Gd d d Z d
d Z d S) 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 e Z d Z d Z d Z d Z d Z d Z d d d d Z d d Z d
d Z
d d
Z d d Z d d Z
d d Z d d Z d d Z d d Z d d Z d d Z d d Z d d! Z d" d# Z d$ d% Z d& d' Z d S)(
async_chatzThis is an abstract class. You must derive from this class, and add
the two methods collect_incoming_data() and found_terminator()i r zlatin-1Nc C s8 d | _ g | _ t | _ t j j | | | d S)N ) ac_in_bufferincomingr
producer_fifoasyncore
dispatcher__init__)selfZsockmap r
-/opt/alt/python34/lib64/python3.4/asynchat.pyr
C s zasync_chat.__init__c C s t d d S)Nz must be implemented in subclass)NotImplementedError)r datar
r
r collect_incoming_dataQ s z async_chat.collect_incoming_datac C s | j j | d S)N)r append)r r r
r
r _collect_incoming_dataT s z!async_chat._collect_incoming_datac C s& d j | j } | j d d =| S)Nr )joinr )r dr
r
r _get_dataW s zasync_chat._get_datac C s t d d S)Nz must be implemented in subclass)r )r r
r
r found_terminator\ s zasync_chat.found_terminatorc C sd t | t r- | j r- t | | j } n* t | t rW | d k rW t d n | | _ d S)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)
isinstancestr use_encodingbytesencodingint
ValueError
terminator)r Ztermr
r
r set_terminator_ s
zasync_chat.set_terminatorc C s | j S)N)r )r r
r
r get_terminatorj s zasync_chat.get_terminatorc C sr y | j | j } WnH t k
r. d SYn3 t k
r` } z | j d SWYd d } ~ Xn Xt | t r | j r t t | j } n | j
| | _
x| j
rmt | j
} | j } | s | j
| j
d | _
q t | t rz| } | | k r4| j
| j
d | _
| j | | _ qj| j
| j
d | | j
| d | _
d | _ | j q t | } | j
j | } | d k r| d k r| j
| j
d | n | j
| | d | _
| j q t | j
| } | rQ| | k rM| j
| j
d | | j
| d | _
n Pq | j
| j
d | _
q Wd S)Nr r )Zrecvac_in_buffer_sizeBlockingIOErrorOSError handle_errorr r r r r r lenr! r r r r findfind_prefix_at_end)r r ZwhyZlbr nZterminator_lenindexr
r
r handle_readr sR
zasync_chat.handle_readc C s | j d S)N)
initiate_send)r r
r
r handle_write s zasync_chat.handle_writec C s | j d S)N)close)r r
r
r handle_close s zasync_chat.handle_closec C s t | t t t f s0 t d t | n | j } t | | k r xQ t d t | | D]$ } | j j
| | | | qd Wn | j j
| | j d S)Nz#data argument must be byte-ish (%r)r ) r r bytearray
memoryview TypeErrortypeac_out_buffer_sizer( ranger r r. )r r Zsabsir
r
r push s %zasync_chat.pushc C s | j j | | j d S)N)r r r. )r Zproducerr
r
r push_with_producer s z async_chat.push_with_producerc C s d S)z4predicate for inclusion in the readable for select()r" r
)r r
r
r readable s zasync_chat.readablec C s | j p | j S)z4predicate for inclusion in the writable for select())r connected)r r
r
r writable s zasync_chat.writablec C s | j j d d S)zAautomatically close this channel once the outgoing queue is emptyN)r r )r r
r
r close_when_done s zasync_chat.close_when_donec C sp xi| j rk| j rk| j d } | sO | j d =| d k rO | j d Sn | j } y | d | } WnD t k
r | j } | r | j j | n
| j d =w Yn Xt | t r | j r t
| | j } n y | j | } Wn t
k
r| j d SYn X| rg| t | k s@| t | k rZ| | d | j d r. rD r
r
r
r r 4 s, H (r c @ s+ e Z d Z d d d Z d d Z d S)simple_produceri c C s | | _ | | _ d S)N)r buffer_size)r r rJ r
r
r r
s zsimple_producer.__init__c C se t | j | j k rK | j d | j } | j | j d | _ | S| j } d | _ | Sd S)Nr )r( r rJ )r resultr
r
r r? s zsimple_producer.moreN)rE rF rG r
r? r
r
r
r rI s rI c @ s[ e Z d Z d d d Z d d Z d d Z d d Z d
d Z d d
Z d S)fifoNc C s( | s t | _ n t | | _ d S)N)r list)r rM r
r
r r
! s z
fifo.__init__c C s
t | j S)N)r( rM )r r
r
r __len__' s z fifo.__len__c C s | j S)N)rM )r r
r
r is_empty* s z
fifo.is_emptyc C s | j d S)Nr )rM )r r
r
r rB - s z
fifo.firstc C s | j j | d S)N)rM r )r r r
r
r r9 0 s z fifo.pushc C s$ | j r d | j j f Sd Sd S)Nr" r )r N)rM popleft)r r
r
r pop3 s zfifo.pop) rE rF rG r
rN rO rB r9 rQ r
r
r
r rL s rL c C sE t | d } x. | r@ | j | d | r@ | d 8} q W| S)Nr" )r( endswith)ZhaystackZneedlelr
r
r r* I s #r* ) rH r collectionsr r r rI rL r* r
r
r
r / s )