3
�ga�( � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
m Z m Z d dl
mZmZ d dlZd dlZddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd l m Z m Z dd
l m Z m!Z!m"Z" ddl#m$Z$m%Z% ej&� de� �Z'dZ(e(� d�Z)de� �Z*d0e j+ee
e,gdf e e ej-ej.f d�dd�Z/edd� �Z0d1e1e2e2ej3d�dd�Z4e5d�d d �Z6eed �d d �Z7e j+ed!�d"d#�Z8e j+e5d!�d$d%�Z9e j+e5d!�d&d'�Z:dd(d)� fe j+ee5d*�d+d,�Z;e2e2e2d-�d.d/�Z<dS )2� N)�contextmanager)�Path)�Callable�Iterable�Union)�urlparse�
urlunparse� )� __version__)�RequestHeadersDict)�UTF8)�plugin_manager)�get_httpie_session)�!AVAILABLE_SSL_VERSION_ARG_MAPPING�HTTPieHTTPSAdapter)�compress_request�prepare_request_body�#get_multipart_data_and_content_type)�get_expired_cookies� repr_dictz+application/x-www-form-urlencoded; charset=zapplication/jsonz , */*;q=0.5zHTTPie/)�args�
config_dir�request_body_read_callback�returnc c s( d }d }| j s| jr<t|| j p"| j| jjd�| jd�}|j}t| ||d�}t| �}t| �}t | j
| j t |d �d�}|r�|j
|d � |j|_| jr�| jj| jjd�|_n|jr�|j|d< | jr�t|� tjf |�} |j| �}
| jr�t| j|
jd �|
_| j�r|
j�rt |
| jd
kd � d } g } x�|
�r�|
V | j �s�|j f d
|
ji|��}
t | j �� |j f d|
i|
|��}W d Q R X | t!|jjdd��7 } | d
7 } |j"�r�| j#�r�| | j#k�r�tj$�| j%�r�|j"}
| j&�r"|V �q"|V P �q"W |�r$|j'� �s | j �r$|j|_|j(dd� | D �� |j)� d S )NZHost)r Z session_name�host�url)r � base_headersr �verify)� ssl_version�ciphersr �headers)�type�raw_auth�auth)�orig_url� prepped_urlr )�request�alwaysr r r& z
Set-Cookie� c s s | ]}|d V qdS )�nameN� )�.0Zcookier* r* � /usr/lib/python3.6/client.py� <genexpr>� s z#collect_messages.<locals>.<genexpr>)*ZsessionZsession_read_onlyr r �getr �make_request_kwargs�make_send_kwargs�#make_send_kwargs_mergeable_from_env�build_requests_sessionr r �boolZupdate_headersZcookiesZ auth_pluginZ auth_typer" r# �debug� dump_request�requestsZRequestZprepare_requestZ
path_as_is�ensure_path_as_is�compress�bodyr �offlineZmerge_environment_settings� max_headers�sendr �nextZ
max_redirectsZTooManyRedirectsZfollow�allZis_newZremove_cookiesZsave)r r r Zhttpie_sessionZhttpie_session_headersZrequest_kwargsZ send_kwargsZ send_kwargs_mergeable_from_env�requests_sessionr& Zprepared_requestZresponse_countZexpired_cookiesZsend_kwargs_mergedZresponser* r* r, �collect_messages" s�
r@ c c s2 t jj}| ptd�t j_z
d V W d |t j_X d S )NZInf)�httpZclientZ _MAXHEADERS�float)�limitZorigr* r* r, r; � s
r; )r r r r c C s^ t j� }t|| |rt| nd d�}|jd|� x*tj� D ] }|� }|j|j|j� d� q8W |S )N)r r r zhttps://)�prefixZadapter) r6 �Sessionr r Zmountr
Zget_transport_pluginsrD Z get_adapter)r r r r? Z
https_adapterZ
plugin_clsZtransport_pluginr* r* r, r2 � s r2 )�kwargsc C s t jjdt| �� d�� d S )Nz
>>> requests.request(**z)
)�sys�stderr�writer )rF r* r* r, r5 � s r5 )r r c C sJ t � }x>| j� D ]2\}}|d k r:|j� }t|t�r:|j� }|||< qW |S )N)r �items�strip�
isinstance�str�encode)r Z
final_headersr) �valuer* r* r, �finalize_headers� s
rP )r r c C s` t dti�}| jo| j }| js$|rFt|d< | js<|r\| jr\t|d< n| jr\| j r\t|d< |S )Nz
User-AgentZAcceptz Content-Type) r �
DEFAULT_UA�data�form�json� JSON_ACCEPT�JSON_CONTENT_TYPE�files�FORM_CONTENT_TYPE)r Zdefault_headers� auto_jsonr* r* r, �make_default_headers� s
rZ c C s | j pd dd�S )NF)�timeoutZallow_redirects)r[ )r r* r* r, r0 � s r0 c C sT d }| j r | j }| jr || jf}dd� | jD �dddddd�j| jj� | j�|d�S )Nc S s i | ]}|j |j�qS r* )rO �key)r+ �pr* r* r, �
<dictcomp>� s z7make_send_kwargs_mergeable_from_env.<locals>.<dictcomp>TF)�yes�true�noZfalse)Zproxies�streamr �cert)rc Zcert_key�proxyr. r �lower)r rc r* r* r, r1 � s
r1 c C s | S )Nr* )�chunkr* r* r, �<lambda>� s rg )r r r c
C s� | j }| j}|o| j }| js"|r@t|t�r@|r<tj|�}nd}t| �}|rV|j|� |j| j � | j
r~| j r~d|kr~d|d< t |�}| jr�|s�| j
r�t| j| j| j jd�d�\}|d< | jj� | j|t||| j | j
|jd�d�| j| jj� d�S ) zJ
Translate our `args` into `requests.Request` keyword arguments.
r( zTransfer-Encoding�chunkedz Content-Type)rR �boundaryZ content_typezContent-Length)r9 Zbody_read_callbackrh r: Zcontent_length_header_value)�methodr r rR r# �params)rW rR rS rT rL �dict�dumpsrZ �updater r: rh rP Z multipartr Zmultipart_datari r. rj re r r r# rk rJ )r r r rW rR rY r r* r* r, r/ � s<
r/ )r$ r% r c C s4 t | �t |� }}|j� d|ji�}tt|j� ��S )a�
Handle `--path-as-is` by replacing the path component of the prepared
URL with the path component from the original URL. Other parts stay
untouched because other (welcome) processing on the URL might have
taken place.
<https://github.com/httpie/httpie/issues/895>
<https://ec.haxx.se/http/http-basics#path-as-is>
<https://curl.haxx.se/libcurl/c/CURLOPT_PATH_AS_IS.html>
>>> ensure_path_as_is('http://foo/../', 'http://foo/?foo=bar')
'http://foo/../?foo=bar'
�path)r �_asdictro r �tuple�values)r$ r% Z parsed_origZparsed_preppedZ
final_dictr* r* r, r7 ( s r7 )N)NN)=�argparseZ http.clientrA rT rG �
contextlibr Zpathlibr Ztypingr r r Z urllib.parser r r6 Zurllib3r( r
Z cli.dictsr �encodingr Zplugins.registryr
Zsessionsr Zsslr r Zuploadsr r r Zutilsr r Zdisable_warningsrX rV rU rQ Z Namespace�bytesZPreparedRequestZResponser@ r; r3 rM rE r2 rl r5 rP rZ r0 r1 r/ r7 r* r* r* r, �<module> sJ
a 3