File: //usr/lib/python3.6/site-packages/supervisor/medusa/__pycache__/asynchat_25.cpython-36.opt-1.pyc
3
�k�`L* � @ sj d Z ddlZddlmZ ddlmZ ddlmZ G dd� dej�Z G dd � d �Z
G d
d � d �Z d d
� Z dS )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)� asyncore_25)�long)�as_bytesc @ s� e Zd ZdZdZdZd$dd�Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� 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 Nc C s( d| _ d| _t� | _tjj| ||� d S )N� )� ac_in_buffer�
ac_out_buffer�fifo�
producer_fifo�asyncore�
dispatcher�__init__)�selfZconn�map� r �!/usr/lib/python3.6/asynchat_25.pyr
? s zasync_chat.__init__c C s t d��d S )Nz must be implemented in subclass)�NotImplementedError)r �datar r r �collect_incoming_dataE s z async_chat.collect_incoming_datac C s t d��d S )Nz must be implemented in subclass)r )r r r r �found_terminatorH s zasync_chat.found_terminatorc C s
|| _ dS )zRSet the input delimiter. Can be a fixed string of any length, an integer, or NoneN)�
terminator)r Ztermr r r �set_terminatorK s zasync_chat.set_terminatorc C s | j S )N)r )r r r r �get_terminatorO s zasync_chat.get_terminatorc C s� y| j | j�}W n tjk
r. | j� d S X | j|7 _�x`| j�r�t| j�}| j� }|st| j| j� d| _qBt |t
�s�t |t �r�|}||k r�| j| j� d| _| j |8 _ n2| j| jd |� � | j|d � | _d| _ | j
� qBt|�}| jj|�}|dk�rD|dk�r&| j| jd |� � | j|| d � | _| j
� qBt| j|�}|�r�||k�r�| j| jd | � � | j| d � | _P qB| j| j� d| _qBW d S )Nr r � ���)Zrecv�ac_in_buffer_size�socket�error� handle_errorr �lenr r �
isinstance�intr r r �find�find_prefix_at_end)r r Zlbr �nZterminator_len�indexr r r � handle_readW sJ
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 |�}| jjt|�� | j� d S )N)r r
�push�simple_producerr' )r r r r r r+ � 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 t | j�| jkS )z4predicate for inclusion in the readable for select())r r r )r r r r �readable� s zasync_chat.readablec C s | j dko| jj� o| j S )z4predicate for inclusion in the writable for select()r )r r
�is_empty� connected)r r r r �writable� s
zasync_chat.writablec C s | j jd� dS )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 s� x�t | j�r�| jj� }|d kr:| js6| jj� | j� d S t|t�r`| jj� | j|7 _d S |j� }|r|| j| | _d S | jj� qd S qW d S )N) r r
�firstr �popr) r �bytes�more)r �pr r r r �
refill_buffer� s$
zasync_chat.refill_bufferc C sx | j }t| j�|k r | j� | jrt| jrty,| j| jd |� �}|rR| j|d � | _W n tjk
rr | j� d S X d S )N) �ac_out_buffer_sizer r r8 r0 �sendr r r )r ZobsZnum_sentr r r r' � s zasync_chat.initiate_sendc C s& d| _ d| _x| jr | jj� qW d S )Nr )r r r
r4 )r r r r �discard_buffers� s zasync_chat.discard_buffers)NN)�__name__�
__module__� __qualname__�__doc__r r9 r
r r r r r& r( r* r+ r- r. r1 r2 r8 r' r; r r r r r 6 s&
A
r c @ s e Zd Zddd�Zdd� ZdS )r, � c C s || _ || _d S )N)r � buffer_size)r r rA 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 rA )r �resultr r r r6 � s zsimple_producer.moreN)r@ )r< r= r> r
r6 r r r r r, � s
r, c @ s> e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )r Nc C s |s g | _ n|| _ d S )N)�list)r rC r r r r
� s z
fifo.__init__c C s
t | j�S )N)r rC )r r r r �__len__ s z fifo.__len__c C s
| j g kS )N)rC )r r r r r/ s z
fifo.is_emptyc C s
| j d S )Nr )rC )r r r r r3 s z
fifo.firstc C s | j j|� d S )N)rC �append)r r r r r r+ s z fifo.pushc C s | j rd| j jd�fS dS d S )Nr r )r N)rC r4 )r r r r r4 s zfifo.pop)N) r< r= r> r
rD r/ r3 r+ r4 r r r r r � s
r c C s6 t |�d }x$|r0| j|d |� � r0|d8 }qW |S )Nr )r �endswith)ZhaystackZneedle�lr r r r# $ s r# )
r? r Zsupervisor.medusar r Zsupervisor.compatr r r r r, r r# r r r r �<module>/ s 7(