Format: application/x-python-code
1 o
2
3 7'hm9 � @ s> d dl mZ d dlZd dlZd dlZd dlZddlmZmZ ddl m
4 Z
5 ddl m Z m
6 Z
7 mZmZmZmZmZ ddlmZ ddlmZ dd lmZ d
8 d lmZ d
9 d lm Z d
10 d
11 l m Z d
12 dl m Z ej!e"e#f Z$ej%ej&e$e$f Z'ej(e$e$f Z)e�*d�Z+ d d dd�Z,G dd� de �Z-G dd� de �Z.G dd � d e �Z/dS ) � )� annotationsN� )�
13 SOCKET_OPTION�AsyncNetworkBackend)�
14 ProxyError)�URL�Origin�Request�Response�
15 enforce_bytes�enforce_headers� enforce_url)�default_ssl_context)� AsyncLock)�Trace� )�AsyncHTTPConnection)�AsyncConnectionPool)�AsyncHTTP11Connection)�AsyncConnectionInterfacezhttpcore.proxy�default_headers�+typing.Sequence[tuple[bytes, bytes]] | None�override_headers�return�list[tuple[bytes, bytes]]c sT | du rg nt | �} |du rg nt |�}tdd� |D ��� � fdd�| D �} | | S )zh
16 Append default_headers and override_headers, de-duplicating if a key exists
17 in both cases.
18 Nc s s � | ] \}}|� � V qd S �N��lower��.0�key�value� r" �PC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpcore/_async/http_proxy.py� <genexpr>- s � z merge_headers.<locals>.<genexpr>c s$ g | ]\}}|� � � vr||f�qS r" r r �Z has_overrider" r# �
19 <listcomp>. s
20 �z!merge_headers.<locals>.<listcomp>)�list�set)r r r" r% r# �
21 merge_headers# s
22 �r) c sJ e Zd ZdZ d*d+� fd#d$�
23 Zd,d(d)�Z� ZS )-�AsyncHTTPProxyzB
24 A connection pool that sends requests via an HTTP proxy.
25 N�
26 TFr � proxy_url�URL | bytes | str�
27 proxy_auth�&tuple[bytes | str, bytes | str] | None�
28 proxy_headers�+HeadersAsMapping | HeadersAsSequence | None� ssl_context�ssl.SSLContext | None�proxy_ssl_context�max_connections�
29 int | None�max_keepalive_connections�keepalive_expiry� float | None�http1�bool�http2�retries�int�
30 local_address�
31 str | None�uds�network_backend�AsyncNetworkBackend | None�socket_options�%typing.Iterable[SOCKET_OPTION] | Noner �Nonec
32 s� t � j||||| |
33 || | |
34 |d� t|dd�| _| jjdkr&|dur&td��|| _|| _t|dd�| _ |dur_t
35 |d d d�}t
36 |d
37 d d�}|d | }d t � |� }d
38 |fg| j | _ dS dS )a�
39 A connection pool for making HTTP requests.
40
41 Parameters:
42 proxy_url: The URL to use when connecting to the proxy server.
43 For example `"http://127.0.0.1:8080/"`.
44 proxy_auth: Any proxy authentication as a two-tuple of
45 (username, password). May be either bytes or ascii-only str.
46 proxy_headers: Any HTTP headers to use for the proxy requests.
47 For example `{"Proxy-Authorization": "Basic <username>:<password>"}`.
48 ssl_context: An SSL context to use for verifying connections.
49 If not specified, the default `httpcore.default_ssl_context()`
50 will be used.
51 proxy_ssl_context: The same as `ssl_context`, but for a proxy server rather than a remote origin.
52 max_connections: The maximum number of concurrent HTTP connections that
53 the pool should allow. Any attempt to send a request on a pool that
54 would exceed this amount will block until a connection is available.
55 max_keepalive_connections: The maximum number of idle HTTP connections
56 that will be maintained in the pool.
57 keepalive_expiry: The duration in seconds that an idle HTTP connection
58 may be maintained for before being expired from the pool.
59 http1: A boolean indicating if HTTP/1.1 requests should be supported
60 by the connection pool. Defaults to True.
61 http2: A boolean indicating if HTTP/2 requests should be supported by
62 the connection pool. Defaults to False.
63 retries: The maximum number of retries when trying to establish
64 a connection.
65 local_address: Local address to connect from. Can also be used to
66 connect using a particular address family. Using
67 `local_address="0.0.0.0"` will connect using an `AF_INET` address
68 (IPv4), while using `local_address="::"` will connect using an
69 `AF_INET6` address (IPv6).
70 uds: Path to a Unix Domain Socket to use instead of TCP sockets.
71 network_backend: A backend instance to use for handling network I/O.
72 ) r2 r5 r7 r8 r: r< rB r= r? rA rD r, ��name� httpNzCThe `proxy_ssl_context` argument is not allowed for the http schemer0 r r. r � :s Basic s Proxy-Authorization)
73 �super�__init__r
74 �
75 _proxy_url�scheme� RuntimeError� _ssl_context�_proxy_ssl_contextr �_proxy_headersr �base64� b64encode)�selfr, r. r0 r2 r4 r5 r7 r8 r: r< r= r? rA rB rD �username�passwordZuserpass�
76 authorization�� __class__r" r# rL ; s> 5�� �
77 ��zAsyncHTTPProxy.__init__�originr r c C sV |j dkrt| jj| j|| j| j| jd�S t| jj| j|| j | j| j| j
78 | j | jd� S )NrI )� proxy_originr0 �
79 remote_originr8 rB r4 ) r\ r0 r] r2 r4 r8 r: r< rB ) rN �AsyncForwardHTTPConnectionrM r[ rR �_keepalive_expiryZ_network_backendrQ �AsyncTunnelHTTPConnectionrP �_http1�_http2�rU r[ r" r" r# �create_connection� s(
80 ��z AsyncHTTPProxy.create_connection)NNNNr+ NNTFr NNNN) r, r- r. r/ r0 r1 r2 r3 r4 r3 r5 r6 r7 r6 r8 r9 r: r; r< r; r= r> r? r@ rA r@ rB rC rD rE r rF )r[ r r r )�__name__�
81 __module__� __qualname__�__doc__rL rd �
82 __classcell__r" r" rY r# r* 6 s$ �Wr* c @ s| e Zd Z d+d,dd�Zd-dd�Zd.dd�Zd/d d �Zd0d d �Zd1d!d"�Zd1d#d$�Z d1d%d&�Z
83 d1d'd(�Z d0d)d*�Z dS )2r^ Nr\ r r] r0 r1 r8 r9 rB rC rD rE r4 r3 r rF c C s2 t |||||d�| _|| _t|dd�| _|| _d S )N�r[ r8 rB rD r2 r0 rG )r � _connection�
84 _proxy_originr rR �_remote_origin)rU r\ r] r0 r8 rB rD r4 r" r" r# rL � s
85 �
86 z#AsyncForwardHTTPConnection.__init__�requestr r
87 c � s\ �t | j|j�}t| jj| jj| jjt|j �d�}t
88 |j |||j |j
89 d�}| j�|�I d H S )N�rN �host�port�target)�method�url�headers�content�
90 extensions)r) rR ru r rl rN rp rq �bytesrt r rs �streamrw rk �handle_async_request)rU rn ru rt Z
91 proxy_requestr" r" r# rz � s ���z/AsyncForwardHTTPConnection.handle_async_requestr[ r; c C �
92 || j kS r �rm rc r" r" r# �can_handle_request� �
93 z-AsyncForwardHTTPConnection.can_handle_requestc � � �| j �� I d H d S r �rk �aclose�rU r" r" r# r� � � �z!AsyncForwardHTTPConnection.aclose�strc C �
94 | j �� S r �rk �infor� r" r" r# r� � r~ z AsyncForwardHTTPConnection.infoc C r� r �rk � is_availabler� r" r" r# r� � r~ z'AsyncForwardHTTPConnection.is_availablec C r� r �rk � has_expiredr� r" r" r# r� � r~ z&AsyncForwardHTTPConnection.has_expiredc C r� r �rk �is_idler� r" r" r# r� � r~ z"AsyncForwardHTTPConnection.is_idlec C r� r �rk � is_closedr� r" r" r# r� � r~ z$AsyncForwardHTTPConnection.is_closedc C � d| j j� d| �� � d�S �N�<z [z]>�rZ re r� r� r" r" r# �__repr__� � z#AsyncForwardHTTPConnection.__repr__)NNNNN)r\ r r] r r0 r1 r8 r9 rB rC rD rE r4 r3 r rF �rn r r r
95 �r[ r r r; �r rF �r r� �r r; �
96 re rf rg rL rz r} r� r� r� r� r� r� r� r" r" r" r# r^ � s �
97
98
99
100
101
102
103
104 r^ c @ s� e Zd Z d0d1dd�Zd2d d �Zd3d d �Zd4d!d"�Zd5d$d%�Zd6d&d'�Zd6d(d)�Z d6d*d+�Z
105 d6d,d-�Z d5d.d/�Z dS )7r` NTFr\ r r] r2 r3 r4 r0 r r8 r9 r: r; r< rB rC rD rE r rF c C s^ t ||| |
106 |d�| _|| _|| _|| _|| _t|dd�| _|| _|| _ || _
107 t � | _ d| _
108 d S )Nrj r0 rG F)r rk rl rm rP rQ r rR r_ ra rb r �
109 _connect_lock�
110 _connected) rU r\ r] r2 r4 r0 r8 r: r< rB rD r" r" r# rL � s"
111 �
112 z"AsyncTunnelHTTPConnection.__init__rn r r
113 c
114 � sL �|j �di �}|�dd �}| j4 I d H �� | j�sd| jj| jjf }t| jj | jj| jj|d�}t
115 d|fdg| j �}t d|||j d�}| j
116 �|�I d H }|jd k sX|jd
117 kry|j �d d �} | jd
118 dd�}
119 d|j|
120 f } | j
121 �� I d H t| ��|j d } | jd u r�t� n| j}
122 | jr�ddgndg}|
123 �|� |
124 | jj�d
125 �|d�}tdt||�4 I d H �}| jdi |��I d H } | |_W d �I d H n 1 I d H s�w Y | �d�}|d uo�|� � dk}|s�| jr�| j s�ddl m } || j| | j d�| _
126 n
127 t!| j| | j d�| _
128 d| _W d �I d H n 1 I d H �sw Y | j
129 �|�I d H S ) N�timeout�connects %b:%dro s Host)s Accepts */*s CONNECT)rs rt ru rw �� i+ Z
130 reason_phrase� �ascii�ignore)�errorsz%d %sZnetwork_streamzhttp/1.1�h2)r2 �server_hostnamer� � start_tls�
131 ssl_objectr )�AsyncHTTP2Connection)r[ ry r8 Tr" )"rw �getr� r� rm rp rq r rl rN r) rR r rk rz �status�decoder� r rP r rb �set_alpn_protocolsr �loggerr� � return_value�get_extra_info�selected_alpn_protocolra r< r� r_ r )rU rn Ztimeoutsr� rr Z connect_urlZconnect_headersZconnect_requestZconnect_responseZ reason_bytesZ
132 reason_str�msgry r2 �alpn_protocols�kwargs�tracer� Zhttp2_negotiatedr� r" r" r# rz s� � ���
133 �
134
135 ��
136 �(�
137
138 �
139 ��*�Jz.AsyncTunnelHTTPConnection.handle_async_requestr[ c C r{ r r| rc r" r" r# r} Y r~ z,AsyncTunnelHTTPConnection.can_handle_requestc � r r r� r� r" r" r# r� \ r� z AsyncTunnelHTTPConnection.acloser� c C r� r r� r� r" r" r# r� _ r~ z AsyncTunnelHTTPConnection.infoc C r� r r� r� r" r" r# r� b r~ z&AsyncTunnelHTTPConnection.is_availablec C r� r r� r� r" r" r# r� e r~ z%AsyncTunnelHTTPConnection.has_expiredc C r� r r� r� r" r" r# r� h r~ z!AsyncTunnelHTTPConnection.is_idlec C r� r r� r� r" r" r# r� k r~ z#AsyncTunnelHTTPConnection.is_closedc C r� r� r� r� r" r" r# r� n r� z"AsyncTunnelHTTPConnection.__repr__)NNNNTFNN)r\ r r] r r2 r3 r4 r3 r0 r r8 r9 r: r; r< r; rB rC rD rE r rF r� r� r� r� r� r� r" r" r" r# r` � s& �
140
141 P
142
143
144
145
146
147 r` )NN)r r r r r r )0�
148 __future__r rS �logging�ssl�typingZ_backends.baser r � _exceptionsr Z_modelsr r r r
149 r r r
150 �_sslr Z_synchronizationr �_tracer �
151 connectionr Zconnection_poolr Zhttp11r Z
152 interfacesr �Unionrx r� Z ByteOrStr�Sequence�TupleZHeadersAsSequence�MappingZHeadersAsMapping� getLoggerr� r) r* r^ r` r" r" r" r# �<module> s2 $
153 �s@