3
V'ícß' ã @ sÌ d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
m Z m Z ddl
mZ ddlmZ ddlmZ d Zd
Zd d ZG d
d deZG dd deZG dd deZG dd deZdS )z]
requests.auth
~~~~~~~~~~~~~
This module contains the authentication handlers for Requests.
é N)Ú b64encodeé )ÚurlparseÚstrÚ
basestring)Úextract_cookies_to_jar)Úto_native_string)Úparse_dict_headerz!application/x-www-form-urlencodedzmultipart/form-datac C s t | ts&tjdj| td t| } t |tsPtjdjt|td t|}t | trd| jd} t |trx|jd}dt t
dj | |fj }|S )z Returns a Basic Auth string.z·Non-string usernames will no longer be supported in Requests 3.0.0. Please convert the object you've passed in ({!r}) to a string or bytes object in the near future to avoid problems.)Úcategoryz·Non-string passwords will no longer be supported in Requests 3.0.0. Please convert the object you've passed in ({!r}) to a string or bytes object in the near future to avoid problems.Úlatin1zBasic ó :)
Ú
isinstancer ÚwarningsÚwarnÚformatÚDeprecationWarningr ÚtypeÚencoder r ÚjoinÚstrip)ÚusernameÚpasswordÚauthstr© r ú1/tmp/pip-build-gk9425m9/requests/requests/auth.pyÚ_basic_auth_str s&
r c @ s e Zd ZdZdd ZdS )ÚAuthBasez4Base class that all auth implementations derive fromc C s t dd S )Nz Auth hooks must be callable.)ÚNotImplementedError)ÚselfÚrr r r Ú__call__K s zAuthBase.__call__N)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r r r r r H s r c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) Ú
HTTPBasicAuthz?Attaches HTTP Basic Authentication to the given Request object.c C s || _ || _d S )N)r r )r r r r r r Ú__init__R s zHTTPBasicAuth.__init__c C s( t | jt|dd k| jt|dd kgS )Nr r )Úallr Úgetattrr )r Úotherr r r Ú__eq__V s zHTTPBasicAuth.__eq__c C s
| |k S )Nr )r r) r r r Ú__ne__\ s zHTTPBasicAuth.__ne__c C s t | j| j|jd< |S )NÚ
Authorization)r r r Úheaders)r r r r r r _ s zHTTPBasicAuth.__call__N)r! r" r# r$ r& r* r+ r r r r r r% O s
r% c @ s e Zd ZdZdd ZdS )Ú
HTTPProxyAuthz=Attaches HTTP Proxy Authentication to a given Request object.c C s t | j| j|jd< |S )NzProxy-Authorization)r r r r- )r r r r r r g s zHTTPProxyAuth.__call__N)r! r" r# r$ r r r r r r. d s r. c @ sP e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dS )ÚHTTPDigestAuthz@Attaches HTTP Digest Authentication to the given Request object.c C s || _ || _tj | _d S )N)r r Ú threadingÚlocalÚ
_thread_local)r r r r r r r& o s zHTTPDigestAuth.__init__c C s@ t | jds<d| j_d| j_d| j_i | j_d | j_d | j_d S )NÚinitTÚ r )Úhasattrr2 r3 Ú
last_nonceÚ nonce_countÚchalÚposÚ
num_401_calls)r r r r Úinit_per_thread_stateu s z$HTTPDigestAuth.init_per_thread_statec s | j jd }| j jd }| j jjd}| j jjd}| j jjd}d |dkrTd}n|j }|dksl|dkrzd d
} | n@|d krd d
}
|
n*|dkr¦dd } | n|dkrºdd } | fdd}
dkrÒdS d}t|}|jpæd}|jrü|d|j 7 }d| j|| jf }d||f } |} |}|| j j krJ| j j
d7 _
nd| j _
d| j j
}t | j j
j d }||j d 7 }|t
j j d 7 }|tjd 7 }tj|j dd
}|dkrÐ d|||f }|sê|
|d||f }n<|d ksd |jd kr"d!|||d |f }|
||}ndS || j _ d"| j||||f }|rT|d#| 7 }|rf|d$| 7 }|rx|d%| 7 }|r|d&||f 7 }d'| S )(z
:rtype: str
ÚrealmÚnonceÚqopÚ algorithmÚopaqueNÚMD5zMD5-SESSc S s" t | tr| jd} tj| j S )Nzutf-8)r
r r ÚhashlibÚmd5Ú hexdigest)Úxr r r Úmd5_utf8 s
z4HTTPDigestAuth.build_digest_header.<locals>.md5_utf8ÚSHAc S s" t | tr| jd} tj| j S )Nzutf-8)r
r r rB Úsha1rD )rE r r r Úsha_utf8 s
z4HTTPDigestAuth.build_digest_header.<locals>.sha_utf8zSHA-256c S s" t | tr| jd} tj| j S )Nzutf-8)r
r r rB Úsha256rD )rE r r r Ú sha256_utf8 s
z7HTTPDigestAuth.build_digest_header.<locals>.sha256_utf8zSHA-512c S s" t | tr| jd} tj| j S )Nzutf-8)r
r r rB Úsha512rD )rE r r r Ú sha512_utf8£ s
z7HTTPDigestAuth.build_digest_header.<locals>.sha512_utf8c s d| |f S )Nz%s:%sr )ÚsÚd)Ú hash_utf8r r Ú<lambda>© s z4HTTPDigestAuth.build_digest_header.<locals>.<lambda>ú/ú?z%s:%s:%sz%s:%sr z%08xzutf-8é é Úauthú,z%s:%s:%s:%s:%sz>username="%s", realm="%s", nonce="%s", uri="%s", response="%s"z
, opaque="%s"z, algorithm="%s"z
, digest="%s"z , qop="auth", nc=%s, cnonce="%s"z Digest %s)r2 r8 ÚgetÚupperr ÚpathÚqueryr r r6 r7 r r ÚtimeÚctimeÚosÚurandomrB rH rD Úsplit)r ÚmethodÚurlr< r= r> r? r@ Ú
_algorithmrF rI rK rM ÚKDÚentdigÚp_parsedrZ ÚA1ÚA2ÚHA1ÚHA2ÚncvaluerN ÚcnonceÚrespdigÚnoncebitÚbaser )rP r Úbuild_digest_header s~
z"HTTPDigestAuth.build_digest_headerc K s |j rd| j_dS )z)Reset num_401_calls counter on redirects.r N)Ú is_redirectr2 r: )r r Úkwargsr r r Úhandle_redirectå s z HTTPDigestAuth.handle_redirectc K s" d|j kodk n s&d| j_|S | jjdk rD|jjj| jj |jjdd}d|j koh| jjdk r| j jd7 _t
j d t
j d
}t
|jd|dd | j_|j |j |jj }t|j|j|j |j|j | j|j|j|jd < |jj|f|}|j j | ||_|S d| j_|S )
zo
Takes the given response and tries digest-auth, if needed.
:rtype: requests.Response
i iô r Nzwww-authenticater4 Údigesté zdigest )Úflags)Úcountr, ) Ú status_coder2 r: r9 ÚrequestÚbodyÚseekr- rX ÚlowerÚreÚcompileÚ
IGNORECASEr Úsubr8 ÚcontentÚcloseÚcopyr Ú_cookiesÚrawÚprepare_cookiesrp ra rb Ú
connectionÚsendÚhistoryÚappend)r r rr Ús_authÚpatÚprepÚ_rr r r Ú
handle_401ê s.
zHTTPDigestAuth.handle_401c C s~ | j | jjr&| j|j|j|jd< y|jj | j_ W n t
k
rT d | j_ Y nX |j d| j |j d| j
d| j_|S )Nr, Úresponser )r; r2 r6 rp ra rb r- rz Útellr9 ÚAttributeErrorÚ
register_hookr rs r: )r r r r r r s zHTTPDigestAuth.__call__c C s( t | jt|dd k| jt|dd kgS )Nr r )r' r r( r )r r) r r r r* * s zHTTPDigestAuth.__eq__c C s
| |k S )Nr )r r) r r r r+ 0 s zHTTPDigestAuth.__ne__N) r! r" r# r$ r& r; rp rs r r r* r+ r r r r r/ l s
f,r/ )r$ r^ r} r\ rB r0 r Úbase64r Úcompatr r r Úcookiesr Ú_internal_utilsr Úutilsr Ú CONTENT_TYPE_FORM_URLENCODEDÚCONTENT_TYPE_MULTI_PARTr Úobjectr r% r. r/ r r r r Ú<module> s$ ,