File: //usr/local/lib/python3.7/site-packages/requests/__pycache__/auth.cpython-37.pyc
B
���g�' � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z m
Z
mZ ddlmZ ddlmZ d Zd
Zd d � ZG d
d� d�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )z]
requests.auth
~~~~~~~~~~~~~
This module contains the authentication handlers for Requests.
� N)� b64encode� )�to_native_string)�
basestring�str�urlparse)�extract_cookies_to_jar)�parse_dict_headerz!application/x-www-form-urlencodedzmultipart/form-datac C s� t | t�s&tjd�| �td� t| �} t |t�sPtjd�t|��td� t|�}t | t�rd| �d�} t |t�rx|�d�}dt t
d� | |f��� � � }|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 �7/usr/local/lib/python3.7/site-packages/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 d��d S )Nz Auth hooks must be callable.)�NotImplementedError)�self�rr r r �__call__H s zAuthBase.__call__N)�__name__�
__module__� __qualname__�__doc__r r r r r r E 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__O s zHTTPBasicAuth.__init__c C s( t | jt|dd �k| jt|dd �kg�S )Nr r )�allr �getattrr )r �otherr r r �__eq__S 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% L 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 f s zHTTPProxyAuth.__call__N)r! r" r# r$ r r r r r r. c 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 || _ || _t�� | _d S )N)r r � threading�local�
_thread_local)r r r r r r r& n s zHTTPDigestAuth.__init__c C s@ t | jd�s<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_statet s z$HTTPDigestAuth.init_per_thread_statec s� | j jd }| j jd }| j j�d�}| j j�d�}| j j�d�}d� |dkrTd}n|�� }|dksl|dkrzd d
� } | � n@|d kr�d d
� }
|
� n*|dkr�dd� } | � n|dkr�dd� } | � � fdd�}
� dkr�dS d}t|�}|jp�d}|jr�|d|j� �7 }| j� d|� d| j� �}|� d|� �}� |�}� |�}|| j j k�rT| j j
d7 _
nd| j _
| j j
d�}t | j j
�� d�}||� d�7 }|t
�� � d�7 }|t�d �7 }t�|��� dd � }|dk�r� |� d|� d|� ��}|�s�|
||� d|� ��}nD|d k�sd |�d �k�r<|� d|� d|� d |� �}|
||�}ndS || j _ d!| j� d"|� d#|� d$|� d%|� d&� }|�r�|d'|� d&�7 }|�r�|d(|� d&�7 }|�r�|d)|� d&�7 }|�r�|d*|� d+|� d&�7 }d,|� �S )-z
:rtype: str
�realm�nonce�qop� algorithm�opaqueN�MD5zMD5-SESSc S s" t | t�r| �d�} t�| ��� 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 | t�r| �d�} t�| ��� 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 | t�r| �d�} t�| ��� 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 | t�r| �d�} t�| ��� S )Nzutf-8)r
r r rB �sha512rD )rE r r r � sha512_utf8� s
z7HTTPDigestAuth.build_digest_header.<locals>.sha512_utf8c s � | � d|� ��S )N�:r )�s�d)� hash_utf8r r �<lambda>� � z4HTTPDigestAuth.build_digest_header.<locals>.<lambda>�/�?rN r �08xzutf-8� � �auth�,z:auth:z
username="z
", realm="z
", nonce="z", uri="z
", response="�"z
, opaque="z
, algorithm="z
, digest="z, qop="auth", nc=z
, cnonce="zDigest )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_parsedr^ �A1�A2�HA1�HA2�ncvaluerO �cnonce�respdig�noncebit�baser )rQ r �build_digest_header~ sz
&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 krdk s$n d| j_|S | jjdk rB|jj�| jj� |j�dd�}d|� � k�r| jjdk �r| j jd7 _t
j d t
j d
�}t
|jd|dd ��| j_|j |�� |j�� }t|j|j|j� |�|j� | �|j|j�|jd < |jj|f|�}|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- r\ �lower�re�compile�
IGNORECASEr �subr8 �content�close�copyr �_cookies�raw�prepare_cookiesrt re rf �
connection�send�history�append)r r rv �s_auth�pat�prep�_rr r r �
handle_401� s.
zHTTPDigestAuth.handle_401c C s~ | � � | jjr&| �|j|j�|jd< y|j�� | j_ W n t
k
rT d | j_ Y nX |� d| j � |� d| j
� d| j_|S )Nr, �responser )r; r2 r6 rt re rf r- r~ �tellr9 �AttributeError�
register_hookr� rw r: )r r r r r r s zHTTPDigestAuth.__call__c C s( t | jt|dd �k| jt|dd �kg�S )Nr r )r' r r( r )r r) r r r r* 2 s zHTTPDigestAuth.__eq__c C s
| |k S )Nr )r r) r r r r+ : s zHTTPDigestAuth.__ne__N) r! r" r# r$ r& r; rt rw r� r r* r+ r r r r r/ k s
n-r/ )r$ rB rb r� r0 r` r �base64r �_internal_utilsr �compatr r r �cookiesr �utilsr � CONTENT_TYPE_FORM_URLENCODED�CONTENT_TYPE_MULTI_PARTr r r% r. r/ r r r r �<module> s$ ,