3 |`�hUs�@s�dZddlZddlZddlZddlmZddlmZddlm Z m Z mZmZm Z mZddlmZmZmZmZddlmZmZmZdd lmZmZdd lmZddlmZmZm Z ddl!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'ddl(m)Z)ddlm*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ddl0m1Z1ddlm2Z2ej3dk�rNy ej4Z5Wne6k �rJej7Z5YnXnejZ5efdd�Z8efdd�Z9Gdd�de:�Z;Gdd�de;�ZIsz!merge_setting..)� isinstancer r�update�items)Zrequest_settingZsession_setting� dict_classZmerged_settingZ none_keys�keyr&r&r*� merge_setting1s r1cCs@|dks|jd�gkr|S|dks0|jd�gkr4|St|||�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)�getr1)Z request_hooksZ session_hooksr/r&r&r*�merge_hooksPs r4c@s>eZdZdd�Zdd�Zddd �Zd d�Zdd �Zdd�ZdS)�SessionRedirectMixincCs,|jr(|jd}tr|jd�}t|d�SdS)z7Receives a Response. Returns a redirect URI or ``None``�location�latin1�utf8N)Zis_redirect�headersr�encoder)�self�respr6r&r&r*�get_redirect_targetas z(SessionRedirectMixin.get_redirect_targetcCs�t|�}t|�}|j|jkr dS|jdkrL|jdkrL|jdkrL|jd krLdS|j|jk}|j|jk}tj|jd�df}|r�|j|kr�|j|kr�dS|p�|S) zFDecide whether Authorization header should be removed when redirectingTZhttp�PN�https�F)r>N)r@N)r Zhostname�schemeZportrr3)r;Zold_urlZnew_urlZ old_parsedZ new_parsedZchanged_portZchanged_schemeZdefault_portr&r&r*�should_strip_authvs z&SessionRedirectMixin.should_strip_authFNTc ksdg} |j|�}t|j�j}�xB|�r^|j�} | j|�| dd�|_y |jWn(tt t fk rv|jjdd�YnXt |j�|jkr�td|j|d��|j�|jd�r�t|j�}dt|j�|f}t|�}|jd kr�|r�|j|d �}n|jr�|j}|j�}|j�st|jt|��}nt|�}t|�| _|j| |�|jtjtjfk�rtd}x|D]}| jj|d��qVWd| _ | j}y |d=Wnt!k �r�YnXt"| j#||j�t$| j#|j%�| j&| j#�|j'| |�}|j(| |�| j)dk �o�d|k�p�d |k}|�r t*| �| }|�r|Vq|j+|f|||||dd�| ��}t"|j%| |j�|j|�}|VqWdS)zBReceives a Response. Returns a generator of Responses or Requests.rNF)Zdecode_contentzExceeded %s redirects.)r2z//z%s:%s�)�fragment�Content-Length�Content-Type�Transfer-EncodingZCookie)�stream�timeout�verify�cert�proxies�allow_redirects)rErFrG),r=r �urlrD�copy�append�history�contentrr�RuntimeError�raw�read�len� max_redirectsr�close� startswithrrA�_replaceZgeturlZnetlocrr�rebuild_method�status_coder$Ztemporary_redirectZpermanent_redirectr9�popZbody�KeyErrorrZ_cookiesr�cookiesZprepare_cookies�rebuild_proxies�rebuild_authZ_body_positionr#�send)r;r<�reqrHrIrJrKrL�yield_requestsZadapter_kwargsZhistrNZprevious_fragment�prepared_requestZparsed_rurlZparsedZpurged_headers�headerr9Z rewindabler&r&r*�resolve_redirects�s| z&SessionRedirectMixin.resolve_redirectscCsR|j}|j}d|kr*|j|jj|�r*|d=|jr8t|�nd}|dk rN|j|�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)r9rNrB�request� trust_envr Zprepare_auth)r;rer2r9rNZnew_authr&r&r*ras z!SessionRedirectMixin.rebuild_authc Cs�|dk r|ni}|j}|j}t|�j}|j�}|jd�}t||d�}|jr~|r~t||d�} | j|| jd��} | r~|j || �d|kr�|d=yt ||�\}}Wntk r�d\}}YnX|jd�r�|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)rj�allzProxy-Authorizationr?)NN)r9rNr rArOr3r!rir� setdefaultr"r^rYr) r;rerLr9rNrAZnew_proxiesrjZbypass_proxyZenviron_proxies�proxyZusernameZpasswordr&r&r*r`s* z$SessionRedirectMixin.rebuild_proxiescCsX|j}|jtjkr|dkrd}|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)�methodr\r$Z see_other�foundZmoved)r;rer2rqr&r&r*r[?sz#SessionRedirectMixin.rebuild_method)FNTNNF) �__name__� __module__�__qualname__r=rBrgrar`r[r&r&r&r*r5_s p+r5c@s�eZdZdZdddddddd d ddd dg Zdd�Zdd�Zdd�Zdd�Zd7dd�Z dd�Z dd�Zdd �Zd8d!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') r9r_�authrL�hooks�paramsrJrKZprefetch�adaptersrHrirWcCsrt�|_d|_i|_t�|_i|_d|_d|_d|_ t |_d|_t i�|_t�|_|jdt��|jdt��dS)NFTzhttps://zhttp://)rr9rwrLrrxryrHrJrKrrWrirr_rrz�mountr)r;r&r&r*�__init__os zSession.__init__cCs|S)Nr&)r;r&r&r*� __enter__�szSession.__enter__cGs|j�dS)N)rX)r;�argsr&r&r*�__exit__�szSession.__exit__c Cs�|jpi}t|tj�st|�}ttt�|j�|�}|j}|jrV|rV|jrVt |j �}t�}|j|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 )r/) rqrN�files�data�jsonr9ryrwr_rx)r_r,rZ CookieJarrrr rwrir rNrZpreparerq�upperr�r�r�r1r9rryr4rx)r;rhr_Zmerged_cookiesrw�pr&r&r*�prepare_request�s* zSession.prepare_requestNTcCstt|j�||||pi||pi|||d� }|j|�}|p8i}|j|j|| ||�}| | d�}|j|�|j|f|�}|S)aConstructs 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 ) rqrNr9r�r�r�ryrwr_rx)rIrM)rr�r��merge_environment_settingsrNr-rb)r;rqrNryr�r9r_r�rwrIrMrLrxrHrJrKr�rcZprepZsettingsZsend_kwargsr<r&r&r*rh�s() zSession.requestcKs|jdd�|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 rMTro)rlrh)r;rN�kwargsr&r&r*r3szSession.getcKs|jdd�|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 rMTZOPTIONS)rlrh)r;rNr�r&r&r*�options&szSession.optionscKs|jdd�|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 rMFrn)rlrh)r;rNr�r&r&r*�head1szSession.headcKs|jd|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 rp)r�r�)rh)r;rNr�r�r�r&r&r*�post<szSession.postcKs|jd|fd|i|��S)auSends 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�)rh)r;rNr�r�r&r&r*�putIs zSession.putcKs|jd|fd|i|��S)awSends 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�)rh)r;rNr�r�r&r&r*�patchUs z Session.patchcKs|jd|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 ZDELETE)rh)r;rNr�r&r&r*�deleteaszSession.deletec Ks~|jd|j�|jd|j�|jd|j�|jd|j�t|t�rJtd��|jdd�}|j d�}|j }|j|jd�}t �}|j|f|�}t �|} t| d �|_td ||f|�}|jr�x |jD]} t|j| j| j�q�Wt|j||j�|j||f|�}|�r dd�|D�ng}|�r.|jd |�|j�}||_|�sny"t|j||fddi|��|_Wntk �rlYnX|�sz|j|S)zISend a given PreparedRequest. :rtype: requests.Response rHrJrKrLz#You can only send PreparedRequests.rMT)rN)Zsecondsr2cSsg|]}|�qSr&r&)r'r<r&r&r*r+�sz Session.send..rrd)rlrHrJrKrLr,r� ValueErrorr]r3rx�get_adapterrN�preferred_clockrbr�elapsedrrQrr_rhrTrg�insert�nextZ_next� StopIterationrR) r;rhr�rMrHrx�adapter�start�rr�r<�genrQr&r&r*rbksB "zSession.sendc Cs�|jrr|dk r|jd�nd}t||d�}x |j�D]\}} |j|| �q2W|dksZ|dkrrtjjd�pptjjd�}t||j�}t||j �}t||j �}t||j�}||||d�S)z^ Check the environment and merge it with some settings. :rtype: dict Nrj)rjTZREQUESTS_CA_BUNDLEZCURL_CA_BUNDLE)rJrLrHrK)rir3rr.rl�os�environr1rLrHrJrK) r;rNrLrHrJrKrjZenv_proxiesr(r)r&r&r*r��sz"Session.merge_environment_settingscCs>x,|jj�D]\}}|j�j|j��r|SqWtd|��dS)z~ Returns the appropriate connection adapter for the given URL. :rtype: requests.adapters.BaseAdapter z*No connection adapters were found for '%s'N)rzr.�lowerrYr)r;rN�prefixr�r&r&r*r��szSession.get_adaptercCs x|jj�D]}|j�qWdS)z+Closes all adapters and as such the sessionN)rz�valuesrX)r;r)r&r&r*rX�sz Session.closecsB||j�<�fdd�|jD�}x|D]}|jj|�|j|<q$WdS)zwRegisters a connection adapter to a prefix. Adapters are sorted in descending order by prefix length. cs g|]}t|�t��kr|�qSr&)rV)r'r()r�r&r*r+�sz!Session.mount..N)rzr])r;r�r�Zkeys_to_mover0r&)r�r*r{�s z Session.mountcs�fdd��jD�}|S)Ncsi|]}t�|d�|�qS)N)�getattr)r'�attr)r;r&r*� �sz(Session.__getstate__..)� __attrs__)r;�stater&)r;r*�__getstate__�szSession.__getstate__cCs&x |j�D]\}}t|||�q WdS)N)r.�setattr)r;r�r��valuer&r&r*�__setstate__�szSession.__setstate__)NNNNNNNTNNNNNN)NN)N)N)rsrtru�__doc__r�r|r}rr�rhr3r�r�r�r�r�r�rbr�r�rXr{r�r�r&r&r&r*rvVs2 7) D IrvcCst�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 )rvr&r&r&r*�session�sr�)>r�r��sysZtimeZdatetimerrwr�compatrrrrr r r_rrr rZmodelsrrrrxrrZ_internal_utilsrZutilsrrr� exceptionsrrrrZ structuresrrzrrrr r!r"r#Zstatus_codesr$r%�platformZperf_counterr��AttributeErrorZclockr1r4�objectr5rvr�r&r&r&r*� s@ x%