a dbq@sdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z mZddlmZmZmZmZddlmZmZmZdd lmZmZdd lmZdd lmZmZdd l 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.ddl/m0Z0ddlm1Z1ej2dkrHz ej3Z4Wne5yDej6Z4Yn0nejZ4e fddZ7e fddZ8Gddde9Z:Gddde:Z;ddZIz!merge_setting..) isinstancer rupdateitems)request_settingsession_setting dict_classmerged_setting none_keyskeyr'r'r+ merge_setting1s r7cCs@|dus|dgkr|S|dus0|dgkr4|St|||S)zProperly merges both requests and session hooks. This is necessary because when request_hooks == {'response': []}, the merge breaks Session hooks entirely. Nresponse)getr7) request_hooks session_hooksr3r'r'r+ merge_hooksPs r<c@s>eZdZddZddZddd Zd d Zd d ZddZdS)SessionRedirectMixincCs,|jr(|jd}tr|d}t|dSdS)z7Receives a Response. Returns a redirect URI or ``None``locationlatin1utf8N) is_redirectheadersrencoder)selfrespr>r'r'r+get_redirect_targetas    z(SessionRedirectMixin.get_redirect_targetcCsdt|}t|}|j|jkr dS|jdkrL|jdvrL|jdkrL|jdvrLdS|j|jkpb|j|jkS)zFDecide whether Authorization header should be removed when redirectingThttp)PNhttps)iNF)r hostnameschemeport)rDold_urlnew_url old_parsed new_parsedr'r'r+should_strip_authvs z&SessionRedirectMixin.should_strip_authFNTc  ksVg} ||} t|jj} | rR|} | || dd|_z |jWn&tt t fyp|j j ddYn0t |j|jkrtd|j|d|| drt|j}dt|j| f} t| }|jd kr| r|j| d }n |jr|j} |} |jst|jt| } nt| } t| | _|| ||jtjtjfvrjd }|D]}| j|dqNd| _ | j}z |d =Wnt!yYn0t"| j#||j t$| j#|j%| &| j#|'| |}|(| || j)duod |vpd|v}|rt*| | }|r|Vq|j+|f|||||dd| }t"|j%| |j ||} |VqdS)zBReceives a Response. Returns a generator of Responses or Requests.rNF)decode_contentzExceeded %s redirects.)r8z//z%s:%s)fragment)Content-Lengthz Content-TypeTransfer-EncodingCookierUrV)streamtimeoutverifycertproxiesallow_redirects),rFr urlrTcopyappendhistorycontentrr RuntimeErrorrawreadlen max_redirectsrclose startswithrrK_replacegeturlnetlocrrrebuild_method status_coder#temporary_redirectpermanent_redirectrBpopbodyKeyErrorr _cookiesrcookiesprepare_cookiesrebuild_proxies rebuild_auth_body_positionr"send)rDrEreqrXrYrZr[r\yield_requestsadapter_kwargshistr^previous_fragmentprepared_request parsed_rurlparsedpurged_headersheaderrB rewindabler'r'r+resolve_redirectss               z&SessionRedirectMixin.resolve_redirectscCsR|j}|j}d|vr*||jj|r*|d=|jr8t|nd}|durN||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. AuthorizationN)rBr^rQrequest trust_envr prepare_auth)rDrr8rBr^new_authr'r'r+rxs z!SessionRedirectMixin.rebuild_authc Cs|dur |ni}|j}|j}t|j}|}|d}t||d}|jr||s|t||d} | || d} | r|| || d|vr|d=zt ||\} } Wnt yd\} } Yn0| r| rt | | |d<|S)aThis 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_proxyrallzProxy-Authorization)NN) rBr^r rKr_r9r rr setdefaultr!rsr) rDrr\rBr^rK new_proxiesr bypass_proxyenviron_proxiesproxyusernamepasswordr'r'r+rw s*        z$SessionRedirectMixin.rebuild_proxiescCsX|j}|jtjkr|dkrd}|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)methodrnr# see_otherfoundmoved)rDrr8rr'r'r+rm4sz#SessionRedirectMixin.rebuild_method)FNTNNF) __name__ __module__ __qualname__rFrQrrxrwrmr'r'r'r+r=_s q)r=c@seZdZdZgdZddZddZddZd d Zd+d dZ ddZ ddZ ddZ d,ddZ d-ddZd.ddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd S)/SessionaA 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') ) rBruauthr\hooksparamsrZr[ZprefetchadaptersrXrrgcCsrt|_d|_i|_t|_i|_d|_d|_d|_ t |_ d|_ t i|_t|_|dt|dtdS)NFTzhttps://zhttp://)rrBrr\rrrrXrZr[rrgrr rurrmountrrDr'r'r+__init__ds zSession.__init__cCs|Sr&r'rr'r'r+ __enter__szSession.__enter__cGs |dSr&)rh)rDargsr'r'r+__exit__szSession.__exit__c Cs|jpi}t|tjst|}ttt|j|}|j}|jrR|sR|jsRt |j }t }|j |j |j |j|j|jt|j|jtdt|j|jt||j|t|j|jd |S)aConstructs 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 )r3) rr^filesdatajsonrBrrrur)rur.r CookieJarr rr rrrr^rpreparerupperrrrr7rBrrr<r)rDrrumerged_cookiesrpr'r'r+prepare_requests.       zSession.prepare_requestNTc Csxt|||||pi||pi||| d }||}| p8i} ||j| | ||}| | d}|||j|fi|}|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 ) rr^rBrrrrrrur)rYr])rrrmerge_environment_settingsr^r/rz)rDrr^rrrBrurrrYr]r\rrXrZr[rr{prepsettings send_kwargsrEr'r'r+rs.)   zSession.requestcKs |dd|jd|fi|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 r]TrrrrDr^kwargsr'r'r+r9s z Session.getcKs |dd|jd|fi|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 r]TOPTIONSrrr'r'r+optionss zSession.optionscKs |dd|jd|fi|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 r]Frrrr'r'r+head&s z Session.headcKs|jd|f||d|S)aSends 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 r)rrr)rDr^rrrr'r'r+post1s z Session.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 PUTrrrDr^rrr'r'r+put>s z Session.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 PATCHrrrr'r'r+patchJs z Session.patchcKs|jd|fi|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 DELETErrr'r'r+deleteVszSession.deletec Ks|d|j|d|j|d|j|d|jt|trJtd|dd}| d}|j }|j |j d}t }|j|fi|}t |} t| d |_td ||fi|}|jr|jD]} t|j| j| jqt|j||j|j||fi|} |rd d | Dng} | r6| d || }| |_|stz"t|j||fddi||_WntyrYn0|s|j|S)zISend a given PreparedRequest. :rtype: requests.Response rXrZr[r\z#You can only send PreparedRequests.r]T)r^)secondsr8cSsg|]}|qSr'r')r(rEr'r'r+r,r-z Session.send..rr|)rrXrZr[r\r.r ValueErrorrqr9r get_adapterr^preferred_clockrzrelapsedrrar rurrdrinsertnext_next StopIterationrb) rDrrr]rXradapterstartrrrEgenrar'r'r+rz`sB       "z Session.sendc Cs|jrn|dur|dnd}t||d}|D]\}} ||| q0|dusV|durntjdpltjd}t||j}t||j }t||j }t||j }||||dS)z^ Check the environment and merge it with some settings. :rtype: dict NrrTREQUESTS_CA_BUNDLECURL_CA_BUNDLE)rZr\rXr[) rr9rr0rosenvironr7r\rXrZr[) rDr^r\rXrZr[r env_proxiesr)r*r'r'r+rs        z"Session.merge_environment_settingscCs>|jD]"\}}||r |Sq 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)rr0lowerrir)rDr^prefixrr'r'r+rs zSession.get_adaptercCs|jD] }|q dS)z+Closes all adapters and as such the sessionN)rvaluesrh)rDr*r'r'r+rhsz Session.closecs>||j<fdd|jD}|D]}|j||j|<q"dS)zwRegisters a connection adapter to a prefix. Adapters are sorted in descending order by prefix length. cs g|]}t|tkr|qSr')rf)r(r)rr'r+r,r-z!Session.mount..N)rrq)rDrr keys_to_mover6r'rr+rs z Session.mountcsfddjD}|S)Ncsi|]}|t|dqSr&)getattr)r(attrrr'r+ r-z(Session.__getstate__..) __attrs__)rDstater'rr+ __getstate__szSession.__getstate__cCs"|D]\}}t|||qdSr&)r0setattr)rDrrvaluer'r'r+ __setstate__szSession.__setstate__)NNNNNNNTNNNNNN)NN)N)N)rrr__doc__rrrrrrr9rrrrrrrzrrrhrrrr'r'r'r+rKs07) G    I rcCstS)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 )rr'r'r'r+sessions r)=rrsystimedatetimerrrcompatrrrrr r rur r r rmodelsrrrrrr_internal_utilsrutilsrr exceptionsrrrr structuresrrrrrrr r!r" status_codesr#r$platform perf_counterrAttributeErrorZclockr7r<objectr=rrr'r'r'r+s@             m%