File: //lib/python3.6/site-packages/supervisor/medusa/__pycache__/producers.cpython-36.pyc
3
�k�` # � @ s� d Z ddlmZ ddlmZ G dd� d�ZG dd� d�ZG dd � d �ZG d
d � d �ZG d d
� d
�Z G dd� d�Z
G dd� d�Z G dd� d�Z G dd� d�Z
G dd� d�Zy ddlZW n ek
r� dZY nX G dd� d�ZG dd � d �ZdS ) z;$Id: producers.py,v 1.9 2004/04/21 13:56:28 akuchling Exp $� )�find_prefix_at_end)�as_bytesc @ s" e Zd ZdZddd�Zdd� ZdS ) �simple_producerzproducer for a string� c C s || _ || _d S )N)�data� buffer_size)�selfr r � r � /usr/lib/python3.6/producers.py�__init__ s zsimple_producer.__init__c C sJ t | j�| jkr6| jd | j� }| j| jd � | _|S | j}d| _|S d S )N� )�lenr r )r �resultr r r
�more s zsimple_producer.moreN)r )�__name__�
__module__� __qualname__�__doc__r r r r r r
r s
r c @ s" e Zd ZdZddd�Zdd� ZdS ) �scanning_producerz:like simple_producer, but more efficient for large strings� c C s || _ || _d| _d S )Nr )r r �pos)r r r r r r
r # s zscanning_producer.__init__c C sZ | j t| j�k rR| j }tt| j�| j | j �}| j||� }| j t|�7 _ |S dS d S )Nr )r r
r �minr )r ZlpZrpr r r r
r ( s zscanning_producer.moreN)r )r r r r r r r r r r
r ! s
r c @ s e Zd ZdZdd� Zdd� ZdS )�lines_producerz producer for a list of linesc C s
|| _ d S )N)�lines)r r r r r
r 8 s zlines_producer.__init__c C s: | j r2| j d d� }| j dd � | _ dj|�d S dS d S )N�2 z
� )r �join)r �chunkr r r
r ; s
zlines_producer.moreN)r r r r r r r r r r
r 5 s r c @ s e Zd ZdZdd� Zdd� ZdS )�buffer_list_producerz producer for a list of stringsc C s d| _ || _d S )Nr )�index�buffers)r r r r r
r H s z buffer_list_producer.__init__c C s6 | j t| j�krdS | j| j }| j d7 _ |S d S )Nr � )r r
r )r r r r r
r L s
zbuffer_list_producer.moreN)r r r r r r r r r r
r C s r c @ s$ e Zd ZdZd Zdd� Zdd� ZdS )
�
file_producerz(producer wrapper for file[-like] objectsr! � c C s d| _ || _d S )Nr )�done�file)r r% r r r
r Z s zfile_producer.__init__c C s< | j r
dS | jj| j�}|s4| jj� | `d| _ dS |S d S )Nr r! )r$ r% �read�out_buffer_size�close)r r r r r
r ^ s
zfile_producer.moreNi )r r r r r' r r r r r r
r" T s r" c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�output_producerz;Acts like an output file; suitable for capturing sys.stdoutc C s
d| _ d S )Nr )r )r r r r
r t s zoutput_producer.__init__c C s& |j d�}dj|�}| j|7 _d S )N�
z
)�splitr r )r r r r r r
�writew s
zoutput_producer.writec C s | j | d | _ d S )Nz
)r )r �liner r r
� writeline| s zoutput_producer.writelinec C s | j dj|� d | _ d S )Nz
)r r )r r r r r
�
writelines s zoutput_producer.writelinesc C s d S )Nr )r r r r
�flush� s zoutput_producer.flushc G s d S )Nr )r �argsr r r
� softspace� s zoutput_producer.softspacec C s0 | j r(| j d d� }| j dd � | _ |S dS d S )Ni r )r )r r r r r
r � s
zoutput_producer.moreN) r r r r r r, r. r/ r0 r2 r r r r r
r) r s r) c @ s e Zd ZdZdd� Zdd� ZdS )�composite_producerz$combine a fifo of producers into onec C s
|| _ d S )N)� producers)r r4 r r r
r � s zcomposite_producer.__init__c C s> x8t | j�r4| jd }|j� }|r&|S | jjd� qW dS d S )Nr r )r
r4 r �pop)r �p�dr r r
r � s
zcomposite_producer.moreN)r r r r r r r r r r
r3 � s r3 c @ s" e Zd ZdZd
dd�Zdd� ZdS ) �globbing_producerz�
'glob' the output from a producer into a particular buffer size.
helps reduce the number of calls to send(). [this appears to
gain about 30% performance on requests to a single channel]
r! r# c C s || _ d| _|| _d S )Nr )�producer�bufferr )r r9 r r r r
r � s zglobbing_producer.__init__c C sD x2t | j�| jk r2| jj� }|r.| j| | _qP qW | j}d| _|S )Nr )r
r: r r9 r )r r �rr r r
r � s
zglobbing_producer.moreN� )r< )r r r r r r r r r r
r8 � s
r8 c @ s e Zd ZdZdd� Zdd� ZdS )�hooked_producerz�
A producer that will call <function> when it empties,.
with an argument of the number of bytes produced. Useful
for logging/instrumentation purposes.
c C s || _ || _d| _d S )Nr )r9 �function�bytes)r r9 r> r r r
r � s zhooked_producer.__init__c C sF | j r>| j j� }|s(d | _ | j| j� n| jt|�7 _|S dS d S )Nr )r9 r r> r? r
)r r r r r
r � s
zhooked_producer.moreN)r r r r r r r r r r
r= � s r= c @ s" e Zd ZdZddd�Zdd� ZdS )�chunked_producera' A producer that implements the 'chunked' transfer coding for HTTP/1.1.
Here is a sample usage:
request['Transfer-Encoding'] = 'chunked'
request.push (
producers.chunked_producer (your_producer)
)
request.done()
Nc C s || _ || _d S )N)r9 �footers)r r9 rA r r r
r � s zchunked_producer.__init__c C sd | j r\| j j� }|r4dt|� }t|�d | d S d | _ | jrVdjdg| j �d S dS ndS d S )Nz%xs
� 0s
s 0
r )r9 r r
r rA r )r r �sr r r
r � s
zchunked_producer.more)N)r r r r r r r r r r
r@ � s
r@ Nc @ s" e Zd ZdZddd�Zdd� ZdS ) �compressed_producerz:
Compress another producer on-the-fly, using ZLIB
� c C s || _ tj|�| _d S )N)r9 �zlibZ compressobj�
compressor)r r9 �levelr r r
r s z compressed_producer.__init__c C sJ | j rBd}x2|s<| j j� }|s.d | _ | jj� S | jj|�}q W |S dS d S )Nr )r9 r rG r0 �compress)r Zcdatar r r r
r s
zcompressed_producer.moreN)rE )r r r r r r r r r r
rD � s
rD c @ s" e Zd ZdZd dd�Zdd� ZdS )
�escaping_producerz0A producer that escapes a sequence of characters�
.�
..c C s" || _ || _|| _d| _t| _d S )Nr )r9 �esc_from�esc_tor: r )r r9 rM rN r r r
r s
zescaping_producer.__init__c C sn | j }| j}| j| jj� }|rf|j||�}| j||�}|rZ|| d � | _|d | � S d| _|S n|S d S )Nr )rM rN r: r9 r �replacer )r rM rN r: �ir r r
r ' s zescaping_producer.moreN)rK rL )r r r r r r r r r r
rJ s
rJ )ZRCS_IDZasynchatr Zsupervisor.compatr r r r r r" r) r3 r8 r= r@ rF � ImportErrorrD rJ r r r r
�<module> s$