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 || _ || _|| _|| _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)r � 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 r0 �kwargs)� __class__r r r! S s zRawStream.__init__)r" c C s | j j| j�S )N)r r& r0 )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 )r2 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)
r/ r! r Z content_type�mime�encodingZ
stdout_isattyZstdout_encodingr �output_encoding)r �envr6 r7 r1 r: )r2 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_linesr3 r r
r9 r r: )r �line�lfr r r r& x s
zEncodedStream.iter_body) r r r r r3 r �strr! r r+ r& r4 r r )r2 r r5 [ s r5 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)r/ r! rC rB )r rB rC r1 )r2 r r r! � s zPrettyStream.__init__)r" c C s | j j| jj�j| j�S )N)rC 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 |�\| _}| j
|�V d S t � �| j
|�| V d}qW d S )NTr< F) r r= r3 rB �
get_converterr8 � bytearrayr �extend�convert� 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)Zcontentr8 ) �
isinstancer@ r
r9 rC Z format_bodyr8 r r: )r r( r r r rH � s
zPrettyStream.process_body)r r r r r3 r r
r! r+ r% r r& r r@ rH r4 r r )r2 r rA � s rA 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< ) rE r r& r3 rB rD r8 r rF rG rH )r rI rJ r( r r r r& � s z BufferedPrettyStream.iter_bodyNi ( )r r r r r3 r r+ r& r r r r rL � s rL N) �abcr r � itertoolsr Ztypingr r r Z
processingr r
�contextr r9 r
r r Zmodelsr r � Exceptionr r r r- r5 rA rL r r r r �<module> s 0%6