3
|`hUs @ 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 ej3dkrNy
ej4Z5W n e6k
rJ ej7Z5Y nX nejZ5e fddZ8e fddZ9G dd de:Z;G dd de;ZI s z!merge_setting..)
isinstancer
r updateitems)Zrequest_settingZsession_setting
dict_classZmerged_settingZ none_keyskeyr& r& r*
merge_setting1 s
r1 c C s@ |dks|j dg kr| S | dks0| j dg kr4|S t| ||S )zProperly merges both requests and session hooks.
This is necessary because when request_hooks == {'response': []}, the
merge breaks Session hooks entirely.
Nresponse)getr1 )Z
request_hooksZ
session_hooksr/ r& r& r* merge_hooksP s
r4 c @ s> e Zd Zdd Zdd Zddd Zd
d Zd d
Zdd ZdS )SessionRedirectMixinc C s, |j r(|jd }tr |jd}t|dS dS )z7Receives a Response. Returns a redirect URI or ``None``locationlatin1utf8N)Z is_redirectheadersr encoder )selfrespr6 r& r& r* get_redirect_targeta s
z(SessionRedirectMixin.get_redirect_targetc C s t |}t |}|j|jkr dS |jdkrL|jdkrL|jdkrL|jd krLdS |j|jk}|j|jk}tj|jddf}| r|j|kr|j|krdS |p|S )
zFDecide whether Authorization header should be removed when redirectingTZhttpP Nhttps F)r> N)r@ N)r ZhostnameschemeZportr r3 )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 sd g }
| j |} t|jj} xB| r^|j }
|
j| |
dd |_y
|j W n( tt t
fk
rv |j j dd Y nX t
|j| jkrtd| j |d|j | jdrt|j}dt|j| f } t| }|jd kr| r|j| d
}n |jr|j} |j } |jst|jt| } nt| } t| |
_| j|
| |jtj tj fkrtd}x |D ]}|
j j |d qVW d|
_ |
j }y
|d= W n t!k
r Y nX t"|
j#||j t$|
j#| j% |
j&|
j# | j'|
|}| j(|
| |
j)dk od |kpd
|k}|r
t*|
|
}|r |V q | j+|f|||||dd| }t"| j%|
|j | j |} |V q W dS )zBReceives a Response. Returns a generator of Responses or Requests.r NF)Zdecode_contentzExceeded %s redirects.)r2 z//z%s:%s )fragmentContent-Length Content-TypeTransfer-EncodingZCookie)streamtimeoutverifycertproxiesallow_redirects)rE rF rG ),r= r urlrD copyappendhistorycontentr r RuntimeErrorrawreadlen
max_redirectsr close
startswithr rA _replaceZgeturlZnetlocr r rebuild_method status_coder$ Ztemporary_redirectZpermanent_redirectr9 popZbodyKeyErrorr Z_cookiesr cookiesZprepare_cookiesrebuild_proxies rebuild_authZ_body_positionr# send)r; r< reqrH rI rJ rK rL yield_requestsZadapter_kwargsZhistrN Zprevious_fragmentprepared_requestZ parsed_rurlZparsedZpurged_headersheaderr9 Z
rewindabler& r& r* resolve_redirects s|
z&SessionRedirectMixin.resolve_redirectsc C sR |j }|j}d|kr*| j|jj|r*|d= | jr8t|nd}|dk rN|j| dS )zWhen 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)r9 rN rB request trust_envr Z prepare_auth)r; re r2 r9 rN Znew_authr& r& r* ra s
z!SessionRedirectMixin.rebuild_authc
C s |dk r |ni }|j }|j}t|j}|j }|jd}t||d}| jr~| r~t||d} | j|| jd}
|
r~|j ||
d|kr|d= yt
|| \} } W n t k
r d\} } Y nX |j d r| r| rt
| | |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
Nno_proxy)rj allzProxy-Authorizationr? )NN)r9 rN r rA rO r3 r! ri r
setdefaultr" r^ rY r )
r; re rL r9 rN rA Z new_proxiesrj Z bypass_proxyZenviron_proxiesproxyZusernameZpasswordr& r& r* r` s*
z$SessionRedirectMixin.rebuild_proxiesc C sX |j }|jtjkr |dkr d}|jtjkr6|dkr6d}|jtjkrN|dkrNd}||_ dS )zWhen being redirected we may want to change the method of the request
based on certain specs or browser behavior.
HEADGETPOSTN)methodr\ r$ Z see_otherfoundZmoved)r; re r2 rq r& r& r* r[ ? s z#SessionRedirectMixin.rebuild_method)FNTNNF) __name__
__module__ __qualname__r= rB rg ra r` r[ r& r& r& r* r5 _ s
p+r5 c @ s e Zd ZdZdddddddd d
d d d
dg
Zdd Zdd Zdd Zdd Zd7ddZ 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')
r9 r_ authrL hooksparamsrJ rK ZprefetchadaptersrH ri rW c C sr t | _d | _i | _t | _i | _d| _d| _d | _ t
| _ d| _ t
i | _t | _| jdt | jdt d S )NFTzhttps://zhttp://)r r9 rw rL r rx ry rH rJ rK r rW ri r r_ r rz mountr )r; r& r& r* __init__o s
zSession.__init__c C s | S )Nr& )r; r& r& r* __enter__ s zSession.__enter__c G s | j d S )N)rX )r; argsr& r& r* __exit__ s zSession.__exit__c
C s |j pi }t|tjs t|}ttt | j |}|j}| jrV| rV| j rVt |j
}t }|j |j
j |j
|j|j|jt|j| jtdt|j| jt|| j|t|j| jd
|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/ )
rq rN filesdatajsonr9 ry rw r_ rx )r_ r, r Z CookieJarr r r
rw ri r rN r Zpreparerq upperr r r r1 r9 r ry r4 rx )r; rh r_ Zmerged_cookiesrw pr& r& r* prepare_request s*
zSession.prepare_requestNTc C st t |j ||||pi ||p i ||| d
}| j|}| p8i } | j|j| |
||}| |
d}|j| | 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
)
rq rN r9 r r r ry rw r_ rx )rI rM )r r r merge_environment_settingsrN r- rb )r; rq rN ry r r9 r_ r rw rI rM rL rx rH rJ rK r rc ZprepZsettingsZ send_kwargsr< r& r& r* rh s( )
zSession.requestc K s |j dd | jd|f|S )zSends 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
rM Tro )rl rh )r; rN kwargsr& r& r* r3 s z Session.getc K s |j dd | jd|f|S )zSends 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
rM TZOPTIONS)rl rh )r; rN r r& r& r* options& s zSession.optionsc K s |j dd | jd|f|S )zSends 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
rM Frn )rl rh )r; rN r r& r& r* head1 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
rp )r r )rh )r; rN 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 )rh )r; rN r r r& r& r* putI 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 )rh )r; rN r r r& r& r* patchU s
z
Session.patchc K s | j d|f|S )zSends 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; rN r r& r& r* deletea s zSession.deletec
K s~ |j d| j |j d| j |j d| j |j d| j t|trJtd|jdd}|j d}|j
}| j |j d}t
}|j|f|}t
| } t| d |_td
||f|}|jrx |jD ]}
t| j|
j|
j qW t| j||j | j||f|} |r
d d | D ng } | r.| jd
| | j }| |_|sny"t| j||fddi||_W n tk
rl Y nX |sz|j |S )zISend a given PreparedRequest.
:rtype: requests.Response
rH rJ rK rL z#You can only send PreparedRequests.rM T)rN )Zsecondsr2 c S s g | ]}|qS r& r& )r' r< r& r& r* r+ s z Session.send..r rd ) rl rH rJ rK rL r, r
ValueErrorr] r3 rx get_adapterrN preferred_clockrb r elapsedr rQ r r_ rh rT rg insertnextZ_next
StopIterationrR )
r; rh r rM rH rx adapterstartrr r< genrQ r& r& r* rb k sB
"z Session.sendc
C s | j rr|dk r|jdnd}t||d}x |j D ]\}} |j|| q2W |dksZ|dkrrtjjdpptjjd}t|| j}t|| j }t|| j
}t|| j }||||dS )z^
Check the environment and merge it with some settings.
:rtype: dict
Nrj )rj TZREQUESTS_CA_BUNDLEZCURL_CA_BUNDLE)rJ rL rH rK ) ri r3 r r. rl osenvironr1 rL rH rJ rK )
r; rN rL rH rJ rK rj Z env_proxiesr( r) r& r& r* r s z"Session.merge_environment_settingsc C s> x,| j j D ] \}}|j j|j r |S q W 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)rz r. lowerrY r )r; rN prefixr r& r& r* r s zSession.get_adapterc C s x| j j D ] }|j q W dS )z+Closes all adapters and as such the sessionN)rz valuesrX )r; r) r& r& r* rX s z
Session.closec sB || j < fdd| j D }x |D ]}| j j|| j |< q$W 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& )rV )r' r( )r r& r* r+ s z!Session.mount..N)rz r] )r; r r Z keys_to_mover0 r& )r r* r{ s
z
Session.mountc s fdd j D }|S )Nc s i | ]}t |d |qS )N)getattr)r' attr)r; r& r*
s z(Session.__getstate__..) __attrs__)r; stater& )r; r* __getstate__ s zSession.__getstate__c C s& x |j D ]\}}t| || q
W d S )N)r. setattr)r; r r valuer& r& r* __setstate__ s zSession.__setstate__)NNNNNNNTNNNNNN)NN)N)N)rs rt ru __doc__r r| r} r r rh r3 r r r r r r rb r r rX r{ r r r& r& r& r* rv V s2
7)
D
I rv 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
)rv r& r& r& r* session s r )>r r sysZtimeZdatetimer rw r compatr r r r r r
r_ r r r
r Zmodelsr r r rx r r Z_internal_utilsr Zutilsr r r
exceptionsr r r r Z
structuresr rz r r r r r! r" r# Z status_codesr$ r% platformZ perf_counterr AttributeErrorZclockr1 r4 objectr5 rv r r& r& r& r* s@
x %