o
    NEg8                     @   s  d dl mZ d dlmZmZmZmZmZ d dlZd dl	m
Z
 d dlmZ ddlmZ ddlmZ erNd d	lmZ dd
lmZ edddZG dd dee ZdZG dd dZejddddG dd dZejddddG dd dZejddddG dd dZejddddG dd dZejddddG dd dZejddddG dd dZejddddG d d! d!Z ejddddG d"d# d#Z!ejddddG d$d% d%Z"ejddddG d&d' d'Z#ejddddG d(d) d)Z$ejddddG d*d+ d+Z%ejddddG d,d- d-Z&ejddddG d.d/ d/Z'ejddddG d0d1 d1Z(G d2d3 d3Z)dS )4    )SimpleNamespace)TYPE_CHECKING	AwaitableOptionalTypeTypeVarN)CIMultiDict)URL   )ClientResponse)Signal)Protocol)ClientSession_ParamT_contraT)contravariantc                	   @   s*   e Zd Zdedededed fddZdS )_SignalCallback_SignalCallback__client_session!_SignalCallback__trace_config_ctx_SignalCallback__paramsreturnNc                 C   s   d S N )selfr   r   r   r   r   i/var/www/eduai.edurigo.com/doc_train/edurigo_ai/Puru/venv/lib/python3.10/site-packages/aiohttp/tracing.py__call__   s   z_SignalCallback.__call__)__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r      s    r   )TraceConfigTraceRequestStartParamsTraceRequestEndParamsTraceRequestExceptionParams TraceConnectionQueuedStartParamsTraceConnectionQueuedEndParams TraceConnectionCreateStartParamsTraceConnectionCreateEndParamsTraceConnectionReuseconnParamsTraceDnsResolveHostStartParamsTraceDnsResolveHostEndParamsTraceDnsCacheHitParamsTraceDnsCacheMissParamsTraceRequestRedirectParamsTraceRequestChunkSentParams TraceResponseChunkReceivedParamsc                   @   sL  e Zd ZdZefdee ddfddZ	d9dee defdd	Zd:d
dZ	e
d;ddZe
		d<ddZe
		d=ddZe
d>ddZe
		d?ddZe
		d@ddZe
		dAdd Ze
		!dBd"d#Ze
		$dCd%d&Ze
		'dDd(d)Ze
		*dEd+d,Ze
		-dFd.d/Ze
		0dGd1d2Ze
dHd4d5Ze
dId7d8ZdS )Jr   zJFirst-class used to trace requests launched via ClientSession
    objects.trace_config_ctx_factoryr   Nc                 C   s   t | | _t | | _t | | _t | | _t | | _t | | _t | | _t | | _t | | _	t | | _
t | | _t | | _t | | _t | | _t | | _|| _d S r   )r   _on_request_start_on_request_chunk_sent_on_response_chunk_received_on_request_end_on_request_exception_on_request_redirect_on_connection_queued_start_on_connection_queued_end_on_connection_create_start_on_connection_create_end_on_connection_reuseconn_on_dns_resolvehost_start_on_dns_resolvehost_end_on_dns_cache_hit_on_dns_cache_miss_trace_config_ctx_factory)r   r.   r   r   r   __init__4   s\   
zTraceConfig.__init__trace_request_ctxc                 C   s   | j |dS )z( Return a new trace_config_ctx instance )r@   )r>   )r   r@   r   r   r   trace_config_ctxg   s   zTraceConfig.trace_config_ctxc                 C   s   | j   | j  | j  | j  | j  | j  | j  | j  | j	  | j
  | j  | j  | j  | j  | j  d S r   )r/   freezer0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r   r   r   r   rB   m   s   













zTraceConfig.freeze0Signal[_SignalCallback[TraceRequestStartParams]]c                 C      | j S r   )r/   rC   r   r   r   on_request_start~      zTraceConfig.on_request_start4Signal[_SignalCallback[TraceRequestChunkSentParams]]c                 C   rE   r   )r0   rC   r   r   r   on_request_chunk_sent      z!TraceConfig.on_request_chunk_sent9Signal[_SignalCallback[TraceResponseChunkReceivedParams]]c                 C   rE   r   )r1   rC   r   r   r   on_response_chunk_received   rJ   z&TraceConfig.on_response_chunk_received.Signal[_SignalCallback[TraceRequestEndParams]]c                 C   rE   r   )r2   rC   r   r   r   on_request_end   rG   zTraceConfig.on_request_end4Signal[_SignalCallback[TraceRequestExceptionParams]]c                 C   rE   r   )r3   rC   r   r   r   on_request_exception   rJ   z TraceConfig.on_request_exception3Signal[_SignalCallback[TraceRequestRedirectParams]]c                 C   rE   r   )r4   rC   r   r   r   on_request_redirect   rJ   zTraceConfig.on_request_redirect9Signal[_SignalCallback[TraceConnectionQueuedStartParams]]c                 C   rE   r   )r5   rC   r   r   r   on_connection_queued_start   rJ   z&TraceConfig.on_connection_queued_start7Signal[_SignalCallback[TraceConnectionQueuedEndParams]]c                 C   rE   r   )r6   rC   r   r   r   on_connection_queued_end   rJ   z$TraceConfig.on_connection_queued_end9Signal[_SignalCallback[TraceConnectionCreateStartParams]]c                 C   rE   r   )r7   rC   r   r   r   on_connection_create_start   rJ   z&TraceConfig.on_connection_create_start7Signal[_SignalCallback[TraceConnectionCreateEndParams]]c                 C   rE   r   )r8   rC   r   r   r   on_connection_create_end   rJ   z$TraceConfig.on_connection_create_end7Signal[_SignalCallback[TraceConnectionReuseconnParams]]c                 C   rE   r   )r9   rC   r   r   r   on_connection_reuseconn   rJ   z#TraceConfig.on_connection_reuseconn7Signal[_SignalCallback[TraceDnsResolveHostStartParams]]c                 C   rE   r   )r:   rC   r   r   r   on_dns_resolvehost_start   rJ   z$TraceConfig.on_dns_resolvehost_start5Signal[_SignalCallback[TraceDnsResolveHostEndParams]]c                 C   rE   r   )r;   rC   r   r   r   on_dns_resolvehost_end   rJ   z"TraceConfig.on_dns_resolvehost_end/Signal[_SignalCallback[TraceDnsCacheHitParams]]c                 C   rE   r   )r<   rC   r   r   r   on_dns_cache_hit   rG   zTraceConfig.on_dns_cache_hit0Signal[_SignalCallback[TraceDnsCacheMissParams]]c                 C   rE   r   )r=   rC   r   r   r   on_dns_cache_miss   rG   zTraceConfig.on_dns_cache_missr   r   N)r   rD   )r   rH   )r   rK   )r   rM   )r   rO   )r   rQ   )r   rS   )r   rU   )r   rW   )r   rY   )r   r[   )r   r]   )r   r_   )r   ra   )r   rc   )r   r   r   __doc__r   r   r?   r   rA   rB   propertyrF   rI   rL   rN   rP   rR   rT   rV   rX   rZ   r\   r^   r`   rb   rd   r   r   r   r   r   0   s    
4

r   )auto_attribsfrozenslotsc                   @   s*   e Zd ZU dZeed< eed< ded< dS )r   z1 Parameters sent by the `on_request_start` signalmethodurlCIMultiDict[str]headersN)r   r   r   rf   str__annotations__r	   r   r   r   r   r      
   
 r   c                   @   *   e Zd ZU dZeed< eed< eed< dS )r,   z6 Parameters sent by the `on_request_chunk_sent` signalrk   rl   chunkNr   r   r   rf   ro   rp   r	   bytesr   r   r   r   r,      rq   r,   c                   @   rr   )r-   z; Parameters sent by the `on_response_chunk_received` signalrk   rl   rs   Nrt   r   r   r   r   r-      rq   r-   c                   @   2   e Zd ZU dZeed< eed< ded< eed< dS )r    z/ Parameters sent by the `on_request_end` signalrk   rl   rm   rn   responseNr   r   r   rf   ro   rp   r	   r   r   r   r   r   r          
 r    c                   @   rv   )r!   z5 Parameters sent by the `on_request_exception` signalrk   rl   rm   rn   	exceptionN)r   r   r   rf   ro   rp   r	   BaseExceptionr   r   r   r   r!      ry   r!   c                   @   rv   )r+   z4 Parameters sent by the `on_request_redirect` signalrk   rl   rm   rn   rw   Nrx   r   r   r   r   r+      ry   r+   c                   @      e Zd ZdZdS )r"   z; Parameters sent by the `on_connection_queued_start` signalNr   r   r   rf   r   r   r   r   r"   
      r"   c                   @   r|   )r#   z9 Parameters sent by the `on_connection_queued_end` signalNr}   r   r   r   r   r#     r~   r#   c                   @   r|   )r$   z; Parameters sent by the `on_connection_create_start` signalNr}   r   r   r   r   r$     r~   r$   c                   @   r|   )r%   z9 Parameters sent by the `on_connection_create_end` signalNr}   r   r   r   r   r%     r~   r%   c                   @   r|   )r&   z8 Parameters sent by the `on_connection_reuseconn` signalNr}   r   r   r   r   r&     r~   r&   c                   @      e Zd ZU dZeed< dS )r'   z9 Parameters sent by the `on_dns_resolvehost_start` signalhostNr   r   r   rf   ro   rp   r   r   r   r   r'   #     
 r'   c                   @   r   )r(   z7 Parameters sent by the `on_dns_resolvehost_end` signalr   Nr   r   r   r   r   r(   *  r   r(   c                   @   r   )r)   z1 Parameters sent by the `on_dns_cache_hit` signalr   Nr   r   r   r   r   r)   1  r   r)   c                   @   r   )r*   z2 Parameters sent by the `on_dns_cache_miss` signalr   Nr   r   r   r   r   r*   8  r   r*   c                
   @   sL  e Zd ZdZdddededdfdd	Zd
ededdddfddZ	d
edede
ddfddZd
edede
ddfddZd
ededddeddf
ddZd
ededddeddf
ddZd
ededddeddf
ddZd0ddZd0dd Zd0d!d"Zd0d#d$Zd0d%d&Zd'eddfd(d)Zd'eddfd*d+Zd'eddfd,d-Zd'eddfd.d/ZdS )1TracezcInternal class used to keep together the main dependencies used
    at the moment of send a signal.sessionr   trace_configrA   r   Nc                 C   s   || _ || _|| _d S r   )_trace_config_trace_config_ctx_session)r   r   r   rA   r   r   r   r?   C  s   
zTrace.__init__rk   rl   rn   rm   c                    &   | j j| j| jt|||I d H S r   )r   rF   sendr   r   r   )r   rk   rl   rn   r   r   r   send_request_startM     

zTrace.send_request_startrs   c                    r   r   )r   rI   r   r   r   r,   r   rk   rl   rs   r   r   r   send_request_chunk_sentV  r   zTrace.send_request_chunk_sentc                    r   r   )r   rL   r   r   r   r-   r   r   r   r   send_response_chunk_received_  r   z"Trace.send_response_chunk_receivedrw   c              	      (   | j j| j| jt||||I d H S r   )r   rN   r   r   r   r    r   rk   rl   rn   rw   r   r   r   send_request_endh     
zTrace.send_request_endrz   c              	      r   r   )r   rP   r   r   r   r!   )r   rk   rl   rn   rz   r   r   r   send_request_exceptionu  r   zTrace.send_request_exceptionc              	      r   r   )r   r4   r   r   r   r+   r   r   r   r   send_request_redirect  r   zTrace.send_request_redirectc                        | j j| j| jt I d H S r   )r   rT   r   r   r   r"   rC   r   r   r   send_connection_queued_start     
z"Trace.send_connection_queued_startc                    r   r   )r   rV   r   r   r   r#   rC   r   r   r   send_connection_queued_end  r   z Trace.send_connection_queued_endc                    r   r   )r   rX   r   r   r   r$   rC   r   r   r   send_connection_create_start  r   z"Trace.send_connection_create_startc                    r   r   )r   rZ   r   r   r   r%   rC   r   r   r   send_connection_create_end  r   z Trace.send_connection_create_endc                    r   r   )r   r\   r   r   r   r&   rC   r   r   r   send_connection_reuseconn  r   zTrace.send_connection_reuseconnr   c                    "   | j j| j| jt|I d H S r   )r   r^   r   r   r   r'   r   r   r   r   r   send_dns_resolvehost_start     
z Trace.send_dns_resolvehost_startc                    r   r   )r   r`   r   r   r   r(   r   r   r   r   send_dns_resolvehost_end  r   zTrace.send_dns_resolvehost_endc                    r   r   )r   rb   r   r   r   r)   r   r   r   r   send_dns_cache_hit  r   zTrace.send_dns_cache_hitc                    r   r   )r   rd   r   r   r   r*   r   r   r   r   send_dns_cache_miss  r   zTrace.send_dns_cache_missre   )r   r   r   rf   r   r   r?   ro   r	   r   ru   r   r   r   r   r{   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   ?  s    


	
	
	







r   )*typesr   typingr   r   r   r   r   attr	multidictr   yarlr	   client_reqrepr   signalsr   typing_extensionsr   clientr   r   r   __all__r   sr   r,   r-   r    r!   r+   r"   r#   r$   r%   r&   r'   r(   r)   r*   r   r   r   r   r   <module>   s\    
 "			