Edit File: guess.cpython-36.pyc
3 �P\P � @ sR d Z ddlmZ ddlmZ ddlmZmZ G dd� dej�Z G dd � d e �Z d S ) z-The module containing the code for GuessAuth.� )�auth)�cookies� )�_digest_auth_compat�http_proxy_digestc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � GuessAuthz7Guesses the auth type by the WWW-Authentication header.c C s || _ || _d | _d | _d S )N)�username�passwordr �pos)�selfr r � r �/usr/lib/python3.6/guess.py�__init__ s zGuessAuth.__init__c C s� | j d k r|jjj| j � |j |jj� |jj� }t|d�sHt j � |_ t j |j |j|j� |j |j � tj| j| j�| _| j|�}|jj|f|�}|jj|� ||_|S )N�_cookies)r �request�body�seek�content�raw� release_conn�copy�hasattrr �RequestsCookieJarr �extract_cookies_to_jar�prepare_cookiesr Z HTTPBasicAuthr r � connection�send�history�append)r �r�kwargs�prep�_rr r r �_handle_basic_auth_401 s z GuessAuth._handle_basic_auth_401c C sf t j| j| j�| _y| jj� W n tk r4 Y nX t| jd�rV| jjd krVd| j_| jj |f|�S )N� num_401_callsr ) � auth_compatZHTTPDigestAuthr r r �init_per_thread_state�AttributeErrorr r$ � handle_401)r r r r r r �_handle_digest_auth_401' s z!GuessAuth._handle_digest_auth_401c K s> |j jdd�j� }d|kr&| j||�S d|kr:| j||�S dS )z/Resends a request with auth headers, if needed.zwww-authenticate� �basic�digestN)�headers�get�lowerr# r) )r r r Zwww_authenticater r r r( : s zGuessAuth.handle_401c C sL | j d k r| j |�S y|jj� | _W n tk r8 Y nX |jd| j� |S )N�response)r r �tellr r' � register_hookr( )r r r r r �__call__E s zGuessAuth.__call__N) �__name__� __module__� __qualname__�__doc__r r# r) r( r3 r r r r r s r c sF e Zd ZdZd � fdd� Zdd� Zdd� Zd d � Z� fd d �Z� Z S )�GuessProxyAuthzZ Guesses the auth type by WWW-Authentication and Proxy-Authentication headers Nc s( t t| �j||� || _|| _d | _d S )N)�superr8 r �proxy_username�proxy_password� proxy_auth)r r r r: r; )� __class__r r r W s zGuessProxyAuth.__init__c C s� | j d k r|jjj| j � |j |jj� |jj� }t|d�sHt j � |_ t j |j |j|j� |j |j � tj| j| j�| _| j|�}|jj|f|�}|jj|� ||_|S )Nr )r r r r r r r r r r r r r r r Z HTTPProxyAuthr: r; r< r r r r )r r r r! r" r r r �_handle_basic_auth_407^ s z%GuessProxyAuth._handle_basic_auth_407c C sH t j| j| jd�| _y| jj� W n tk r6 Y nX | jj|f|�S )N)r r ) r ZHTTPProxyDigestAuthr: r; r< r r& r' � handle_407)r r r r r r �_handle_digest_auth_407s s z&GuessProxyAuth._handle_digest_auth_407c K s> |j jdd�j� }d|kr&| j||�S d|kr:| j||�S d S )NzProxy-Authenticater* r+ r, )r- r. r/ r> r@ )r r r Zproxy_authenticater r r r? s zGuessProxyAuth.handle_407c sX | j d k r| j |�}y|jj� | _W n tk r8 Y nX |jd| j� tt| �j |�S )Nr0 ) r< r r1 r r' r2 r? r9 r8 r3 )r r )r= r r r3 � s zGuessProxyAuth.__call__)NNNN) r4 r5 r6 r7 r r> r@ r? r3 � __classcell__r r )r= r r8 R s r8 N) r7 Zrequestsr r r* r r% r ZAuthBaser r8 r r r r �<module> s I
Back to File Manager