File: //proc/1233/root/usr/local/lib/python3.10/dist-packages/httpx/__pycache__/_auth.cpython-310.pyc
o
���gs. � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
ddl m Z ddl
mZmZmZ ddlmZmZmZ ejrId d lmZ g d
�ZG d d � d �ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�Z G dd� dej �Z dS )� )� annotationsN)� b64encode)�parse_http_list� )�
ProtocolError)�Cookies�Request�Response)�to_bytes�to_str�unquote)�_Hash)�Auth� BasicAuth�
DigestAuth� NetRCAuthc @ s6 e Zd ZdZdZdZddd�Zdd d
�Zdd d
�ZdS )r a�
Base class for all authentication schemes.
To implement a custom authentication scheme, subclass `Auth` and override
the `.auth_flow()` method.
If the authentication scheme does I/O such as disk access or network calls, or uses
synchronization primitives such as locks, you should override `.sync_auth_flow()`
and/or `.async_auth_flow()` instead of `.auth_flow()` to provide specialized
implementations that will be used by `Client` and `AsyncClient` respectively.
F�requestr �return�)typing.Generator[Request, Response, None]c c s � |V dS )a
Execute the authentication flow.
To dispatch a request, `yield` it:
```
yield request
```
The client will `.send()` the response back into the flow generator. You can
access it like so:
```
response = yield request
```
A `return` (or reaching the end of the generator) will result in the
client returning the last response obtained from the server.
You can dispatch as many requests as is necessary.
N� ��selfr r r �6/usr/local/lib/python3.10/dist-packages/httpx/_auth.py� auth_flow& s �
zAuth.auth_flowc c s^ � | j r|�� | �|�}t|�} |V }| jr |�� z|�|�}W n
ty- Y dS w q)z�
Execute the authentication flow synchronously.
By default, this defers to `.auth_flow()`. You should override this method
when the authentication scheme does I/O and/or uses concurrency primitives.
TN)�requires_request_body�readr �next�requires_response_body�send�
StopIteration�r r �flow�responser r r �sync_auth_flow> s �
��zAuth.sync_auth_flow�(typing.AsyncGenerator[Request, Response]c C sj �| j r |�� I dH | �|�}t|�} |V }| jr"|�� I dH z|�|�}W n
ty3 Y dS w q)z�
Execute the authentication flow asynchronously.
By default, this defers to `.auth_flow()`. You should override this method
when the authentication scheme does I/O and/or uses concurrency primitives.
N)r �areadr r r r r r r r r �async_auth_flowW s �
��zAuth.async_auth_flowN�r r r r )r r r r$ ) �__name__�
__module__� __qualname__�__doc__r r r r# r& r r r r r s
r c @ s$ e Zd ZdZddd�Zdd d �Zd
S )� FunctionAuthz�
Allows the 'auth' argument to be passed as a simple callable function,
that takes the request, and returns a new, modified request.
�func�#typing.Callable[[Request], Request]r �Nonec C s
|| _ d S �N��_func)r r- r r r �__init__w s
zFunctionAuth.__init__r r r c c s � | � |�V d S r0 r1 r r r r r z s �zFunctionAuth.auth_flowN)r- r. r r/ r' )r( r) r* r+ r3 r r r r r r, q s
r, c @ s. e Zd ZdZddd�Zdd d
�Zddd�ZdS )r zy
Allows the 'auth' argument to be passed as a (username, password) pair,
and uses HTTP Basic authentication.
�username� str | bytes�passwordr r/ c C s | � ||�| _d S r0 )�_build_auth_header� _auth_header�r r4 r6 r r r r3 � s zBasicAuth.__init__r r r c c s � | j |jd< |V d S )N�
Authorization)r8 �headersr r r r r � s �
zBasicAuth.auth_flow�strc C �, d� t|�t|�f�}t|��� }d|� �S �N� :zBasic ��joinr
r �decode�r r4 r6 �userpass�tokenr r r r7 � �
z BasicAuth._build_auth_headerN�r4 r5 r6 r5 r r/ r' �r4 r5 r6 r5 r r< �r( r) r* r+ r3 r r7 r r r r r ~ s
r c @ s0 e Zd ZdZdddd�Zdd d
�Zddd�ZdS )r zT
Use a 'netrc' file to lookup basic auth credentials based on the url host.
N�file�
str | Noner r/ c C s dd l }|� |�| _d S )Nr )�netrc� _netrc_info)r rJ rL r r r r3 � s zNetRCAuth.__init__r r r c c sR � | j �|jj�}|d u s|d s|V d S | j|d |d d�|jd< |V d S )N� r )r4 r6 r: )rM �authenticators�url�hostr7 r; )r r � auth_infor r r r � s �
�
zNetRCAuth.auth_flowr4 r5 r6 r< c C r= r> r@ rC r r r r7 � rF z NetRCAuth._build_auth_headerr0 )rJ rK r r/ r' rH rI r r r r r � s
r c @ s� e Zd ZU ejejejejejejejejd�Zde d< d)d d
�Z
d*dd�Z d+dd�Z d,dd�Z
d-d d �Zd.d"d#�Zd/d&d'�Zd(S )0r )�MD5zMD5-SESS�SHAzSHA-SESSzSHA-256z SHA-256-SESSzSHA-512z SHA-512-SESSz*dict[str, typing.Callable[[bytes], _Hash]]�_ALGORITHM_TO_HASH_FUNCTIONr4 r5 r6 r r/ c C s$ t |�| _t |�| _d | _d| _d S )Nr )r
� _username� _password�_last_challenge� _nonce_countr9 r r r r3 � s
zDigestAuth.__init__r r r c c s� � | j r| �|| j �|jd< |V }|jdksd|jvr d S |j�d�D ] }|�� �d�r. nq#d S | �|||�| _ d| _| �|| j �|jd< |j rRt
|j �j |d� |V d S )Nr: i� zwww-authenticatezdigest r �r ) rX r7 r; � status_code�get_list�lower�
startswith�_parse_challengerY �cookiesr �set_cookie_header)r r r" � auth_headerr r r r � s* �
��
�
zDigestAuth.auth_flowr" r rb r< �_DigestAuthChallengec
C s� |� d�\}}}|�� dksJ �i }t|�D ]}|�� �dd�\} }
t|
�|| < qz4|d �� } |d �� } |�dd�}
d |v rF|d �� nd
}d |v rR|d �� nd
}t| | |
||d �W S t yr } z d
}t
||d�|�d
}~ww )z�
Returns a challenge from a Digest WWW-Authenticate header.
These take the form of:
`Digest realm="realm@host.com",qop="auth,auth-int",nonce="abc",opaque="xyz"`
� �digest�=r �realm�nonce� algorithmrS �opaqueN�qop)rg rh ri rj rk z(Malformed Digest WWW-Authenticate headerrZ ) � partitionr] r �strip�splitr �encode�getrc �KeyErrorr )r r r" rb �scheme�_�fields� header_dict�field�key�valuerg rh ri rj rk �exc�messager r r r_ � s(
���zDigestAuth._parse_challenge� challengec sH | j |j�� � d� fdd� }d�| j|j| jf�}|jj}d�|j �
� |f�}||�}d| j }| � | j |j
�} | j d7 _ ||�}
|j�� �d �rW|d�|
|j
| f��}
| j|j|d
�} | d u rj|
|j
|g} n |
|j
|| | |g} | j|j|j
||d�| ��|j�
� d �}
|jr�|j|
d < | r�d
|
d< ||
d< | |
d< d| �|
� S )N�data�bytesr c s � | �� � �� S r0 )� hexdigestro )r| �� hash_funcr r re s z-DigestAuth._build_auth_header.<locals>.digestr? s %08xr z-sessrZ )r4 rg rh �urir" ri rj � authrk �nc�cnoncezDigest )r| r} r r} )rU ri �upperrA rV rg rW rP �raw_path�methodro rY �_get_client_noncerh r] �endswith� _resolve_qoprk rj �_get_header_value)r r r{ re �A1�path�A2�HA2�nc_valuer� �HA1rk � digest_data� format_argsr r r r7 � s<
�
z DigestAuth._build_auth_header� nonce_count�intrh r} c C sL t |��� }||7 }|t�� �� 7 }|t�d�7 }t�|��� d d� �� S )N� � ) r< ro �time�ctime�os�urandom�hashlib�sha1r~ )r r� rh �sr r r r� / s
z DigestAuth._get_client_nonce�
header_fields�dict[str, bytes]c
C sb d}d}d}d}t |�� �D ] \}\}}|dkr |d7 }||vr"|n|} || �|t|��7 }q|S )N)ri rk r� z{}="{}"z{}={}� r z, )� enumerate�items�formatr )
r r� �NON_QUOTED_FIELDS�QUOTED_TEMPLATE�NON_QUOTED_TEMPLATE� header_value�irv rx �templater r r r� 7 s ��z DigestAuth._get_header_valuerk � bytes | Nonec C sN |d u rd S t �d|�}d|v rdS |dgkrtd��d|�d�}t||d��)Ns , ?r� s auth-intz.Digest auth-int support is not yet implementedzUnexpected qop value "z" in digest authrZ )�rern �NotImplementedErrorr )r rk r �qopsrz r r r r� I s
zDigestAuth._resolve_qopNrG r' )r r r" r rb r< r rc )r r r{ rc r r< )r� r� rh r} r r} )r� r� r r< )rk r� r r r r� )r( r) r* r� �md5r� �sha256�sha512rU �__annotations__r3 r r_ r7 r� r� r� r r r r r � s"
�
0
r c @ s6 e Zd ZU ded< ded< ded< ded< ded< d S )
rc r} rg rh r< ri r� rj rk N)r( r) r* r� r r r r rc W s
rc ) �
__future__r r� r� r� r� �typing�base64r �urllib.requestr � _exceptionsr �_modelsr r r �_utilsr
r r �
TYPE_CHECKINGr
�__all__r r, r r r �
NamedTuplerc r r r r �<module> s* [
)