3
aj & @ 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 )urlparsestr
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 |tsLtjdj|td t|}t | tr`| jd} t |trt|jd}dtt dj
| |fj }|S )z Returns a Basic Auth string.zNon-string usernames will no longer be supported in Requests 3.0.0. Please convert the object you've passed in ({0!r}) to a string or bytes object in the near future to avoid problems.)categoryzNon-string passwords will no longer be supported in Requests 3.0.0. Please convert the object you've passed in ({0!r}) to a string or bytes object in the near future to avoid problems.latin1zBasic :)
isinstancer warningswarnformatDeprecationWarningr encoder r joinstrip)usernamepasswordZauthstr r /usr/lib/python3.6/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)selfrr 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 threadingZlocal
_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 )NinitT r )hasattrr/ r0
last_nonce nonce_countchalpos
num_401_calls)r r r r init_per_thread_stateu s z$HTTPDigestAuth.init_per_thread_statec sj | j jd }| j jd }| j jjd}| j jjd}| j jjd}d |dkrTd}n|j }|dksl|dkrzd d
} | n|d krd d
}
|
fdd} dkrdS d} t|}
|
jpd}|
jr|d|
j 7 }d| j|| jf }d||f } |} |}|| j j kr | j j
d7 _
nd| j _
d| j j
}t | j j
j d}||j d7 }|t
j j d7 }|tjd7 }tj|j dd }|dkr d|||f }|s| |d||f }n<|dksd|jdkrd|||d|f }| ||}ndS || j _ d | j||||f }|r(|d | 7 }|r:|d | 7 }| rL|d | 7 }|rb|d ||f 7 }d!| S )"z
:rtype: str
realmnonceqop algorithmopaqueNZMD5zMD5-SESSc S s" t | tr| jd} tj| j S )Nzutf-8)r
r r hashlibZmd5 hexdigest)xr r r md5_utf8 s
z4HTTPDigestAuth.build_digest_header.<locals>.md5_utf8ZSHAc S s" t | tr| jd} tj| j S )Nzutf-8)r
r r r> sha1r? )r@ r r r sha_utf8 s
z4HTTPDigestAuth.build_digest_header.<locals>.sha_utf8c s d| |f S )Nz%s:%sr )sd) hash_utf8r r <lambda> s z4HTTPDigestAuth.build_digest_header.<locals>.<lambda>/?z%s:%s:%sz%s:%sr z%08xzutf-8 Zauth,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)r/ r5 getupperr pathZqueryr r r3 r4 r r timeZctimeosurandomr> rB r? split)r methodurlr9 r: r; r< r= Z
_algorithmrA rC ZKDZentdigZp_parsedrO ZA1ZA2ZHA1ZHA2ZncvaluerD ZcnonceZrespdigZnoncebitbaser )rF r build_digest_header sr
z"HTTPDigestAuth.build_digest_headerc K s |j rd| j_dS )z)Reset num_401_calls counter on redirects.r N)Z is_redirectr/ r7 )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-authenticater1 Zdigest zdigest )flags)countr* ) Z status_coder/ r7 r6 Zrequestbodyseekr+ rM lowerrecompile
IGNORECASEr subr5 Zcontentclosecopyr Z_cookiesrawZprepare_cookiesrW rT rU Z
connectionsendhistoryappend)r r rX Zs_authZpatZprepZ_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* Zresponser )r8 r/ r3 rW rT rU r+ r] tellr6 AttributeErrorZ
register_hookrj rY r7 )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) $ s zHTTPDigestAuth.__ne__N) r r r! r" r$ r8 rW rY rj r r( r) r r r r r- l s
Z,r- )r" rQ r` rP r> r. r base64r compatr r r Zcookiesr Z_internal_utilsr Zutilsr Z CONTENT_TYPE_FORM_URLENCODEDZCONTENT_TYPE_MULTI_PARTr objectr r# r, r- r r r r <module> s$ ,