3
P\y @ s> d Z ddlZddlmZmZ ddlmZ G dd dejZ dS )z*The module containing HTTPProxyDigestAuth. N)cookiesutils )_digest_auth_compatc sf e Zd ZdZejdejdZ fddZe dd Z
e
j dd Z
fd d
Z d d Z
d
d Z ZS )HTTPProxyDigestAuthaO HTTP digest authentication between proxy
:param stale_rejects: The number of rejects indicate that:
the client may wish to simply retry the request
with a new encrypted response, without reprompting the user for a
new username and password. i.e., retry build_digest_header
:type stale_rejects: int
zdigest )flagsc s$ t t| j|| d| _| j d S )Nr )superr __init__
stale_rejectsinit_per_thread_state)selfargskwargs) __class__ '/usr/lib/python3.6/http_proxy_digest.pyr s z HTTPProxyDigestAuth.__init__c C s t | dd }|d kr| jS |jS )N
_thread_local)getattr_stale_rejectsr
)r thread_localr r r r
s z!HTTPProxyDigestAuth.stale_rejectsc C s& t | dd }|d kr || _n||_d S )Nr )r r r
)r valuer r r r r
" s c s, yt t| j W n tk
r& Y nX d S )N)r r r AttributeError)r )r r r r * s z)HTTPProxyDigestAuth.init_per_thread_statec K s, |j dko| jdk r$|jjd}|dkr4tdn| jj|sD|S tj| jj d|dd| _
d |j jkrd
| j
kr| j
d
j d kr| jd7 _n| j
d
j d krtd
|j
|j |j j }tj|j|j |j |j|j | j|j|j|jd < |jj|f|}|jj| ||_ |S |S dS )zHandle HTTP 407 only once, otherwise give up
:param r: current response
:returns: responses, along with the new response
i zproxy-authenticateNzRproxy server violated RFC 7235:407 response MUST contain header proxy-authenticate r )countzProxy-AuthorizationZstaletrueZfalsezUser or password is invalid) Z status_coder
headersgetIOError_patmatchr Zparse_dict_headersubZchalZrequestlowerZcontentclosecopyr Zextract_cookies_to_jarZ_cookiesrawZprepare_cookiesbuild_digest_headermethodurlZ
connectionsendhistoryappend)r rr Zs_authZprepZ_rr r r
handle_4071 s6
z HTTPProxyDigestAuth.handle_407c C s6 | j | jr$| j|j|j|jd< |jd| j |S )NzProxy-AuthorizationZresponse)r Z
last_noncer& r' r( r Z
register_hookr- )r r, r r r __call___ s z HTTPProxyDigestAuth.__call__)__name__
__module__ __qualname____doc__recompile
IGNORECASEr r propertyr
setterr r- r.
__classcell__r r )r r r
s .r )
r2 r3 Zrequestsr r r r ZauthZHTTPDigestAuthr r r r r s