3
�gaw � @ s( d dl Z d dlZd dlmZmZmZmZmZ d dlZddl m
Z
ddl m Z m
Z
mZ ddlmZmZ ddlmZmZmZmZmZ d Zej� Zdeejej f e
e j d �d d
�Z eeeef e d�dd�Z dee d�dd�Z!e j e
eejej f e e d�dd�Z"e
e j ee eed e#f d�dd�Z$dS )� N)�IO�TextIO�Tuple�Type�Union� )� Environment)� HTTPRequest� HTTPResponse� HTTPMessage� )�
Conversion�
Formatting)�
BaseStream�BufferedPrettyStream�
EncodedStream� PrettyStream� RawStreamz
F)�requests_message�env�argsc C s� |p|s d S t ||| ||d�|j|jp*|jd�}y*|jrNd|jkrNtf |� n
tf |� W nP tk
r� } z4|j pv|j
}| r�|j t j kr�|j
jd� n� W Y d d }~X nX d S )N)r r r � with_body� with_headers)�stream�outfile�flushZcolors�
)� build_output_stream_for_message�stdout�
stdout_isattyr Z
is_windows�prettify� write_stream_with_colors_win� write_stream�OSError�debug� traceback�errnoZEPIPE�stderr�write)r r r r r Zwrite_stream_kwargs�eZshow_traceback� r* � /usr/lib/python3.6/writer.py�
write_message s& r, )r r r c C sL y
|j }W n tk
r" |}Y nX x"| D ]}|j|� |r*|j� q*W dS )zWrite the output stream.N)�buffer�AttributeErrorr( r )r r r Zbuf�chunkr* r* r+ r" 6 s
r" r c C sN d}|j }x>| D ]6}||kr.|j|j|�� n |jj|� |r|j� qW dS )z�Like `write`, but colorized chunks are written as text
directly to `outfile` to ensure it gets processed by colorama.
Applies only to Windows and colorized terminal output.
s [N)�encodingr( �decoder- r )r r r Zcolorr0 r/ r* r* r+ r! H s
r! )r r r r r c c sl t jtt jtit|� }t|| |d�\}}|f ||�||d�|��E d H |jrh|rht|dd� rht V d S )N)r r � message_type)�msgr r Zis_body_upload_chunkF)
�requests�PreparedRequestr �Responser
�type�get_stream_type_and_kwargsr �getattr�MESSAGE_SEPARATOR_BYTES)r r r r r r2 � stream_class�
stream_kwargsr* r* r+ r ] s
r )r r r2 �returnc C s� | j r,|j r,t}d|jr"tjntji}ndt}d| i}|tkrT|j|j |j
d�� |jr�|jrdt nt }|jt
� t| |j|j|j|jd�d�� ||fS )zMPick the right stream type and kwargs for it based on `env` and `args`.
Z
chunk_sizer )Zmime_overwriteZencoding_overwrite)r �groupsZ color_schemeZ
explicit_json�format_options)Z
conversionZ
formatting)r r r r ZCHUNK_SIZE_BY_LINEZ
CHUNK_SIZEr r
�updateZ
response_mimeZresponse_charsetr r r
r ZstyleZjsonr? )r r r2 r; r< r* r* r+ r8 z s,
r8 )FF)%�argparser& Ztypingr r r r r r4 �contextr Zmodelsr r
r Z
processingr
r Zstreamsr r r r r ZMESSAGE_SEPARATOR�encoder: r5 r6 Z Namespacer, �boolr" r! r �dictr8 r* r* r* r+ �<module> s4 "