3
�ga � @ s� d dl mZmZ d dlmZ d dlmZmZmZ ddl m
Z
m Z ddl m
Z
ddlmZmZmZ dd lmZ d
ZG d d � d e�ZG d
d� de�ZG dd� ded�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Z dS )� )�ABCMeta�abstractmethod)�chain)�Callable�Iterable�Union� )�
Conversion�
Formatting� )� Environment)� smart_decode� smart_encode�UTF8)� HTTPMessages�
+-----------------------------------------+
| NOTE: binary data not shown in terminal |
+-----------------------------------------+c @ s e Zd ZdZdS )�DataSuppressedErrorN)�__name__�
__module__� __qualname__�message� r r � /usr/lib/python3.6/streams.pyr s r c @ s e Zd ZdZeZdS )�BinarySuppressedErrorzbAn error indicating that the body is binary and won't be written,
e.g., for terminal output).N)r r r �__doc__�BINARY_SUPPRESSED_NOTICEr r r r r r s r c @ sb e Zd ZdZdeeegdf d�dd�Zed�dd �Ze e
e d�d
d ��Z e
e d�d d
�Z dS )�
BaseStreamz&Base HTTP message output stream class.TN)�msg�on_body_chunk_downloadedc C s( |s |s t �|| _|| _|| _|| _dS )z�
:param msg: a :class:`models.HTTPMessage` subclass
:param with_headers: if `True`, headers will be included
:param with_body: if `True`, body will be included
N)�AssertionErrorr � with_headers� with_bodyr )�selfr r r r r r r �__init__ s
zBaseStream.__init__)�returnc C s | j jj� S )zReturn the headers' bytes.)r �headers�encode)r! r r r � get_headers3 s zBaseStream.get_headersc C s dS )z)Return an iterator over the message body.Nr )r! r r r � iter_body7 s zBaseStream.iter_bodyc c s� | j r| j� V dV | jr�y,x&| j� D ]}|V | jr(| j|� q(W W n6 tk
r~ } z| j rfdV |jV W Y dd}~X nX dS )z#Return an iterator over `self.msg`.s
�
N)r r& r r' r r r )r! �chunk�er r r �__iter__; s
zBaseStream.__iter__)TTN)
r r r r r r �bytesr" r& r r r' r+ r r r r r s r )� metaclassc s> e Zd ZdZd
ZdZef� fdd� Zee d�dd �Z � Z
S ) � RawStreamz5The message is streamed in chunks with no processing.i �d r c s t � jf |� || _d S )N)�superr" �
chunk_size)r! r1 �kwargs)� __class__r r r" S s zRawStream.__init__)r# c C s | j j| j�S )N)r r' r1 )r! r r r r' W s zRawStream.iter_bodyi � ) r r r r �
CHUNK_SIZEZCHUNK_SIZE_BY_LINEr" r r, r' �
__classcell__r r )r3 r r. M s
r. c sH e Zd ZdZdZe� ddfeed�� fdd�
Zee d�dd �Z
� Z S )
�
EncodedStreamz�Encoded HTTP message stream.
The message bytes are converted to an encoding suitable for
`self.env.stdout`. Unicode errors are replaced and binary data
is suppressed. The body is always streamed by line.
r N)�mime_overwrite�encoding_overwritec sN t � jf |� |p| jj| _|p&| jj| _|jr8|j}n| jj}|pFt| _ d S )N)
r0 r" r Z content_type�mime�encodingZ
stdout_isattyZstdout_encodingr �output_encoding)r! �envr7 r8 r2 r; )r3 r r r" e s zEncodedStream.__init__)r# c c sL xF| j j| j�D ]4\}}d|kr&t� �t|| j�}t|| j�| V qW d S )N� )r �
iter_linesr4 r r
r: r r; )r! �line�lfr r r r' x s
zEncodedStream.iter_body) r r r r r4 r �strr" r r, r' r5 r r )r3 r r6 [ s r6 c sd e Zd ZdZdZeed�� fdd� Zed�dd�Z e
e d�d d
�Z e e
ef ed �d d
�Z� ZS )� PrettyStreamz�In addition to :class:`EncodedStream` behaviour, this stream applies
content processing.
Useful for long-lived HTTP responses that stream by lines
such as the Twitter streaming API.
r )�
conversion�
formattingc s t � jf |� || _|| _d S )N)r0 r" rD rC )r! rC rD r2 )r3 r r r" � s zPrettyStream.__init__)r# c C s | j j| jj�j| j�S )N)rD Zformat_headersr r$ r% r; )r! r r r r& � s zPrettyStream.get_headersc c s� d}| j j| j�}x�|D ]�\}}d|kr�|r�| jj| j�}|r�t� }x0t||fg|�D ] \}}|j|� |j|� qVW |j |�\| _}t
|t �s�t �| j
|�V d S t� �| j
|�| V d}qW d S )NTr= F)r r> r4 rC �
get_converterr9 � bytearrayr �extend�convert�
isinstancerA r � process_bodyr )r! Z first_chunkr> r? r@ � converter�bodyr r r r' � s$
zPrettyStream.iter_body)r) r# c C s4 t |t�st|| j�}| jj|| jd�}t|| j�S )N)Zcontentr9 ) rI rA r
r: rD Z format_bodyr9 r r; )r! r) r r r rJ � s
zPrettyStream.process_body)r r r r r4 r r
r" r, r& r r' r rA rJ r5 r r )r3 r rB � s rB c @ s& e Zd ZdZdZee d�dd�ZdS ) �BufferedPrettyStreamz�The same as :class:`PrettyStream` except that the body is fully
fetched before it's processed.
Suitable regular HTTP responses.
i �
)r# c c sv d }t � }xF| jj| j�D ]4}| rDd|krD| jj| j�}|sDt� �|j|� qW |rf|j |�\| _}| j
|�V d S )Nr= ) rF r r' r4 rC rE r9 r rG rH rJ )r! rK rL r) r r r r' � s z BufferedPrettyStream.iter_bodyNi ( )r r r r r4 r r, r' r r r r rM � s rM N) �abcr r � itertoolsr Ztypingr r r Z
processingr r
�contextr r: r
r r Zmodelsr r � Exceptionr r r r. r6 rB rM r r r r �<module> s 0%6