← Back
Editing: http_proxy_digest.cpython-38.pyc
U �ˀXy � @ s> d Z ddlZddlmZmZ ddlmZ G dd� dej�Z dS )z*The module containing HTTPProxyDigestAuth.� N)�cookies�utils� )�_digest_auth_compatc sf e Zd ZdZejdejd�Z� fdd�Ze 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| _| �� d S )Nr )�superr �__init__� stale_rejects�init_per_thread_state)�self�args�kwargs�� __class__� �J/usr/lib/python3/dist-packages/requests_toolbelt/auth/http_proxy_digest.pyr s z HTTPProxyDigestAuth.__init__c C s t | dd �}|d kr| jS |jS �NZ _thread_local��getattrZ_stale_rejectsr )r � thread_localr r r r s z!HTTPProxyDigestAuth.stale_rejectsc C s& t | dd �}|d kr || _n||_d S r r )r �valuer r r r r " s c s, zt t| ��� 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 dk�r&| jdk �r&|j�d�}|dkr6td��n| j�|�sF|S t�| jj d|dd��| _ d |j jkr�d | j kr�| j d � � d kr�| jd7 _n| j d � � d kr�td ��|j |�� |j �� }t�|j|j |j� |�|j� | �|j|j�|jd < |jj|f|�}|j�|� ||_ |S |S dS )z�Handle 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 )�count�Proxy-AuthorizationZstale�trueZfalsezUser or password is invalid) Z status_coder �headers�get�IOError�_pat�matchr Zparse_dict_header�subZchalZrequest�lowerZcontent�close�copyr Zextract_cookies_to_jarZ_cookies�rawZprepare_cookies�build_digest_header�method�urlZ connection�send�history�append)r �rr Zs_authZprepZ_rr r r � handle_4071 s@ � � � � z HTTPProxyDigestAuth.handle_407c C s6 | � � | jr$| �|j|j�|jd< |�d| j� |S )Nr Zresponse)r Z last_noncer( r) r* r Z register_hookr/ )r r. r r r �__call___ s � z HTTPProxyDigestAuth.__call__)�__name__� __module__� __qualname__�__doc__�re�compile� IGNORECASEr! r �propertyr �setterr r/ r0 � __classcell__r r r r r s .r ) r4 r5 Zrequestsr r r r ZauthZHTTPDigestAuthr r r r r �<module> s
Save File
Cancel