o
    h$!                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZmZmZm	Z	 ddl
mZ ddlmZmZmZ ddlmZmZ ddlmZmZmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlm Z  dZ!e"dZ#de$dee$ fddZ%G dd de Z&dS )    N)TracebackType)IterableIteratorOptionalType   )AutoBackend)SOCKET_OPTIONAsyncNetworkBackendAsyncNetworkStream)ConnectErrorConnectTimeout)OriginRequestResponse)default_ssl_context)	AsyncLock)Trace   )AsyncHTTP11Connection)AsyncConnectionInterfaceg      ?zhttpcore.connectionfactorreturnc                 c   s(    dV  t  D ]	}| d|  V  qdS )z
    Generate a geometric sequence that has a ratio of 2 and starts with 0.

    For example:
    - `factor = 2`: `0, 2, 4, 8, 16, 32, 64, ...`
    - `factor = 3`: `0, 3, 6, 12, 24, 48, 96, ...`
    r   r   N)	itertoolscount)r   n r   Z/var/www/html/flask_server/venv/lib/python3.10/site-packages/httpcore/_async/connection.pyexponential_backoff   s
   r   c                   @   sD  e Zd Z									d.dedeej dee ded	ed
e	dee
 dee
 dee deee  ddfddZdedefddZdedefddZdedefddZd/ddZdefddZdefddZdefdd Zdefd!d"Zde
fd#d$Zde
fd%d&Zd0d'd(Z			d1d)eee  d*ee d+ee ddfd,d-Z dS )2AsyncHTTPConnectionNTFr   originssl_contextkeepalive_expiryhttp1http2retrieslocal_addressudsnetwork_backendsocket_optionsr   c                 C   sb   || _ || _|| _|| _|| _|| _|| _|| _|	d u rt n|	| _	d | _
d| _t | _|
| _d S )NF)_origin_ssl_context_keepalive_expiry_http1_http2_retries_local_address_udsr   _network_backend_connection_connect_failedr   _request_lock_socket_options)selfr    r!   r"   r#   r$   r%   r&   r'   r(   r)   r   r   r   __init__%   s   
zAsyncHTTPConnection.__init__requestc              
      s  |  |jjstd|jj d| j z^| j4 I d H J | jd u r]| |I d H }|d}|d uo9|	 dk}|sB| j
rS| jsSddlm} || j|| jd| _n
t| j|| jd| _W d   I d H  n1 I d H smw   Y  W n ty } zd| _|d }~ww | j|I d H S )	NzAttempted to send request to z on connection to 
ssl_objecth2r   )AsyncHTTP2Connection)r    streamr"   T)can_handle_requesturlr    RuntimeErrorr*   r5   r3   _connectget_extra_infoselected_alpn_protocolr.   r-   r$   r<   r,   r   BaseExceptionr4   handle_async_request)r7   r9   r=   r:   http2_negotiatedr<   excr   r   r   rE   C   sB   




(z(AsyncHTTPConnection.handle_async_requestc                    s  |j di }|j dd }|dd }| j}ttd}	 z| jd u rf| jjd| jj	| j
|| jd}tdt||4 I d H }| jjdi |I d H }	|	|_W d   I d H  n1 I d H s`w   Y  n8| j|| jd	}td
t||4 I d H }| jjdi |I d H }	|	|_W d   I d H  n1 I d H sw   Y  | jjdv r| jd u rt n| j}
| jrddgndg}|
| |
|p| jjd|d}tdt||4 I d H }|	jdi |I d H }	|	|_W d   I d H  |	W S 1 I d H sw   Y  |	W S  ttfyC   |dkr |d8 }t|}tdt||4 I d H }| j|I d H  W d   I d H  n1 I d H s<w   Y  Y nw q)Ntimeoutsni_hostnameconnect)r   Tascii)hostportr&   rH   r)   connect_tcp)pathrH   r)   connect_unix_socket)   httpss   wsszhttp/1.1r;   )r!   server_hostnamerH   	start_tlsr   r   retryr   )
extensionsgetr/   r   RETRIES_BACKOFF_FACTORr1   r*   rL   decoderM   r0   r6   r   loggerr2   rN   return_valuerP   schemer+   r   r.   set_alpn_protocolsrS   r   r   nextsleep)r7   r9   timeoutsrI   rH   retries_leftdelayskwargstracer=   r!   alpn_protocolsdelayr   r   r   rA   g   s~   

(
(


*zAsyncHTTPConnection._connectc                 C   s
   || j kS N)r*   )r7   r    r   r   r   r>      s   
z&AsyncHTTPConnection.can_handle_requestc              	      sh   | j d ur2tdtd i 4 I d H  | j  I d H  W d   I d H  d S 1 I d H s+w   Y  d S d S )Nclose)r3   r   rY   acloser7   r   r   r   rh      s   
.zAsyncHTTPConnection.aclosec                 C   s6   | j d u r| jo| jjdkp| j o| j S | j  S )NrQ   )r3   r.   r*   r[   r-   r4   is_availableri   r   r   r   rj      s   

z AsyncHTTPConnection.is_availablec                 C      | j d u r| jS | j  S rf   )r3   r4   has_expiredri   r   r   r   rl         

zAsyncHTTPConnection.has_expiredc                 C   rk   rf   )r3   r4   is_idleri   r   r   r   rn      rm   zAsyncHTTPConnection.is_idlec                 C   rk   rf   )r3   r4   	is_closedri   r   r   r   ro      rm   zAsyncHTTPConnection.is_closedc                 C   s"   | j d u r| jr
dS dS | j  S )NzCONNECTION FAILED
CONNECTING)r3   r4   infori   r   r   r   rq      s   

zAsyncHTTPConnection.infoc                 C   s   d| j j d|   dS )N<z [z]>)	__class____name__rq   ri   r   r   r   __repr__   s   zAsyncHTTPConnection.__repr__c                    s   | S rf   r   ri   r   r   r   
__aenter__   s   zAsyncHTTPConnection.__aenter__exc_type	exc_value	tracebackc                    s   |   I d H  d S rf   )rh   )r7   rw   rx   ry   r   r   r   	__aexit__   s   zAsyncHTTPConnection.__aexit__)	NNTFr   NNNN)r   N)r   r   )NNN)!rt   
__module____qualname__r   r   ssl
SSLContextfloatboolintstrr
   r   r	   r8   r   r   rE   r   rA   r>   rh   rj   rl   rn   ro   rq   ru   rv   r   rD   r   rz   r   r   r   r   r   $   sp    	


$>


r   )'r   loggingr}   typesr   typingr   r   r   r   _backends.autor   _backends.baser	   r
   r   _exceptionsr   r   _modelsr   r   r   _sslr   _synchronizationr   _tracer   http11r   
interfacesr   rW   	getLoggerrY   r   r   r   r   r   r   r   <module>   s$    
