Format: application/x-python-code
1 o
2
3 7'h8 � @ 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�NetworkBackend)�
14 ProxyError)�URL�Origin�Request�Response�
15 enforce_bytes�enforce_headers� enforce_url)�default_ssl_context)�Lock)�Trace� )�HTTPConnection)�ConnectionPool)�HTTP11Connection)�ConnectionInterfacezhttpcore.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" �OC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpcore/_sync/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 )-� HTTPProxyzB
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�NetworkBackend | 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 ��zHTTPProxy.__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 �ForwardHTTPConnectionrM r[ rR �_keepalive_expiryZ_network_backendrQ �TunnelHTTPConnectionrP �_http1�_http2�rU r[ r" r" r# �create_connection� s(
80 ��zHTTPProxy.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 ForwardHTTPConnection.__init__�requestr r
87 c C sT t | j|j�}t| jj| jj| jjt|j �d�}t
88 |j |||j |j
89 d�}| j�|�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_request)rU rn ru rt Z
91 proxy_requestr" r" r# rz � s �� z$ForwardHTTPConnection.handle_requestr[ r; c C �
92 || j kS r �rm rc r" r" r# �can_handle_request� �
93 z(ForwardHTTPConnection.can_handle_requestc C � | j �� d S r �rk �close�rU r" r" r# r� � � zForwardHTTPConnection.close�strc C �
94 | j �� S r �rk �infor� r" r" r# r� � r~ zForwardHTTPConnection.infoc C r� r �rk � is_availabler� r" r" r# r� � r~ z"ForwardHTTPConnection.is_availablec C r� r �rk � has_expiredr� r" r" r# r� � r~ z!ForwardHTTPConnection.has_expiredc C r� r �rk �is_idler� r" r" r# r� � r~ z ForwardHTTPConnection.is_idlec C r� r �rk � is_closedr� r" r" r# r� � r~ z ForwardHTTPConnection.is_closedc C � d| j j� d| �� � d�S �N�<z [z]>�rZ re r� r� r" r" r# �__repr__� � z ForwardHTTPConnection.__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 TunnelHTTPConnection.__init__rn r r
113 c C s |j �di �}|�dd �}| j�� | js�d| jj| jjf }t| jj | jj| jj|d�}t
114 d|fdg| j �}t d|||j d�}| j
115 �|�}|jd k sO|jd
116 krm|j �d d �} | jd
117 dd�}
118 d|j|
119 f } | j
120 �� t| ��|j d } | jd u rzt� n| j}
121 | jr�ddgndg}|
122 �|� |
123 | jj�d
124 �|d�}tdt||��}| jdi |��} | |_W d � n1 s�w Y | �d�}|d uo�|� � dk}|s�| jr�| j s�ddl m } || j| | j d�| _
125 n
126 t!| j| | j d�| _
127 d| _W d � n1 s�w Y | j
128 �|�S ) N�timeout�connects %b:%dro s Host)s Accepts */*s CONNECT)rs rt ru rw �� i+ Z
129 reason_phrase� �ascii�ignore)�errorsz%d %sZnetwork_streamzhttp/1.1�h2)r2 �server_hostnamer� � start_tls�
130 ssl_objectr )�HTTP2Connection)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
131 reason_str�msgry r2 �alpn_protocols�kwargs�tracer� Zhttp2_negotiatedr� r" r" r# rz s� ����
132
133
134 ��
135 � �
136
137 �
138 ��� � Jz#TunnelHTTPConnection.handle_requestr[ c C r{ r r| rc r" r" r# r} Y r~ z'TunnelHTTPConnection.can_handle_requestc C r r r� r� r" r" r# r� \ r� zTunnelHTTPConnection.closer� c C r� r r� r� r" r" r# r� _ r~ zTunnelHTTPConnection.infoc C r� r r� r� r" r" r# r� b r~ z!TunnelHTTPConnection.is_availablec C r� r r� r� r" r" r# r� e r~ z TunnelHTTPConnection.has_expiredc C r� r r� r� r" r" r# r� h r~ z TunnelHTTPConnection.is_idlec C r� r r� r� r" r" r# r� k r~ z TunnelHTTPConnection.is_closedc C r� r� r� r� r" r" r# r� n r� z TunnelHTTPConnection.__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& �
139
140 P
141
142
143
144
145
146 r` )NN)r r r r r r )0�
147 __future__r rS �logging�ssl�typingZ_backends.baser r � _exceptionsr Z_modelsr r r r
148 r r r
149 �_sslr Z_synchronizationr �_tracer �
150 connectionr Zconnection_poolr Zhttp11r Z
151 interfacesr �Unionrx r� Z ByteOrStr�Sequence�TupleZHeadersAsSequence�MappingZHeadersAsMapping� getLoggerr� r) r* r^ r` r" r" r" r# �<module> s2 $
152 �s@