o
    h6                     @   s"  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	m
Z
mZmZmZmZ d dlZddlmZ ddlmZmZmZmZmZ ddlmZmZmZ ddlmZmZ dd	lm Z  d
dl!m"Z" e#dZ$eejej%ej&f Z'G dd de j(Z)G dd de"Z*G dd dZ+G dd deZ,dS )    N)TracebackType)AnyAsyncIterableAsyncIteratorListOptionalTupleTypeUnion   )AsyncNetworkStream)ConnectionNotAvailableLocalProtocolErrorRemoteProtocolError
WriteErrormap_exceptions)OriginRequestResponse)	AsyncLockAsyncShieldCancellation)Trace   )AsyncConnectionInterfacezhttpcore.http11c                   @   s   e Zd ZdZdZdZdZdS )HTTPConnectionStater   r   r      N)__name__
__module____qualname__NEWACTIVEIDLECLOSED r#   r#   V/var/www/html/flask_server/venv/lib/python3.10/site-packages/httpcore/_async/http11.pyr   +   s
    r   c                   @   s  e Zd ZdZdZ	d4dededee ddfdd	Z	d
e
defddZd
e
ddfddZd
e
ddfddZ	d4dejdee ddfddZd
e
deeeeeeeef  ef fddZd
e
dee fddZ	d4dee deejeej f fddZd5ddZd5ddZde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&de%fd+d,Z'd6d-d.Z(			d7d/eee)  d0ee) d1ee* ddfd2d3Z+dS )8AsyncHTTP11Connectioni   i  Noriginstreamkeepalive_expiryreturnc                 C   sF   || _ || _|| _d | _tj| _t | _d| _	t
jt
j| jd| _d S )Nr   )our_rolemax_incomplete_event_size)_origin_network_stream_keepalive_expiry
_expire_atr   r   _stater   _state_lock_request_counth11
ConnectionCLIENTMAX_INCOMPLETE_EVENT_SIZE
_h11_state)selfr&   r'   r(   r#   r#   r$   __init__6   s   zAsyncHTTP11Connection.__init__requestc                    s  |  |jjstd|jj d| j | j4 I d H & | jtjtj	fv r5|  j
d7  _
tj| _d | _nt W d   I d H  n1 I d H sHw   Y  zd|i}zZtdt||4 I d H }| jdi |I d H  W d   I d H  n1 I d H szw   Y  tdt||4 I d H }| jdi |I d H  W d   I d H  n1 I d H sw   Y  W n	 ty   Y nw tdt||4 I d H "}| jdi |I d H \}}}}}||||f|_W d   I d H  n1 I d H sw   Y  | j}	|dks
|jd	krd
|  krdk rn nt|	|}	t||t| ||||	ddW S  tyr }
 zGt 9 tdt|4 I d H }|  I d H  W d   I d H  n1 I d H sNw   Y  W d    |
W d    |
1 sgw   Y  |
d }
~
ww )NzAttempted to send request to z on connection to r   r:   send_request_headerssend_request_bodyreceive_response_headerse   s   CONNECT   i,  )http_versionreason_phrasenetwork_stream)statusheaderscontent
extensionsresponse_closedr#   )can_handle_requesturlr&   RuntimeErrorr,   r1   r0   r   r   r!   r2   r    r/   r   r   logger_send_request_headers_send_request_bodyr   _receive_response_headersreturn_valuer-   methodAsyncHTTP11UpgradeStreamr   HTTP11ConnectionByteStreamBaseExceptionr   _response_closed)r8   r:   kwargstracer@   rC   rA   rD   trailing_datarB   excr#   r#   r$   handle_async_requestH   s   ((((
(

,

z*AsyncHTTP11Connection.handle_async_requestc                    sz   |j di }|dd }ttjti tj|j|jj|j	d}W d    n1 s,w   Y  | j
||dI d H  d S )Ntimeoutwrite)rP   targetrD   rZ   )rF   getr   r3   r   r   rP   rI   r\   rD   _send_eventr8   r:   timeoutsrZ   eventr#   r#   r$   rL      s   z+AsyncHTTP11Connection._send_request_headersc                    s~   |j di }|dd }t|jtsJ |j2 z3 d H W }tj|d}| j||dI d H  q6 | jt |dI d H  d S )NrZ   r[   )datar]   )	rF   r^   
isinstancer'   r   r3   Datar_   EndOfMessage)r8   r:   ra   rZ   chunkrb   r#   r#   r$   rM      s   z(AsyncHTTP11Connection._send_request_bodyrb   rZ   c                    s4   | j |}|d ur| jj||dI d H  d S d S )Nr]   )r7   sendr-   r[   )r8   rb   rZ   bytes_to_sendr#   r#   r$   r_      s
   z!AsyncHTTP11Connection._send_eventc           	         s   |j di }|dd }	 | j|dI d H }t|tjrnt|tjr+|jdkr+nqd|j }|j	
 }| jj\}}||j|j||fS )NrZ   readTr]   r>   s   HTTP/)rF   r^   _receive_eventrd   r3   r   InformationalResponsestatus_coder@   rD   	raw_itemsr7   rW   reason)	r8   r:   ra   rZ   rb   r@   rD   rW   _r#   r#   r$   rN      s    




z/AsyncHTTP11Connection._receive_response_headersc                 C  sb   |j di }|dd }	 | j|dI d H }t|tjr%t|jV  nt|tjtj	fr0d S q)NrZ   rj   Tr]   )
rF   r^   rk   rd   r3   re   bytesrc   rf   PAUSEDr`   r#   r#   r$   _receive_response_body   s   z,AsyncHTTP11Connection._receive_response_bodyc                    s   	 t tjti | j }W d    n1 sw   Y  |tju rG| jj| j|dI d H }|dkr@| jj	tj
kr@d}t|| j| n|S q)NTr]       z/Server disconnected without sending a response.)r   r3   r   r7   
next_event	NEED_DATAr-   rj   READ_NUM_BYTEStheir_stateSEND_RESPONSEreceive_data)r8   rZ   rb   rc   msgr#   r#   r$   rk      s   
z$AsyncHTTP11Connection._receive_eventc              	      s   | j 4 I d H E | jjtju r0| jjtju r0tj| _| j	  | j
d ur/t }|| j
 | _n|  I d H  W d   I d H  d S W d   I d H  d S 1 I d H sSw   Y  d S N)r1   r7   	our_stater3   DONErx   r   r!   r0   start_next_cycler.   time	monotonicr/   aclose)r8   nowr#   r#   r$   rT      s   

. z&AsyncHTTP11Connection._response_closedc                    s   t j| _| j I d H  d S r|   )r   r"   r0   r-   r   r8   r#   r#   r$   r     s   zAsyncHTTP11Connection.aclosec                 C   s
   || j kS r|   )r,   r8   r&   r#   r#   r$   rH     s   
z(AsyncHTTP11Connection.can_handle_requestc                 C      | j tjkS r|   r0   r   r!   r   r#   r#   r$   is_available  s   z"AsyncHTTP11Connection.is_availablec                 C   s<   t  }| jd uo|| jk}| jtjko| jd}|p|S )Nis_readable)r   r   r/   r0   r   r!   r-   get_extra_info)r8   r   keepalive_expiredserver_disconnectedr#   r#   r$   has_expired  s   
z!AsyncHTTP11Connection.has_expiredc                 C   r   r|   r   r   r#   r#   r$   is_idle'     zAsyncHTTP11Connection.is_idlec                 C   r   r|   )r0   r   r"   r   r#   r#   r$   	is_closed*  r   zAsyncHTTP11Connection.is_closedc                 C   s$   t | j}|d| jj d| j S )Nz, HTTP/1.1, , Request Count: )strr,   r0   namer2   r   r#   r#   r$   info-  s
   
zAsyncHTTP11Connection.infoc              	   C   s6   | j j}t| j}d| d|d| jj d| j d	S )N<z [z, r   z]>)	__class__r   r   r,   r0   r   r2   )r8   
class_namer&   r#   r#   r$   __repr__4  s   
zAsyncHTTP11Connection.__repr__c                    s   | S r|   r#   r   r#   r#   r$   
__aenter__?  s   z AsyncHTTP11Connection.__aenter__exc_type	exc_value	tracebackc                    s   |   I d H  d S r|   )r   )r8   r   r   r   r#   r#   r$   	__aexit__B  s   zAsyncHTTP11Connection.__aexit__r|   r)   N)r)   r%   )NNN),r   r   r   rw   r6   r   r   r   floatr9   r   r   rY   rL   rM   r3   Eventr_   r   rq   intr   rN   r   rs   r
   r	   rr   rk   rT   r   boolrH   r   r   r   r   r   r   r   r   rS   r   r   r#   r#   r#   r$   r%   2   st    
K
	






r%   c                   @   s>   e Zd ZdededdfddZdee fddZdd	d
Z	dS )rR   
connectionr:   r)   Nc                 C   s   || _ || _d| _d S )NF)_connection_request_closed)r8   r   r:   r#   r#   r$   r9   L  s   
z#HTTP11ConnectionByteStream.__init__c                 C  s   d| j i}z;tdt| j |4 I d H ! | jjdi |2 z	3 d H W }|V  q6 W d   I d H  W d S 1 I d H s:w   Y  W d S  tyj } zt  |  I d H  W d    |1 s_w   Y  |d }~ww )Nr:   receive_response_bodyr#   )r   r   rK   r   rs   rS   r   r   )r8   rU   rg   rX   r#   r#   r$   	__aiter__Q  s"   
2
z$HTTP11ConnectionByteStream.__aiter__c              	      sj   | j s3d| _ tdt| j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 )NTrG   )r   r   rK   r   r   rT   r   r#   r#   r$   r   _  s   .z!HTTP11ConnectionByteStream.acloser   )
r   r   r   r%   r   r9   r   rq   r   r   r#   r#   r#   r$   rR   K  s    rR   c                	   @   s   e Zd ZdededdfddZddedee defd	d
Z	ddedee ddfddZ
dddZ		ddejdee dee defddZdedefddZdS )rQ   r'   leading_datar)   Nc                 C   s   || _ || _d S r|   )_stream_leading_data)r8   r'   r   r#   r#   r$   r9   g  s   
z!AsyncHTTP11UpgradeStream.__init__	max_bytesrZ   c                    s>   | j r| j d | }| j |d  | _ |S | j||I d H S r|   )r   r   rj   )r8   r   rZ   bufferr#   r#   r$   rj   k  s   zAsyncHTTP11UpgradeStream.readr   c                    s   | j ||I d H  d S r|   )r   r[   )r8   r   rZ   r#   r#   r$   r[   s  s   zAsyncHTTP11UpgradeStream.writec                    s   | j  I d H  d S r|   )r   r   r   r#   r#   r$   r   v  s   zAsyncHTTP11UpgradeStream.aclosessl_contextserver_hostnamec                    s   | j |||I d H S r|   )r   	start_tls)r8   r   r   rZ   r#   r#   r$   r   y  s   z"AsyncHTTP11UpgradeStream.start_tlsr   c                 C   s   | j |S r|   )r   r   )r8   r   r#   r#   r$   r     r   z'AsyncHTTP11UpgradeStream.get_extra_infor|   r   )NN)r   r   r   r   rq   r9   r   r   r   rj   r[   r   ssl
SSLContextr   r   r   r   r#   r#   r#   r$   rQ   f  s"    

rQ   )-enumloggingr   r   typesr   typingr   r   r   r   r   r   r	   r
   r3   _backends.baser   _exceptionsr   r   r   r   r   _modelsr   r   r   _synchronizationr   r   _tracer   
interfacesr   	getLoggerrK   re   rf   H11SendEventIntEnumr   r%   rR   rQ   r#   r#   r#   r$   <module>   s4    (
  