U
��.e�r � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
mZ ddlmZmZmZmZ ddlmZmZmZ dd lmZmZ dd
lm Z dd l m Z m Z m Z dd l!m"Z"m#Z#m$Z$m%Z% dd
l&m'Z' ddl(m)Z) ddl m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ ddl0m1Z1 ddlm2Z2 ej3dk�rNz
ej4Z5W n e6k
�rJ ej7Z5Y nX nejZ5e fdd�Z8e fdd�Z9G dd� de:�Z;G dd� de;�ZI s z!merge_setting..)�
isinstancer
r �update�items)Zrequest_settingZsession_setting�
dict_classZmerged_settingZ none_keys�keyr' r' r+ �
merge_setting1 s �� r2 c C s@ |dks|� d�g kr| S | dks0| � d�g kr4|S t| ||�S )z�Properly merges both requests and session hooks.
This is necessary because when request_hooks == {'response': []}, the
merge breaks Session hooks entirely.
N�response)�getr2 )Z
request_hooksZ
session_hooksr0 r' r' r+ � merge_hooksP s
r5 c @ s> e Zd Zdd� Zdd� Zddd �Zd
d � Zd d
� Zdd� ZdS )�SessionRedirectMixinc C s, |j r(|jd }tr |�d�}t|d�S dS )z7Receives a Response. Returns a redirect URI or ``None``�location�latin1�utf8N)Z is_redirect�headersr �encoder )�self�respr7 r' r' r+ �get_redirect_targeta s
z(SessionRedirectMixin.get_redirect_targetc C s� t |�}t |�}|j|jkr dS |jdkrL|jdkrL|jdkrL|jdkrLdS |j|jk}|j|jk}t�|jd�df}|s�|j|kr�|j|kr�dS |p�|S )zFDecide whether Authorization header should be removed when redirectingTZhttp)�P NZhttps)i� NFN)r Zhostname�schemeZportr r4 )r< Zold_urlZnew_urlZ
old_parsedZ
new_parsedZ changed_portZchanged_schemeZ default_portr' r' r+ �should_strip_authv s$ �� �z&SessionRedirectMixin.should_strip_authFNTc
k sZ g }
| � |�} t|j�j} | �rV|�� }
|
�|� |
dd� |_z
|j W n( tt t
fk
rr |j j dd� Y nX t
|j�| jkr�td| j |d��|�� | �d�r�t|j�}dt|j�| f } t| �}|jd kr�| r�|j| d
�}n |jr�|j} |�� } |j�st|jt| ��} nt| �} t| �|
_| �|
|� |jtj tj fk�rld }|D ]}|
j � |d� �qPd|
_ |
j }z
|d = W n t!k
�r� Y nX t"|
j#||j � t$|
j#| j%� |
�&|
j#� | �'|
|�}| �(|
|� |
j)dk �o�d
|k�p�d|k}|�rt*|
� |
}|�r|V q| j+|f|||||dd�| ��}t"| j%|
|j � | � |�} |V qdS )zBReceives a Response. Returns a generator of Responses or Requests.r NF)Zdecode_contentzExceeded %s redirects.)r3 z//z%s:%s� )�fragment)�Content-Lengthz Content-Type�Transfer-EncodingZCookierD rE )�stream�timeout�verify�cert�proxies�allow_redirects),r> r �urlrC �copy�append�history�contentr r � RuntimeError�raw�read�len�
max_redirectsr �close�
startswithr r@ �_replaceZgeturlZnetlocr r �rebuild_method� status_coder$ Ztemporary_redirectZpermanent_redirectr: �popZbody�KeyErrorr Z_cookiesr �cookiesZprepare_cookies�rebuild_proxies� rebuild_authZ_body_positionr# �send)r< r= �reqrF rG rH rI rJ �yield_requestsZadapter_kwargsZhistrL Zprevious_fragment�prepared_requestZ parsed_rurlZparsedZpurged_headers�headerr: Z
rewindabler' r' r+ �resolve_redirects� s�
����
z&SessionRedirectMixin.resolve_redirectsc C sR |j }|j}d|kr*| �|jj|�r*|d= | jr8t|�nd}|dk rN|�|� dS )z�When being redirected we may want to strip authentication from the
request to avoid leaking credentials. This method intelligently removes
and reapplies authentication where possible to avoid credential loss.
Z
AuthorizationN)r: rL rA �request� trust_envr Z prepare_auth)r< rc r3 r: rL Znew_authr' r' r+ r_ s
z!SessionRedirectMixin.rebuild_authc
C s� |dk r |ni }|j }|j}t|�j}|�� }|�d�}t||d�}| jr||s|t||d�} | �|| �d��}
|
r||� ||
� d|kr�|d= zt
|| �\} } W n t k
r� d\} } Y nX | r�| r�t | | �|d< |S )a� This method re-evaluates the proxy configuration by considering the
environment variables. If we are redirected to a URL covered by
NO_PROXY, we strip the proxy configuration. Otherwise, we set missing
proxy keys for this URL (in case they were stripped by a previous
redirect).
This method also replaces the Proxy-Authorization header where
necessary.
:rtype: dict
N�no_proxy�rh �allzProxy-Authorization)NN)
r: rL r r@ rM r4 r! rg r �
setdefaultr" r\ r )
r< rc rJ r: rL r@ Z new_proxiesrh Z bypass_proxyZenviron_proxies�proxyZusernameZpasswordr' r' r+ r^ s*
z$SessionRedirectMixin.rebuild_proxiesc C sX |j }|jtjkr |dkr d}|jtjkr6|dkr6d}|jtjkrN|dkrNd}||_ dS )z�When being redirected we may want to change the method of the request
based on certain specs or browser behavior.
�HEAD�GET�POSTN)�methodrZ r$ Z see_other�foundZmoved)r< rc r3 rp r' r' r+ rY = s z#SessionRedirectMixin.rebuild_method)FNTNNF) �__name__�
__module__� __qualname__r> rA re r_ r^ rY r' r' r' r+ r6 _ s �
q)r6 c @ s� e Zd ZdZdddddddd d
d d d
dg
Zdd� Zdd� Zdd� Zdd� Zd7dd�Z dd � Z
d d � Z d d � Z d8d!d"�Z
d9d#d$�Zd:d%d&�Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� ZdS );�Sessiona� A Requests session.
Provides cookie persistence, connection-pooling, and configuration.
Basic Usage::
>>> import requests
>>> s = requests.Session()
>>> s.get('https://httpbin.org/get')
Or as a context manager::
>>> with requests.Session() as s:
>>> s.get('https://httpbin.org/get')
r: r] �authrJ �hooks�paramsrH rI Zprefetch�adaptersrF rg rU c C sr t � | _d | _i | _t� | _i | _d| _d| _d | _ t
| _ d| _ t
i �| _t� | _| �dt� � | �dt� � d S )NFTzhttps://zhttp://)r r: rv rJ r rw rx rF rH rI r rU rg r r] r ry �mountr �r< r' r' r+ �__init__m s
zSession.__init__c C s | S r& r' r{ r' r' r+ � __enter__� s zSession.__enter__c G s | � � d S r& )rV )r< �argsr' r' r+ �__exit__� s zSession.__exit__c
C s� |j pi }t|tj�s t|�}ttt� | j �|�}|j}| jrR|sR| jsRt |j
�}t � }|j |j
�� |j
|j|j|jt|j| jtd�t|j| j�t|| j�|t|j| j�d�
|S )a� Constructs a :class:`PreparedRequest ` for
transmission and returns it. The :class:`PreparedRequest` has settings
merged from the :class:`Request ` instance and those of the
:class:`Session`.
:param request: :class:`Request` instance to prepare with this
session's settings.
:rtype: requests.PreparedRequest
)r0 )
rp rL �files�data�jsonr: rx rv r] rw )r] r- r Z CookieJarr r r
rv rg r rL r Zpreparerp �upperr� r� r� r2 r: r rx r5 rw )r< rf r] Zmerged_cookiesrv �pr' r' r+ �prepare_request� s0
�
� zSession.prepare_requestNTc C st t |�� ||||pi ||p i ||| d�
}| �|�}| p8i } | �|j| |
||�}| |
d�}|�|� | j|f|�}|S )a Constructs a :class:`Request `, prepares it and sends it.
Returns :class:`Response ` object.
:param method: method for the new :class:`Request` object.
:param url: URL for the new :class:`Request` object.
:param params: (optional) Dictionary or bytes to be sent in the query
string for the :class:`Request`.
:param data: (optional) Dictionary, list of tuples, bytes, or file-like
object to send in the body of the :class:`Request`.
:param json: (optional) json to send in the body of the
:class:`Request`.
:param headers: (optional) Dictionary of HTTP Headers to send with the
:class:`Request`.
:param cookies: (optional) Dict or CookieJar object to send with the
:class:`Request`.
:param files: (optional) Dictionary of ``'filename': file-like-objects``
for multipart encoding upload.
:param auth: (optional) Auth tuple or callable to enable
Basic/Digest/Custom HTTP Auth.
:param timeout: (optional) How long to wait for the server to send
data before giving up, as a float, or a :ref:`(connect timeout,
read timeout) ` tuple.
:type timeout: float or tuple
:param allow_redirects: (optional) Set to True by default.
:type allow_redirects: bool
:param proxies: (optional) Dictionary mapping protocol or protocol and
hostname to the URL of the proxy.
:param stream: (optional) whether to immediately download the response
content. Defaults to ``False``.
:param verify: (optional) Either a boolean, in which case it controls whether we verify
the server's TLS certificate, or a string, in which case it must be a path
to a CA bundle to use. Defaults to ``True``.
:param cert: (optional) if String, path to ssl client cert file (.pem).
If Tuple, ('cert', 'key') pair.
:rtype: requests.Response
)
rp rL r: r� r� r� rx rv r] rw )rG rK )r r� r� �merge_environment_settingsrL r. r` )r< rp rL rx r� r: r] r� rv rG rK rJ rw rF rH rI r� ra ZprepZsettingsZ send_kwargsr= r' r' r+ rf � s6 )�
��
zSession.requestc K s |� dd� | jd|f|�S )z�Sends a GET request. Returns :class:`Response` object.
:param url: URL for the new :class:`Request` object.
:param \*\*kwargs: Optional arguments that ``request`` takes.
:rtype: requests.Response
rK Trn �rk rf �r< rL �kwargsr' r' r+ r4 s z Session.getc K s |� dd� | jd|f|�S )z�Sends a OPTIONS request. Returns :class:`Response` object.
:param url: URL for the new :class:`Request` object.
:param \*\*kwargs: Optional arguments that ``request`` takes.
:rtype: requests.Response
rK TZOPTIONSr� r� r' r' r+ �options$ s zSession.optionsc K s |� dd� | jd|f|�S )z�Sends a HEAD request. Returns :class:`Response` object.
:param url: URL for the new :class:`Request` object.
:param \*\*kwargs: Optional arguments that ``request`` takes.
:rtype: requests.Response
rK Frm r� r� r' r' r+ �head/ s z Session.headc K s | j d|f||d�|��S )a� Sends a POST request. Returns :class:`Response` object.
:param url: URL for the new :class:`Request` object.
:param data: (optional) Dictionary, list of tuples, bytes, or file-like
object to send in the body of the :class:`Request`.
:param json: (optional) json to send in the body of the :class:`Request`.
:param \*\*kwargs: Optional arguments that ``request`` takes.
:rtype: requests.Response
ro )r� r� �rf )r< rL r� r� r� r' r' r+ �post: s z Session.postc K s | j d|fd|i|��S )au Sends a PUT request. Returns :class:`Response` object.
:param url: URL for the new :class:`Request` object.
:param data: (optional) Dictionary, list of tuples, bytes, or file-like
object to send in the body of the :class:`Request`.
:param \*\*kwargs: Optional arguments that ``request`` takes.
:rtype: requests.Response
ZPUTr� r� �r< rL r� r� r' r' r+ �putG s
z Session.putc K s | j d|fd|i|��S )aw Sends a PATCH request. Returns :class:`Response` object.
:param url: URL for the new :class:`Request` object.
:param data: (optional) Dictionary, list of tuples, bytes, or file-like
object to send in the body of the :class:`Request`.
:param \*\*kwargs: Optional arguments that ``request`` takes.
:rtype: requests.Response
ZPATCHr� r� r� r' r' r+ �patchS s
z
Session.patchc K s | j d|f|�S )z�Sends a DELETE request. Returns :class:`Response` object.
:param url: URL for the new :class:`Request` object.
:param \*\*kwargs: Optional arguments that ``request`` takes.
:rtype: requests.Response
ZDELETEr� r� r' r' r+ �delete_ s zSession.deletec
K sz |� d| j� |� d| j� |� d| j� |� d| j� t|t�rJtd��|�dd�}|� d�}|j
}| j |j d�}t
� }|j|f|�}t
� | } t| d �|_td
||f|�}|jr�|jD ]}
t| j|
j|
j� q�t| j||j� | j||f|�} |�rd d � | D �ng } | �r*| �d
|� | �� }| |_|�sjz"t| j||fddi|���|_W n tk
�rh Y nX |�sv|j |S )zISend a given PreparedRequest.
:rtype: requests.Response
rF rH rI rJ z#You can only send PreparedRequests.rK T)rL )Zsecondsr3 c S s g | ]}|�qS r' r' )r( r= r' r' r+ r, � s z Session.send..r rb ) rk rF rH rI rJ r- r �
ValueErrorr[ r4 rw � get_adapterrL �preferred_clockr` r �elapsedr rO r r] rf rR re �insert�nextZ_next�
StopIterationrP )
r< rf r� rK rF rw �adapter�start�rr� r= �genrO r' r' r+ r` i sB
"z Session.sendc
C s� | j rn|dk r|�d�nd}t||d�}|�� D ]\}} |�|| � q0|dksV|dkrntj�d�pltj�d�}t|| j�}t|| j �}t|| j
�}t|| j �}||||d�S )z^
Check the environment and merge it with some settings.
:rtype: dict
Nrh ri TZREQUESTS_CA_BUNDLEZCURL_CA_BUNDLE)rH rJ rF rI ) rg r4 r r/ rk �os�environr2 rJ rF rH rI )
r< rL rJ rF rH rI rh Z env_proxiesr) r* r' r' r+ r� � s
� �z"Session.merge_environment_settingsc C s> | j �� D ]"\}}|�� �|�� �r
| S q
td| ��dS )z~
Returns the appropriate connection adapter for the given URL.
:rtype: requests.adapters.BaseAdapter
z*No connection adapters were found for '%s'N)ry r/ �lowerrW r )r< rL �prefixr� r' r' r+ r� � s
zSession.get_adapterc C s | j �� D ] }|�� q
dS )z+Closes all adapters and as such the sessionN)ry �valuesrV )r< r* r' r' r+ rV � s z
Session.closec s> || j � < � fdd�| j D �}|D ]}| j �|�| j |< q"dS )zwRegisters a connection adapter to a prefix.
Adapters are sorted in descending order by prefix length.
c s g | ]}t |�t � �k r|�qS r' )rT )r( r) �r� r' r+ r, � s z!Session.mount..N)ry r[ )r< r� r� Z keys_to_mover1 r' r� r+ rz � s
z
Session.mountc s � fdd�� j D �}|S )Nc s i | ]}|t � |d ��qS r& )�getattr)r( �attrr{ r' r+ �
� s z(Session.__getstate__..)� __attrs__)r< �stater' r{ r+ � __getstate__� s zSession.__getstate__c C s" |� � D ]\}}t| ||� qd S r& )r/ �setattr)r< r� r� �valuer' r' r+ � __setstate__� s zSession.__setstate__)NNNNNNNTNNNNNN)NN)N)N)rr rs rt �__doc__r� r| r} r r� rf r4 r� r� r� r� r� r� r` r� r� rV rz r� r� r' r' r' r+ ru T s` �7) �
G
I ru c C s t � S )aZ
Returns a :class:`Session` for context-management.
.. deprecated:: 1.0.0
This method has been deprecated since version 1.0.0 and is only kept for
backwards compatibility. New code should use :class:`~requests.sessions.Session`
to create a session. This may be removed at a future date.
:rtype: Session
)ru r' r' r' r+ �session� s r� )>r� r� �sys�timeZdatetimer rv r �compatr r r r r r
r] r r r
r Zmodelsr r r rw r r Z_internal_utilsr Zutilsr r r �
exceptionsr r r r Z
structuresr ry r r r r r! r" r# Z status_codesr$ r% �platform� perf_counterr� �AttributeErrorZclockr2 r5 �objectr6 ru r� r' r' r' r+ � s@
v %