Format: application/x-python-code
1 o
2
3 C'hs. � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
4 Z
5 ddl m Z ddl
6 mZmZmZ ddlmZmZmZ ejrId d lmZ g d
7 �ZG d d � d �ZG d
8 d� de�ZG dd� de�ZG dd� de�ZG dd� de�Z G dd� dej �Z dS )� )� annotationsN)� b64encode)�parse_http_list� )�
9 ProtocolError)�Cookies�Request�Response)�to_bytes�to_str�unquote)�_Hash)�Auth� BasicAuth�
10 DigestAuth� NetRCAuthc @ s6 e Zd ZdZdZdZddd�Zdd d
11 �Zdd d
12 �ZdS )r a�
13 Base class for all authentication schemes.
14
15 To implement a custom authentication scheme, subclass `Auth` and override
16 the `.auth_flow()` method.
17
18 If the authentication scheme does I/O such as disk access or network calls, or uses
19 synchronization primitives such as locks, you should override `.sync_auth_flow()`
20 and/or `.async_auth_flow()` instead of `.auth_flow()` to provide specialized
21 implementations that will be used by `Client` and `AsyncClient` respectively.
22 F�requestr �return�)typing.Generator[Request, Response, None]c c s � |V dS )a
23 Execute the authentication flow.
24
25 To dispatch a request, `yield` it:
26
27 ```
28 yield request
29 ```
30
31 The client will `.send()` the response back into the flow generator. You can
32 access it like so:
33
34 ```
35 response = yield request
36 ```
37
38 A `return` (or reaching the end of the generator) will result in the
39 client returning the last response obtained from the server.
40
41 You can dispatch as many requests as is necessary.
42 N� ��selfr r r �AC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpx/_auth.py� auth_flow& s �
43 zAuth.auth_flowc c s^ � | j r|�� | �|�}t|�} |V }| jr |�� z|�|�}W n
44 ty- Y dS w q)z�
45 Execute the authentication flow synchronously.
46
47 By default, this defers to `.auth_flow()`. You should override this method
48 when the authentication scheme does I/O and/or uses concurrency primitives.
49 TN)�requires_request_body�readr �next�requires_response_body�send�
50 StopIteration�r r Zflow�responser r r �sync_auth_flow> s �
51 ��zAuth.sync_auth_flow�(typing.AsyncGenerator[Request, Response]c C sj �| j r |�� I dH | �|�}t|�} |V }| jr"|�� I dH z|�|�}W n
52 ty3 Y dS w q)z�
53 Execute the authentication flow asynchronously.
54
55 By default, this defers to `.auth_flow()`. You should override this method
56 when the authentication scheme does I/O and/or uses concurrency primitives.
57 N)r Zareadr r r r r r r r r �async_auth_flowW s �
58 ��zAuth.async_auth_flowN�r r r r )r r r r# ) �__name__�
59 __module__� __qualname__�__doc__r r r r" r$ r r r r r s
60
61 r c @ s$ e Zd ZdZddd�Zdd d �Zd
62 S )� FunctionAuthz�
63 Allows the 'auth' argument to be passed as a simple callable function,
64 that takes the request, and returns a new, modified request.
65 �func�#typing.Callable[[Request], Request]r �Nonec C s
66 || _ d S �N�Z_func)r r+ r r r �__init__w s
67 zFunctionAuth.__init__r r r c c s � | � |�V d S r. r/ r r r r r z s �zFunctionAuth.auth_flowN)r+ r, r r- r% )r& r' r( r) r0 r r r r r r* q s
68 r* c @ s. e Zd ZdZddd�Zdd d
69 �Zddd�ZdS )r zy
70 Allows the 'auth' argument to be passed as a (username, password) pair,
71 and uses HTTP Basic authentication.
72 �username� str | bytes�passwordr r- c C s | � ||�| _d S r. )�_build_auth_header� _auth_header�r r1 r3 r r r r0 � s zBasicAuth.__init__r r r c c s � | j |jd< |V d S )N�
73 Authorization)r5 �headersr r r r r � s �
74 zBasicAuth.auth_flow�strc C �, d� t|�t|�f�}t|��� }d|� �S �N� :zBasic ��joinr
75 r �decode�r r1 r3 Zuserpass�tokenr r r r4 � �
76 z BasicAuth._build_auth_headerN�r1 r2 r3 r2 r r- r% �r1 r2 r3 r2 r r9 �r& r' r( r) r0 r r4 r r r r r ~ s
77
78
79 r c @ s0 e Zd ZdZdddd�Zdd d
80 �Zddd�ZdS )r zT
81 Use a 'netrc' file to lookup basic auth credentials based on the url host.
82 N�file�
83 str | Noner r- c C s dd l }|� |�| _d S )Nr )�netrc� _netrc_info)r rF rH r r r r0 � 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 )r1 r3 r7 )rI �authenticators�url�hostr4 r8 )r r Z auth_infor r r r � s �
84 �
85 zNetRCAuth.auth_flowr1 r2 r3 r9 c C r: r; r= r@ r r r r4 � rB z NetRCAuth._build_auth_headerr. )rF rG r r- r% rD rE r r r r r � s
86
87 r c @ s� e Zd ZU ejejejejejejejejd�Zde d< d)d d
88 �Z
89 d*dd�Z d+dd�Z d,dd�Z
90 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_FUNCTIONr1 r2 r3 r r- c C s$ t |�| _t |�| _d | _d| _d S )Nr )r
91 � _username� _password�_last_challenge� _nonce_countr6 r r r r0 � s
92
93
94 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
95 |j �j |d� |V d S )Nr7 i� zwww-authenticatezdigest r �r ) rS r4 r8 � status_codeZget_list�lower�
96 startswith�_parse_challengerT �cookiesr Zset_cookie_header)r r r! � auth_headerr r r r � s* �
97 ��
98 �
99 zDigestAuth.auth_flowr! r r[ r9 �_DigestAuthChallengec
100 C s� |� d�\}}}|�� dksJ �i }t|�D ]}|�� �dd�\} }
101 t|
102 �|| < qz4|d �� } |d �� } |�dd�}
103 d |v rF|d �� nd
104 }d |v rR|d �� nd
105 }t| | |
106 ||d �W S t yr } z d
107 }t
108 ||d�|�d
109 }~ww )z�
110 Returns a challenge from a Digest WWW-Authenticate header.
111 These take the form of:
112 `Digest realm="realm@host.com",qop="auth,auth-int",nonce="abc",opaque="xyz"`
113 � �digest�=r �realm�nonce� algorithmrN �opaqueN�qop)r` ra rb rc rd z(Malformed Digest WWW-Authenticate headerrU ) � partitionrW r �strip�splitr �encode�getr\ �KeyErrorr )r r r! r[ �scheme�_�fieldsZ header_dict�field�key�valuer` ra rb rc rd �exc�messager r r rY � s(
114 ���zDigestAuth._parse_challenge� challengec sH | j |j�� � d� fdd� }d�| j|j| jf�}|jj}d�|j �
115 � |f�}||�}d| j }| � | j |j
116 �} | j d7 _ ||�}
117 |j�� �d �rW|d�|
118 |j
119 | f��}
120 | j|j|d
121 �} | d u rj|
122 |j
123 |g} n |
124 |j
125 || | |g} | j|j|j
126 ||d�| ��|j�
127 � d �}
128 |jr�|j|
129 d < | r�d
130 |
131 d< ||
132 d< | |
133 d< d| �|
134 � S )N�data�bytesr c s � | �� � �� S r. )� hexdigestrh )rt �Z hash_funcr r r^ s z-DigestAuth._build_auth_header.<locals>.digestr< s %08xr z-sessrU )r1 r` ra �urir! rb rc � authrd �nc�cnoncezDigest )rt ru r ru )rP rb �upperr> rQ r` rR rL �raw_path�methodrh rT �_get_client_noncera rW �endswith� _resolve_qoprd rc �_get_header_value)r r rs r^ �A1�path�A2�HA2Znc_valuer{ �HA1rd Z digest_dataZ format_argsr rw r r4 � s<
135 �
136 z DigestAuth._build_auth_header� nonce_count�intra ru c C sL t |��� }||7 }|t�� �� 7 }|t�d�7 }t�|��� d d� �� S )N� � ) r9 rh �time�ctime�os�urandom�hashlib�sha1rv )r r� ra �sr r r r / s
137 z DigestAuth._get_client_nonce�
138 header_fields�dict[str, bytes]c
139 C sb d}d}d}d}t |�� �D ] \}\}}|dkr |d7 }||vr"|n|} || �|t|��7 }q|S )N)rb rd rz z{}="{}"z{}={}� r z, )� enumerate�items�formatr )
140 r r� ZNON_QUOTED_FIELDSZQUOTED_TEMPLATEZNON_QUOTED_TEMPLATE� header_value�irn rp �templater r r r� 7 s ��z DigestAuth._get_header_valuerd � bytes | Nonec C sN |d u rd S t �d|�}d|v rdS |dgkrtd��d|�d�}t||d��)Ns , ?ry s auth-intz.Digest auth-int support is not yet implementedzUnexpected qop value "z" in digest authrU )�rerg �NotImplementedErrorr )r rd r Zqopsrr r r r r� I s
141 zDigestAuth._resolve_qopNrC r% )r r r! r r[ r9 r r\ )r r rs r\ r r9 )r� r� ra ru r ru )r� r� r r9 )rd r� r r r r� )r& r' r( r� �md5r� �sha256�sha512rP �__annotations__r0 r rY r4 r r� r� r r r r r � s"
142 �
143
144
145
146
147 0
148 r c @ s6 e Zd ZU ded< ded< ded< ded< ded< d S )
149 r\ ru r` ra r9 rb r� rc rd N)r& r' r( r� r r r r r\ W s
150 r\ ) �
151 __future__r r� r� r� r� �typing�base64r �urllib.requestr � _exceptionsr Z_modelsr r r �_utilsr
152 r r �
153 TYPE_CHECKINGr
154 �__all__r r* r r r �
155 NamedTupler\ r r r r �<module> s* [
156 )