o
    MK&h.                      @  s   d dl mZ d dlZd dlZ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 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ddZ!G dd deZ"dS )    )annotationsN   )SyncBackend)SOCKET_OPTIONNetworkBackendNetworkStream)ConnectErrorConnectTimeout)OriginRequestResponse)default_ssl_context)Lock)Trace   )HTTP11Connection)ConnectionInterfaceg      ?zhttpcore.connectionfactorfloatreturntyping.Iterator[float]c                 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   L/var/www/html/venv/lib/python3.10/site-packages/httpcore/_sync/connection.pyexponential_backoff   s
   r   c                   @  s   e Zd Z									d>d?ddZd@ddZdAd!d"ZdBd#d$ZdCd%d&ZdDd'd(ZdDd)d*Z	dDd+d,Z
dDd-d.ZdEd0d1ZdEd2d3ZdFd4d5Z			dGdHd<d=ZdS )IHTTPConnectionNTFr   originr
   ssl_contextssl.SSLContext | Nonekeepalive_expiryfloat | Nonehttp1boolhttp2retriesintlocal_address
str | Noneudsnetwork_backendNetworkBackend | Nonesocket_options%typing.Iterable[SOCKET_OPTION] | Noner   Nonec                 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HTTPConnection.__init__requestr   r   c              
   C  s   |  |jjstd|jj d| j zQ| jD | jd u rU| |}|d}|d uo1|	 dk}|s:| j
rK| jsKddlm} || j|| jd| _n
t| j|| jd| _W d    n1 s_w   Y  W n tyv } zd| _|d }~ww | j|S )	NzAttempted to send request to z on connection to 
ssl_objecth2r   )HTTP2Connection)r   streamr!   T)can_handle_requesturlr   RuntimeErrorr0   r;   r9   _connectget_extra_infoselected_alpn_protocolr4   r3   r%   rB   r2   r   BaseExceptionr:   handle_request)r=   r?   rC   r@   http2_negotiatedrB   excr   r   r   rK   E   sB   




zHTTPConnection.handle_requestr   c                 C  s  |j di }|j dd }|dd }| j}ttd}	 z| jd u rX| jjd| jj	| j
|| jd}tdt||}| jjdi |}	|	|_W d    n1 sRw   Y  n+| j|| jd	}td
t||}| jjdi |}	|	|_W d    n1 s~w   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||}|	jdi |}	|	|_W d    |	W S 1 sw   Y  |	W S  ttfy   |dkr |d8 }t|}tdt||}| j| W d    n	1 sw   Y  Y nw q)Ntimeoutsni_hostnameconnect)r   Tascii)hostportr(   rN   r-   connect_tcp)pathrN   r-   connect_unix_socket)   httpss   wsszhttp/1.1rA   )r   server_hostnamerN   	start_tlsr   r   retryr   )
extensionsgetr5   r   RETRIES_BACKOFF_FACTORr7   r0   rR   decoderS   r6   r<   r   loggerr8   rT   return_valuerV   schemer1   r   r4   set_alpn_protocolsrY   r   r	   nextsleep)r=   r?   timeoutsrO   rN   retries_leftdelayskwargstracerC   r   alpn_protocolsdelayr   r   r   rG   i   s|   





zHTTPConnection._connectc                 C  s
   || j kS N)r0   )r=   r   r   r   r   rD      s   
z!HTTPConnection.can_handle_requestc                 C  sL   | j d ur$tdtd i  | j   W d    d S 1 sw   Y  d S d S )Nclose)r9   r   r_   rm   r=   r   r   r   rm      s
   
"zHTTPConnection.closec                 C  s6   | j d u r| jo| jjdkp| j o| j S | j  S )NrW   )r9   r4   r0   ra   r3   r:   is_availablern   r   r   r   ro      s   

zHTTPConnection.is_availablec                 C     | j d u r| jS | j  S rl   )r9   r:   has_expiredrn   r   r   r   rq         

zHTTPConnection.has_expiredc                 C  rp   rl   )r9   r:   is_idlern   r   r   r   rs      rr   zHTTPConnection.is_idlec                 C  rp   rl   )r9   r:   	is_closedrn   r   r   r   rt      rr   zHTTPConnection.is_closedstrc                 C  s"   | j d u r| jr
dS dS | j  S )NzCONNECTION FAILED
CONNECTING)r9   r:   inforn   r   r   r   rw      s   

zHTTPConnection.infoc                 C  s   d| j j d|   dS )N<z [z]>)	__class____name__rw   rn   r   r   r   __repr__   s   zHTTPConnection.__repr__c                 C  s   | S rl   r   rn   r   r   r   	__enter__   s   zHTTPConnection.__enter__exc_typetype[BaseException] | None	exc_valueBaseException | None	tracebacktypes.TracebackType | Nonec                 C  s   |    d S rl   )rm   )r=   r}   r   r   r   r   r   __exit__   s   zHTTPConnection.__exit__)	NNTFr   NNNN)r   r
   r   r    r!   r"   r#   r$   r%   r$   r&   r'   r(   r)   r*   r)   r+   r,   r-   r.   r   r/   )r?   r   r   r   )r?   r   r   r   )r   r
   r   r$   )r   r/   )r   r$   )r   ru   )r   r   )NNN)r}   r~   r   r   r   r   r   r/   )rz   
__module____qualname__r>   rK   rG   rD   rm   ro   rq   rs   rt   rw   r{   r|   r   r   r   r   r   r   &   s4    

$
>







r   )r   r   r   r   )#
__future__r   r   loggingssltypestyping_backends.syncr   _backends.baser   r   r   _exceptionsr   r	   _modelsr
   r   r   _sslr   _synchronizationr   _tracer   http11r   
interfacesr   r]   	getLoggerr_   r   r   r   r   r   r   <module>   s&    

