
    *Gj8                        d dl mZ d dl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Any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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     W/home/rurouni/.hermes/hermes-agent/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:   )r5   )r6   r:   s     r7   trace_config_ctxzTraceConfig.trace_config_ctxX   s    --@Q-RRRr9   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-   r.   r/   r0   r1   r2   r3   r4   r6   s    r7   r>   zTraceConfig.freeze\   sX   %%'''#**,,,(//111##%%%"))+++!((***(//111&--///(//111&--///%,,...&--///$++---%%'''&&(((%,,.....r9   '_TracingSignal[TraceRequestStartParams]c                     | j         S r$   )r%   r?   s    r7   on_request_startzTraceConfig.on_request_startn       %%r9   +_TracingSignal[TraceRequestChunkSentParams]c                     | j         S r$   )r&   r?   s    r7   on_request_chunk_sentz!TraceConfig.on_request_chunk_sentr   s     **r9   0_TracingSignal[TraceResponseChunkReceivedParams]c                     | j         S r$   )r'   r?   s    r7   on_response_chunk_receivedz&TraceConfig.on_response_chunk_receivedx        //r9   %_TracingSignal[TraceRequestEndParams]c                     | j         S r$   )r(   r?   s    r7   on_request_endzTraceConfig.on_request_end~   s    ##r9   +_TracingSignal[TraceRequestExceptionParams]c                     | j         S r$   )r)   r?   s    r7   on_request_exceptionz TraceConfig.on_request_exception   s     ))r9   *_TracingSignal[TraceRequestRedirectParams]c                     | j         S r$   )r*   r?   s    r7   on_request_redirectzTraceConfig.on_request_redirect   s     ((r9   0_TracingSignal[TraceConnectionQueuedStartParams]c                     | j         S r$   )r+   r?   s    r7   on_connection_queued_startz&TraceConfig.on_connection_queued_start   rJ   r9   ._TracingSignal[TraceConnectionQueuedEndParams]c                     | j         S r$   )r,   r?   s    r7   on_connection_queued_endz$TraceConfig.on_connection_queued_end        --r9   0_TracingSignal[TraceConnectionCreateStartParams]c                     | j         S r$   )r-   r?   s    r7   on_connection_create_startz&TraceConfig.on_connection_create_start   rJ   r9   ._TracingSignal[TraceConnectionCreateEndParams]c                     | j         S r$   )r.   r?   s    r7   on_connection_create_endz$TraceConfig.on_connection_create_end   rZ   r9   ._TracingSignal[TraceConnectionReuseconnParams]c                     | j         S r$   )r/   r?   s    r7   on_connection_reuseconnz#TraceConfig.on_connection_reuseconn        ,,r9   ._TracingSignal[TraceDnsResolveHostStartParams]c                     | j         S r$   )r0   r?   s    r7   on_dns_resolvehost_startz$TraceConfig.on_dns_resolvehost_start   rZ   r9   ,_TracingSignal[TraceDnsResolveHostEndParams]c                     | j         S r$   )r1   r?   s    r7   on_dns_resolvehost_endz"TraceConfig.on_dns_resolvehost_end   s     ++r9   &_TracingSignal[TraceDnsCacheHitParams]c                     | j         S r$   )r2   r?   s    r7   on_dns_cache_hitzTraceConfig.on_dns_cache_hit   rC   r9   '_TracingSignal[TraceDnsCacheMissParams]c                     | j         S r$   )r3   r?   s    r7   on_dns_cache_misszTraceConfig.on_dns_cache_miss   s    &&r9   -_TracingSignal[TraceRequestHeadersSentParams]c                     | j         S r$   )r4   r?   s    r7   on_request_headers_sentz#TraceConfig.on_request_headers_sent   rd   r9   r$   r"   N)r"   r@   )r"   rD   )r"   rG   )r"   rK   )r"   rN   )r"   rQ   )r"   rT   )r"   rW   )r"   r[   )r"   r^   )r"   ra   )r"   re   )r"   rh   )r"   rk   )r"   rn   )r"   rq   )__name__
__module____qualname____doc__r   typer8   r   r<   r>   propertyrB   rF   rI   rM   rP   rS   rV   rY   r]   r`   rc   rg   rj   rm   rp   rs    r9   r7   r   r   '   s       PP AP,B ,B(,_(=,B	,B ,B ,B ,B\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- - -r9   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ru   rv   rw   rx   str__annotations__r	   r{   r9   r7   r   r      s6         ::KKK	HHHr9   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ru   rv   rw   rx   r   r   r	   bytesr{   r9   r7   r   r      s1         ??KKK	HHHLLLLLr9   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{   r9   r7   r   r      s1         DDKKK	HHHLLLLLr9   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ru   rv   rw   rx   r   r   r	   r   r{   r9   r7   r   r      sB         88KKK	HHHr9   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)ru   rv   rw   rx   r   r   r	   BaseExceptionr{   r9   r7   r   r      sB         >>KKK	HHHr9   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{   r9   r7   r   r      sB         ==KKK	HHHr9   r   c                       e Zd ZdZdS )r   z:Parameters sent by the `on_connection_queued_start` signalNru   rv   rw   rx   r{   r9   r7   r   r              DDDDr9   r   c                       e Zd ZdZdS )r   z8Parameters sent by the `on_connection_queued_end` signalNr   r{   r9   r7   r   r             BBBBr9   r   c                       e Zd ZdZdS )r   z:Parameters sent by the `on_connection_create_start` signalNr   r{   r9   r7   r   r   
  r   r9   r   c                       e Zd ZdZdS )r   z8Parameters sent by the `on_connection_create_end` signalNr   r{   r9   r7   r   r     r   r9   r   c                       e Zd ZdZdS )r   z7Parameters sent by the `on_connection_reuseconn` signalNr   r{   r9   r7   r   r     s        AAAAr9   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ru   rv   rw   rx   r   r   r{   r9   r7   r   r     s         BB
IIIIIr9   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{   r9   r7   r   r      s         @@
IIIIIr9   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{   r9   r7   r   r   '  s         ::
IIIIIr9   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{   r9   r7   r   r   .  s         ;;
IIIIIr9   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{   r9   r7   r   r   5  s6         AAKKK	HHHr9   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)r6   r   r   r<   s       r7   r8   zTrace.__init__E  s     *!1r9   r   r   r   r   c           	         K   | j         j                            | j        | j        t          |||                     d {V S r$   )r   rB   sendr   r   r   r6   r   r   r   s       r7   send_request_startzTrace.send_request_startO  s^       '8==M"#FC99
 
 
 
 
 
 
 
 	
r9   r   c           	         K   | j         j                            | j        | j        t          |||                     d {V S r$   )r   rF   r   r   r   r   r6   r   r   r   s       r7   send_request_chunk_sentzTrace.send_request_chunk_sentX  s^       '=BBM"'U;;
 
 
 
 
 
 
 
 	
r9   c           	         K   | j         j                            | j        | j        t          |||                     d {V S r$   )r   rI   r   r   r   r   r   s       r7   send_response_chunk_receivedz"Trace.send_response_chunk_receiveda  s^       'BGGM",VS%@@
 
 
 
 
 
 
 
 	
r9   r   c           
         K   | j         j                            | j        | j        t          ||||                     d {V S r$   )r   rM   r   r   r   r   r6   r   r   r   r   s        r7   send_request_endzTrace.send_request_endj  s`       '6;;M"!&#wAA
 
 
 
 
 
 
 
 	
r9   r   c           
         K   | j         j                            | j        | j        t          ||||                     d {V S r$   )r   rP   r   r   r   r   )r6   r   r   r   r   s        r7   send_request_exceptionzTrace.send_request_exceptionw  s`       '<AAM"'WiHH
 
 
 
 
 
 
 
 	
r9   c           
         K   | j         j                            | j        | j        t          ||||                     d {V S r$   )r   r*   r   r   r   r   r   s        r7   send_request_redirectzTrace.send_request_redirect  s`       '<AAM"&vsGXFF
 
 
 
 
 
 
 
 	
r9   c                    K   | j         j                            | j        | j        t                                 d {V S r$   )r   rV   r   r   r   r   r?   s    r7   send_connection_queued_startz"Trace.send_connection_queued_start  U      'BGGM413S3U3U
 
 
 
 
 
 
 
 	
r9   c                    K   | j         j                            | j        | j        t                                 d {V S r$   )r   rY   r   r   r   r   r?   s    r7   send_connection_queued_endz Trace.send_connection_queued_end  U      '@EEM413Q3S3S
 
 
 
 
 
 
 
 	
r9   c                    K   | j         j                            | j        | j        t                                 d {V S r$   )r   r]   r   r   r   r   r?   s    r7   send_connection_create_startz"Trace.send_connection_create_start  r   r9   c                    K   | j         j                            | j        | j        t                                 d {V S r$   )r   r`   r   r   r   r   r?   s    r7   send_connection_create_endz Trace.send_connection_create_end  r   r9   c                    K   | j         j                            | j        | j        t                                 d {V S r$   )r   rc   r   r   r   r   r?   s    r7   send_connection_reuseconnzTrace.send_connection_reuseconn  sU      '?DDM413Q3S3S
 
 
 
 
 
 
 
 	
r9   r   c                    K   | j         j                            | j        | j        t          |                     d {V S r$   )r   rg   r   r   r   r   r6   r   s     r7   send_dns_resolvehost_startz Trace.send_dns_resolvehost_start  sX      '@EEM413QRV3W3W
 
 
 
 
 
 
 
 	
r9   c                    K   | j         j                            | j        | j        t          |                     d {V S r$   )r   rj   r   r   r   r   r   s     r7   send_dns_resolvehost_endzTrace.send_dns_resolvehost_end  sX      '>CCM413OPT3U3U
 
 
 
 
 
 
 
 	
r9   c                    K   | j         j                            | j        | j        t          |                     d {V S r$   )r   rm   r   r   r   r   r   s     r7   send_dns_cache_hitzTrace.send_dns_cache_hit  sW      '8==M413I$3O3O
 
 
 
 
 
 
 
 	
r9   c                    K   | j         j                            | j        | j        t          |                     d {V S r$   )r   rp   r   r   r   r   r   s     r7   send_dns_cache_misszTrace.send_dns_cache_miss  sW      '9>>M413J43P3P
 
 
 
 
 
 
 
 	
r9   c           	         K   | j         j                            | j        | j        t          |||                     d {V S r$   )r   r4   r   r   r   r   r   s       r7   send_request_headerszTrace.send_request_headers  s^       '@EEM")&#w??
 
 
 
 
 
 
 
 	
r9   rt   )ru   rv   rw   rx   r   r   r8   r   r	   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r{   r9   r7   r   r   >  s            "  *	 
 
       

 #
.@
	
 
 
 


 #
,1
	
 
 
 


 #
,1
	
 
 
 


 
 $	

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

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

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


 
 
 


 
 
 


 
 
 


 
 
 

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

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

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

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



 #
.@
	
 
 
 
 
 
r9   r   )'typesr   typingr   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{   r9   r7   <module>r      s   ! ! ! ! ! ! . . . . . . . . . .        ! ! ! ! ! !       ) ) ) ) ) ) L%%%%%%W-TBBBNM?NJKN*]- ]- ]- ]- ]- ]- ]- ]-@ 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
r9   