o
    i.                      @  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   S/home/kim/smarthome/.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   _connectZget_extra_infoselected_alpn_protocolr4   r3   r%   rB   r2   r   BaseExceptionr:   handle_request)r=   r?   rC   r@   Zhttp2_negotiatedrB   excr   r   r   rJ   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(   rL   r-   connect_tcp)pathrL   r-   connect_unix_socket)   httpss   wsszhttp/1.1rA   )r   server_hostnamerL   	start_tlsr   r   retryr   )
extensionsgetr5   r   RETRIES_BACKOFF_FACTORr7   r0   rP   decoderQ   r6   r<   r   loggerr8   rR   return_valuerT   schemer1   r   r4   set_alpn_protocolsrW   r   r	   nextsleep)r=   r?   ZtimeoutsrM   rL   Zretries_leftZ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]   rh   r=   r   r   r   rh      s
   
"zHTTPConnection.closec                 C  s6   | j d u r| jo| jjdkp| j o| j S | j  S )NrU   )r9   r4   r0   r_   r3   r:   is_availableri   r   r   r   rj      s   

zHTTPConnection.is_availablec                 C     | j d u r| jS | j  S rg   )r9   r:   has_expiredri   r   r   r   rl         

zHTTPConnection.has_expiredc                 C  rk   rg   )r9   r:   is_idleri   r   r   r   rn      rm   zHTTPConnection.is_idlec                 C  rk   rg   )r9   r:   	is_closedri   r   r   r   ro      rm   zHTTPConnection.is_closedstrc                 C  s"   | j d u r| jr
dS dS | j  S )NzCONNECTION FAILEDZ
CONNECTING)r9   r:   infori   r   r   r   rq      s   

zHTTPConnection.infoc                 C  s   d| j j d|   dS )N<z [z]>)	__class____name__rq   ri   r   r   r   __repr__   s   zHTTPConnection.__repr__c                 C  s   | S rg   r   ri   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 rg   )rh   )r=   rw   ry   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   rp   )r   r   )NNN)rw   rx   ry   rz   r{   r|   r   r/   )rt   
__module____qualname__r>   rJ   rG   rD   rh   rj   rl   rn   ro   rq   ru   rv   r}   r   r   r   r   r   &   s4    

$
>







r   )r   r   r   r   )#
__future__r   r   loggingssltypestypingZ_backends.syncr   Z_backends.baser   r   r   _exceptionsr   r	   Z_modelsr
   r   r   _sslr   Z_synchronizationr   _tracer   Zhttp11r   Z
interfacesr   r[   	getLoggerr]   r   r   r   r   r   r   <module>   s&    

