
    tni8                        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rdd	lmZ  ed
d          Ze
eeef         ZdZ G d d          Z ej        ddd           G d d                      Z ej        ddd           G d d                      Z ej        ddd           G d d                      Z ej        ddd           G d d                      Z ej        ddd           G d d                      Z ej        ddd           G d d                      Z ej        ddd           G d d                      Z ej        ddd           G d d                       Z ej        ddd           G d! d"                      Z  ej        ddd           G d# d$                      Z! ej        ddd           G d% d&                      Z" ej        ddd           G d' d(                      Z# ej        ddd           G d) d*                      Z$ ej        ddd           G d+ d,                      Z% ej        ddd           G d- d.                      Z& ej        ddd           G d/ d0                      Z' G d1 d2          Z(dS )3    )SimpleNamespace)TYPE_CHECKINGMappingOptionalTypeTypeVarN)Signal)CIMultiDict)URL   )ClientResponse)ClientSession_ParamT_contraT)contravariant)TraceConfigTraceRequestStartParamsTraceRequestEndParamsTraceRequestExceptionParams TraceConnectionQueuedStartParamsTraceConnectionQueuedEndParams TraceConnectionCreateStartParamsTraceConnectionCreateEndParamsTraceConnectionReuseconnParamsTraceDnsResolveHostStartParamsTraceDnsResolveHostEndParamsTraceDnsCacheHitParamsTraceDnsCacheMissParamsTraceRequestRedirectParamsTraceRequestChunkSentParams TraceResponseChunkReceivedParamsTraceRequestHeadersSentParamsc                      e Zd ZdZefdee         ddfdZ	 d)deee	e	f                  defdZ
d*dZed+d
            Ze	 	 d,d            Ze	 	 d-d            Zed.d            Ze	 	 d/d            Ze	 	 d0d            Ze	 	 d1d            Ze	 	 d2d            Ze	 	 d3d            Ze	 	 d4d            Ze	 	 d5d            Ze	 	 d6d             Ze	 	 d7d"            Zed8d$            Zed9d&            Ze	 	 d:d(            ZdS );r   zFFirst-class used to trace requests launched via ClientSession objects.trace_config_ctx_factoryreturnNc                    t          |           | _        t          |           | _        t          |           | _        t          |           | _        t          |           | _        t          |           | _        t          |           | _        t          |           | _        t          |           | _	        t          |           | _
        t          |           | _        t          |           | _        t          |           | _        t          |           | _        t          |           | _        t          |           | _        || _        d S N)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_on_request_headers_sent_trace_config_ctx_factory)selfr#   s     L/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/aiohttp/tracing.py__init__zTraceConfig.__init__*   s@    KQQU,,4LL 	#
 4LL 	( GMTll4LL 	" QWQ
 Q
!
 4LL 	(
 4LL 	&
 4LL 	(
 4LL 	&
 4LL 	%
 4LL 	& 4LL 	$ JPPTKQRV<<4LL 	% *B&&&    trace_request_ctxc                 .    |                      |          S )z&Return a new trace_config_ctx instance)r<   )r7   )r8   r<   s     r9   trace_config_ctxzTraceConfig.trace_config_ctxX   s     --@Q-RRRr;   c                 &   | j                                          | j                                         | j                                         | j                                         | j                                         | j                                         | j                                         | j                                         | j	                                         | j
                                         | j                                         | j                                         | j                                         | j                                         | j                                         | j                                         d S r&   )r'   freezer(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r8   s    r9   r@   zTraceConfig.freeze^   sX   %%'''#**,,,(//111##%%%"))+++!((***(//111&--///(//111&--///%,,...&--///$++---%%'''&&(((%,,.....r;   '_TracingSignal[TraceRequestStartParams]c                     | j         S r&   )r'   rA   s    r9   on_request_startzTraceConfig.on_request_startp       %%r;   +_TracingSignal[TraceRequestChunkSentParams]c                     | j         S r&   )r(   rA   s    r9   on_request_chunk_sentz!TraceConfig.on_request_chunk_sentt   s     **r;   0_TracingSignal[TraceResponseChunkReceivedParams]c                     | j         S r&   )r)   rA   s    r9   on_response_chunk_receivedz&TraceConfig.on_response_chunk_receivedz        //r;   %_TracingSignal[TraceRequestEndParams]c                     | j         S r&   )r*   rA   s    r9   on_request_endzTraceConfig.on_request_end   s    ##r;   +_TracingSignal[TraceRequestExceptionParams]c                     | j         S r&   )r+   rA   s    r9   on_request_exceptionz TraceConfig.on_request_exception   s     ))r;   *_TracingSignal[TraceRequestRedirectParams]c                     | j         S r&   )r,   rA   s    r9   on_request_redirectzTraceConfig.on_request_redirect   s     ((r;   0_TracingSignal[TraceConnectionQueuedStartParams]c                     | j         S r&   )r-   rA   s    r9   on_connection_queued_startz&TraceConfig.on_connection_queued_start   rL   r;   ._TracingSignal[TraceConnectionQueuedEndParams]c                     | j         S r&   )r.   rA   s    r9   on_connection_queued_endz$TraceConfig.on_connection_queued_end        --r;   0_TracingSignal[TraceConnectionCreateStartParams]c                     | j         S r&   )r/   rA   s    r9   on_connection_create_startz&TraceConfig.on_connection_create_start   rL   r;   ._TracingSignal[TraceConnectionCreateEndParams]c                     | j         S r&   )r0   rA   s    r9   on_connection_create_endz$TraceConfig.on_connection_create_end   r\   r;   ._TracingSignal[TraceConnectionReuseconnParams]c                     | j         S r&   )r1   rA   s    r9   on_connection_reuseconnz#TraceConfig.on_connection_reuseconn        ,,r;   ._TracingSignal[TraceDnsResolveHostStartParams]c                     | j         S r&   )r2   rA   s    r9   on_dns_resolvehost_startz$TraceConfig.on_dns_resolvehost_start   r\   r;   ,_TracingSignal[TraceDnsResolveHostEndParams]c                     | j         S r&   )r3   rA   s    r9   on_dns_resolvehost_endz"TraceConfig.on_dns_resolvehost_end   s     ++r;   &_TracingSignal[TraceDnsCacheHitParams]c                     | j         S r&   )r4   rA   s    r9   on_dns_cache_hitzTraceConfig.on_dns_cache_hit   rE   r;   '_TracingSignal[TraceDnsCacheMissParams]c                     | j         S r&   )r5   rA   s    r9   on_dns_cache_misszTraceConfig.on_dns_cache_miss   s    &&r;   -_TracingSignal[TraceRequestHeadersSentParams]c                     | j         S r&   )r6   rA   s    r9   on_request_headers_sentz#TraceConfig.on_request_headers_sent   rf   r;   r&   r$   N)r$   rB   )r$   rF   )r$   rI   )r$   rM   )r$   rP   )r$   rS   )r$   rV   )r$   rY   )r$   r]   )r$   r`   )r$   rc   )r$   rg   )r$   rj   )r$   rm   )r$   rp   )r$   rs   )__name__
__module____qualname____doc__r   r   r:   r   r   strr>   r@   propertyrD   rH   rK   rO   rR   rU   rX   r[   r_   rb   re   ri   rl   ro   rr   ru    r;   r9   r   r   '   s       PP AP,B ,B(,_(=,B	,B ,B ,B ,B^ @DS S!)'#s(*;!<S	S S S S/ / / /$ & & & X& +	6+ + + X+
 0	;0 0 0 X0
 $ $ $ X$ *	6* * * X*
 )	5) ) ) X)
 0	;0 0 0 X0
 .	9. . . X.
 0	;0 0 0 X0
 .	9. . . X.
 -	9- - - X-
 .	9. . . X.
 ,	7, , , X,
 & & & X& ' ' ' X' -	8- - - X- - -r;   r   )auto_attribsfrozenslotsc                   2    e Zd ZU dZeed<   eed<   ded<   dS )r   z0Parameters sent by the `on_request_start` signalmethodurlCIMultiDict[str]headersNrw   rx   ry   rz   r{   __annotations__r   r}   r;   r9   r   r      s6         ::KKK	HHHr;   r   c                   2    e Zd ZU dZeed<   eed<   eed<   dS )r   z5Parameters sent by the `on_request_chunk_sent` signalr   r   chunkNrw   rx   ry   rz   r{   r   r   bytesr}   r;   r9   r   r      s1         ??KKK	HHHLLLLLr;   r   c                   2    e Zd ZU dZeed<   eed<   eed<   dS )r    z:Parameters sent by the `on_response_chunk_received` signalr   r   r   Nr   r}   r;   r9   r    r       s1         DDKKK	HHHLLLLLr;   r    c                   <    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` signalr   r   r   r   responseNrw   rx   ry   rz   r{   r   r   r   r}   r;   r9   r   r      sB         88KKK	HHHr;   r   c                   <    e Zd ZU dZeed<   eed<   ded<   eed<   dS )r   z4Parameters sent by the `on_request_exception` signalr   r   r   r   	exceptionN)rw   rx   ry   rz   r{   r   r   BaseExceptionr}   r;   r9   r   r      sB         >>KKK	HHHr;   r   c                   <    e Zd ZU dZeed<   eed<   ded<   eed<   dS )r   z3Parameters sent by the `on_request_redirect` signalr   r   r   r   r   Nr   r}   r;   r9   r   r      sB         ==KKK	HHHr;   r   c                       e Zd ZdZdS )r   z:Parameters sent by the `on_connection_queued_start` signalNrw   rx   ry   rz   r}   r;   r9   r   r             DDDDr;   r   c                       e Zd ZdZdS )r   z8Parameters sent by the `on_connection_queued_end` signalNr   r}   r;   r9   r   r             BBBBr;   r   c                       e Zd ZdZdS )r   z:Parameters sent by the `on_connection_create_start` signalNr   r}   r;   r9   r   r     r   r;   r   c                       e Zd ZdZdS )r   z8Parameters sent by the `on_connection_create_end` signalNr   r}   r;   r9   r   r     r   r;   r   c                       e Zd ZdZdS )r   z7Parameters sent by the `on_connection_reuseconn` signalNr   r}   r;   r9   r   r     s        AAAAr;   r   c                       e Zd ZU dZeed<   dS )r   z8Parameters sent by the `on_dns_resolvehost_start` signalhostNrw   rx   ry   rz   r{   r   r}   r;   r9   r   r     s         BB
IIIIIr;   r   c                       e Zd ZU dZeed<   dS )r   z6Parameters sent by the `on_dns_resolvehost_end` signalr   Nr   r}   r;   r9   r   r   "  s         @@
IIIIIr;   r   c                       e Zd ZU dZeed<   dS )r   z0Parameters sent by the `on_dns_cache_hit` signalr   Nr   r}   r;   r9   r   r   )  s         ::
IIIIIr;   r   c                       e Zd ZU dZeed<   dS )r   z1Parameters sent by the `on_dns_cache_miss` signalr   Nr   r}   r;   r9   r   r   0  s         ;;
IIIIIr;   r   c                   2    e Zd ZU dZeed<   eed<   ded<   dS )r!   z7Parameters sent by the `on_request_headers_sent` signalr   r   r   r   Nr   r}   r;   r9   r!   r!   7  s6         AAKKK	HHHr;   r!   c            
       F   e Zd ZdZdddededdfdZd	ed
eddddfdZ	d	ed
ede
ddfdZd	ed
ede
ddfdZd	ed
edddeddf
dZd	ed
edddeddf
dZd	ed
edddeddf
dZd!dZd!dZd!dZd!dZd!dZdeddfdZdeddfdZdeddfdZdeddfdZd	ed
eddddfd ZdS )"TracezInternal dependency holder class.

    Used to keep together the main dependencies used
    at the moment of send a signal.
    sessionr   trace_configr>   r$   Nc                 0    || _         || _        || _        d S r&   )_trace_config_trace_config_ctx_session)r8   r   r   r>   s       r9   r:   zTrace.__init__G  s     *!1r;   r   r   r   r   c           	         K   | j         j                            | j        | j        t          |||                     d {V S r&   )r   rD   sendr   r   r   r8   r   r   r   s       r9   send_request_startzTrace.send_request_startQ  s^       '8==M"#FC99
 
 
 
 
 
 
 
 	
r;   r   c           	         K   | j         j                            | j        | j        t          |||                     d {V S r&   )r   rH   r   r   r   r   r8   r   r   r   s       r9   send_request_chunk_sentzTrace.send_request_chunk_sentZ  s^       '=BBM"'U;;
 
 
 
 
 
 
 
 	
r;   c           	         K   | j         j                            | j        | j        t          |||                     d {V S r&   )r   rK   r   r   r   r    r   s       r9   send_response_chunk_receivedz"Trace.send_response_chunk_receivedc  s^       'BGGM",VS%@@
 
 
 
 
 
 
 
 	
r;   r   c           
         K   | j         j                            | j        | j        t          ||||                     d {V S r&   )r   rO   r   r   r   r   r8   r   r   r   r   s        r9   send_request_endzTrace.send_request_endl  s`       '6;;M"!&#wAA
 
 
 
 
 
 
 
 	
r;   r   c           
         K   | j         j                            | j        | j        t          ||||                     d {V S r&   )r   rR   r   r   r   r   )r8   r   r   r   r   s        r9   send_request_exceptionzTrace.send_request_exceptiony  s`       '<AAM"'WiHH
 
 
 
 
 
 
 
 	
r;   c           
         K   | j         j                            | j        | j        t          ||||                     d {V S r&   )r   r,   r   r   r   r   r   s        r9   send_request_redirectzTrace.send_request_redirect  s`       '<AAM"&vsGXFF
 
 
 
 
 
 
 
 	
r;   c                    K   | j         j                            | j        | j        t                                 d {V S r&   )r   rX   r   r   r   r   rA   s    r9   send_connection_queued_startz"Trace.send_connection_queued_start  U      'BGGM413S3U3U
 
 
 
 
 
 
 
 	
r;   c                    K   | j         j                            | j        | j        t                                 d {V S r&   )r   r[   r   r   r   r   rA   s    r9   send_connection_queued_endz Trace.send_connection_queued_end  U      '@EEM413Q3S3S
 
 
 
 
 
 
 
 	
r;   c                    K   | j         j                            | j        | j        t                                 d {V S r&   )r   r_   r   r   r   r   rA   s    r9   send_connection_create_startz"Trace.send_connection_create_start  r   r;   c                    K   | j         j                            | j        | j        t                                 d {V S r&   )r   rb   r   r   r   r   rA   s    r9   send_connection_create_endz Trace.send_connection_create_end  r   r;   c                    K   | j         j                            | j        | j        t                                 d {V S r&   )r   re   r   r   r   r   rA   s    r9   send_connection_reuseconnzTrace.send_connection_reuseconn  sU      '?DDM413Q3S3S
 
 
 
 
 
 
 
 	
r;   r   c                    K   | j         j                            | j        | j        t          |                     d {V S r&   )r   ri   r   r   r   r   r8   r   s     r9   send_dns_resolvehost_startz Trace.send_dns_resolvehost_start  sX      '@EEM413QRV3W3W
 
 
 
 
 
 
 
 	
r;   c                    K   | j         j                            | j        | j        t          |                     d {V S r&   )r   rl   r   r   r   r   r   s     r9   send_dns_resolvehost_endzTrace.send_dns_resolvehost_end  sX      '>CCM413OPT3U3U
 
 
 
 
 
 
 
 	
r;   c                    K   | j         j                            | j        | j        t          |                     d {V S r&   )r   ro   r   r   r   r   r   s     r9   send_dns_cache_hitzTrace.send_dns_cache_hit  sW      '8==M413I$3O3O
 
 
 
 
 
 
 
 	
r;   c                    K   | j         j                            | j        | j        t          |                     d {V S r&   )r   rr   r   r   r   r   r   s     r9   send_dns_cache_misszTrace.send_dns_cache_miss  sW      '9>>M413J43P3P
 
 
 
 
 
 
 
 	
r;   c           	         K   | j         j                            | j        | j        t          |||                     d {V S r&   )r   r6   r   r   r   r!   r   s       r9   send_request_headerszTrace.send_request_headers  s^       '@EEM")&#w??
 
 
 
 
 
 
 
 	
r;   rv   )rw   rx   ry   rz   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;   r9   r   r   @  s            "  *	 
 
       

 #
.@
	
 
 
 


 #
,1
	
 
 
 


 #
,1
	
 
 
 


 
 $	

 !
 

 
 
 


 
 $	

 !
 

 
 
 


 
 $	

 !
 

 
 
 

 
 
 


 
 
 


 
 
 


 
 
 


 
 
 


S 
T 
 
 
 


3 
4 
 
 
 


S 
T 
 
 
 


c 
d 
 
 
 



 #
.@
	
 
 
 
 
 
r;   r   ))typesr   typingr   r   r   r   r   attr	aiosignalr	   	multidictr
   yarlr   client_reqrepr   clientr   r   _TracingSignal__all__r   sr   r   r    r   r   r   r   r   r   r   r   r   r   r   r   r!   r   r}   r;   r9   <module>r      s   ! ! ! ! ! ! B B B B B B B B B B B B B B        ! ! ! ! ! !       ) ) ) ) ) ) L%%%%%%W-TBBBNM?NJKN*_- _- _- _- _- _- _- _-D T$d333              43  T$d333       43 T$d333       43 T$d333       43 T$d333       43 T$d333       43 T$d333E E E E E E E 43E T$d333C C C C C C C 43C T$d333E E E E E E E 43E T$d333C C C C C C C 43C T$d333B B B B B B B 43B T$d333       43 T$d333       43 T$d333       43 T$d333       43 T$d333              43 G
 G
 G
 G
 G
 G
 G
 G
 G
 G
r;   