3
\@, @ s@ d dl Z d dlmZ G dd de jZG dd dZdd ZdS ) N)dequec @ s e Zd ZdZdZdZdZd'ddZdd Zd d
Z d d Z
d
d Z dd Z dd Z
dd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& ZdS )(
async_chati r zlatin-1Nc C s( d| _ g | _t | _tjj| || d S )N ) ac_in_bufferincomingr
producer_fifoasyncore
dispatcher__init__)selfZsockmap r
/usr/lib64/python3.6/asynchat.pyr
C s zasync_chat.__init__c C s t dd 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 dj | j}| jd d = |S )Nr )joinr )r dr
r
r _get_dataW s zasync_chat._get_datac C s t dd S )Nz must be implemented in subclass)r )r r
r
r found_terminator\ s zasync_chat.found_terminatorc C sB t |tr | jr t|| j}nt |tr8|dk r8td|| _d S )Nr z-the number of received bytes must be positive)
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 s y| j | j}W n: tk
r$ d S tk
rJ } z | j d S d }~X nX t|trh| jrhtt| j }| j
| | _
xV| j
rt | j
}| j }|s| j
| j
d| _
qxt|tr|}||k r| j
| j
d| _
| j| | _n2| j
| j
d | | j
|d | _
d| _| j qxt |}| j
j|}|dkrp|dkrR| j
| j
d | | j
|| d | _
| j qxt| j
|}|r||kr| j
| j
d | | j
| d | _
P qx| j
| j
d| _
qxW d 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 |tttfs tdt|| j}t||krdx>tdt||D ] }| j j
|||| qBW n | 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 dS )Nr" r
)r r
r
r readable s zasync_chat.readablec C s | j p | j S )N)r connected)r r
r
r writable s zasync_chat.writablec C s | j jd d S )N)r r )r r
r
r close_when_done s zasync_chat.close_when_donec C s x| j o| jr| j d }|s<| j d= |d kr<| j d S | j}y|d | }W n8 tk
r |j }|r|| j j| n| j d= wY nX t|tr| j rt
|| j }y| j |}W n t
k
r | j d S X |r|t|k s|t|k r||d | j d< n| j d= d S W d S )Nr )r r< r1 r6 r4 more
appendleftr r r r r sendr& r' r( )r firstZobsr Znum_sentr
r
r r. s8
zasync_chat.initiate_sendc C s d| _ | jd d = | jj d S )Nr )r r r clear)r r
r
r discard_buffers s zasync_chat.discard_buffers)NN)__name__
__module__ __qualname__r$ r6 r r r
r r r r r r! r- r/ r1 r9 r: r; r= r> r. rD r
r
r
r r 4 s*
H (r c @ s e Zd ZdddZdd ZdS )simple_producer c C s || _ || _d S )N)r buffer_size)r r rJ r
r
r r
s zsimple_producer.__init__c C sJ t | j| jkr6| jd | j }| j| jd | _|S | j}d| _|S d S )Nr )r( r rJ )r resultr
r
r r? s zsimple_producer.moreN)rI )rE rF rG r
r? r
r
r
r rH s
rH c C s6 t |d }x$|r0| j|d | r0|d8 }qW |S )Nr" )r( endswith)ZhaystackZneedlelr
r
r r* / s r* )r collectionsr r r rH r* r
r
r
r 0 s
\