3
�ga�" � @ sf d dl Z d dlZd dlZd dlZd dlmZmZmZmZ d dl Z d dl
m Z d dl m Z
ddlm Z ddlmZmZmZmZ ddlmZ ddlmZ ddlmZ dd lm Z m Z m Z dd
l m Z dd l!m"Z"m#Z# ej$e� feee%e&f e"d �d
d�Z'e j(ee j)e j*f ee+e+f d�dd�Z,e j(ee"d�dd�Z-ed�dd�Z.eee%e&f e%ee% d�dd�Z/dS )� N)�List�Optional�Tuple�Union)� __version__� )� OUT_REQ_BODY� OUT_REQ_HEAD�
OUT_RESP_BODY�
OUT_RESP_HEAD)�collect_messages)� Environment)�
Downloader)�
write_message� write_stream�MESSAGE_SEPARATOR_BYTES)�plugin_manager)�
ExitStatus�http_status_to_exit_status)�args�returnc C s� | ^}} t jj|�|_t| |j�} tj� ddlm } |j
j rJ|j
j | } d| k}|p\d| k}|rzt |� | dgkrzt
jS t
j}y|j| |d�}W nv tk
r� |jjd� |r�� t
j}Y �n� tk
�r } z*|jt
jkr�|jjd� |r� t
j}W Y dd}~X �nzX yt||d�}W �nd tk
�rN |jjd� |�rB� t
j}Y �n4 tk
�r� } z.|jt
jk�r�|jjd� |�r�� t
j}W Y dd}~X n� tjk
�r� t
j}|jd|j � d �� Y n� tj k
�r� t
j }|jd
|j � d �� Y n� t k
�r� } zlt!|�} t"|d ��rH|j#}
t"|
d
��rH| � d|
j$� d|
j%� �} |jt&|�j'� d| � �� |�rj� t
j}W Y dd}~X nX |S )z�
The main function.
Pre-process args, handle some special types of invocations,
and run the main program with error handling.
Return exit status code.
r )�parserz--debugz --traceback)r �env�
NzRequest timed out (zs).z$Too many redirects (--max-redirects=z).�request�urlz while doing a z request to URL: z: )(�os�path�basename� program_name�decode_raw_args�stdin_encodingr Zload_installed_pluginsZcli.definitionr �configZdefault_options�print_debug_infor �SUCCESS�
parse_args�KeyboardInterrupt�stderr�writeZ ERROR_CTRL_C�
SystemExit�code�ERROR�program�requestsZTimeoutZ
ERROR_TIMEOUT� log_errorZtimeoutZTooManyRedirectsZERROR_TOO_MANY_REDIRECTSZ
max_redirects� Exception�str�hasattrr �methodr �type�__name__) r r r r Zinclude_debug_infoZinclude_traceback� exit_statusZ parsed_args�e�msgr � r8 �/usr/lib/python3.6/core.py�main s|
r: )r �messager c C s8 t jt| jkt| jkft jt| jkt| jkfit|� S )N) r- �PreparedRequestr �output_optionsr �Responser r
r3 )r r; r8 r8 r9 �get_output_optionsp s
r? )r r r c s6 t j}d}d�d}�fdd�}td�� ��fdd� }�zƈ jrdd� _t� j�j� jd�}|j � j
� t � �j j
|d �}d
}d
} �x|D ]�}
t|
tj�} t� |
d �\} }
|
}| r�| s�|
r�|sj r�|� d
}| �r�s�|
�|
�rlt|
jttf� }| }|�o�j}nf|
}� j�s|�rlt|
j� jd �}|t jk�rl�j �sJ� jd
k�rl�jd|
jj� d|
jj � �dd� t |
�� | |d� |
} q�W |�r�|� |�r�|t jk�r�|j �j |d�\}}t ||d
d� |j!� |j"�r�t j#}�jd|jj$� d|jj%� �� |S |�r|j& �r|j'� � j�r0� j(�r0� jj)� X dS )z3
The main program without error handling.
Nc s t � jd� j�jt� d S )N�buffer)�getattr�stdoutr( r r8 )r r8 r9 �separate� s zprogram.<locals>.separate)�chunkc sL t t� jko�o| �}|rHtj� }d|_| |_�j|_t|�� ddd� d S )NTF)�requests_messager r � with_body� with_headers) �boolr r= r- r< Zis_body_upload_chunk�body�headersr )rD Zshould_pipe_to_stdoutr7 )r r �initial_requestr8 r9 �request_body_read_callback� s
z+program.<locals>.request_body_read_callbackT)� output_fileZ
progress_fileZresume)r Z
config_dirrL F)r r; )Z http_status�followr zHTTP � Zwarning)�level)rE r r rG rF )Z initial_url�final_response)�streamZoutfile�flushzIncomplete download: size=z
; downloaded=)*r r$ �bytesZdownloadrN r rM r' Zdownload_resumeZ pre_requestrJ r r" Z directory�
isinstancer- r< r? Z
stdout_isattyrI r0 Z check_statusr Z status_code�quietr. �raw�status�reasonr �startr r ZfinishZ interruptedr+ Z
total_sizeZ
downloadedZfinishedZfailedZoutput_file_specified�close)r r r5 Z
downloaderrQ rC rL ZmessagesZforce_separatorZprev_with_bodyr; Z
is_requestrG rF Z
do_write_bodyZis_streamed_uploadZdownload_streamZ download_tor8 )r r rK r9 r, � sl
""
r, )r c C s� | j jdt� d�dt� d�dt� d�dtj� dtj� d�tj � � dtj
� � �g� | j j d� | j j t | �� | j j d� | j j t t
�� | j j d� d S )NzHTTPie r z Requests z Pygments zPython rO z
)r' �
writelines�httpie_version�requests_version�pygments_version�sys�version�
executable�platform�system�releaser( �reprr )r r8 r8 r9 r# � s
r# )r r! r c s � fdd�| D �S )zS
Convert all bytes args to str
by decoding them using stdin encoding.
c s& g | ] }t |�tkr |j� �n|�qS r8 )r3 rT �decode)�.0�arg)r! r8 r9 �
<listcomp>� s z#decode_raw_args.<locals>.<listcomp>r8 )r r! r8 )r! r9 r � s
r )0�argparser rc r` Ztypingr r r r r- Zpygmentsr r_ r^ � r] Z
cli.constantsr r r
r Zclientr �contextr
Z downloadsr Z
output.writerr r r Zplugins.registryr rX r r �argvr0 rT r: Z Namespacer< r> rH r? r, r# r r8 r8 r8 r9 �<module> s0 &[
\