���ZF% � @ s� d d l 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
d) Z Gd d � d � Z e j
d d d d � Gd d � d � � Z e j
d d d d � Gd d
� d
� � Z e j
d d d d � Gd d � d � � Z e j
d d d d � Gd d � d � � Z e j
d d d d � Gd d � d � � Z e j
d d d d � Gd d
� d
� � Z e j
d d d d � Gd d � d � � Z e j
d d d d � Gd! d � d � � Z e j
d d d d � Gd" d � d � � Z e j
d d d d � Gd# d � d � � Z e j
d d d d � Gd$ d � d � � Z e j
d d d d � Gd% d � d � � Z e j
d d d d � Gd& d � d � � Z Gd' d( � d( � Z d S)*� )�SimpleNamespaceN)� CIMultiDict)�URL� )�ClientResponse)�Signal� TraceConfig�TraceRequestStartParams�TraceRequestEndParams�TraceRequestExceptionParams� TraceConnectionQueuedStartParams� TraceConnectionQueuedEndParams� TraceConnectionCreateStartParams� TraceConnectionCreateEndParams� TraceConnectionReuseconnParams� TraceDnsResolveHostStartParams� TraceDnsResolveHostEndParams�TraceDnsCacheHitParams�TraceDnsCacheMissParams�TraceRequestRedirectParamsc @ s* e Z d Z d Z e d d � Z d d 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 d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d! d" � � Z d S)#r zJFirst-class used to trace requests launched via ClientSession
objects.c C s� t | � | _ t | � | _ t | � | _ t | � | _ t | � | _ t | � | _ t | � | _ t | � | _ t | � | _ t | � | _
t | � | _ t | � | _ t | � | _
| | _ d S)N)r �_on_request_start�_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)�selfZtrace_config_ctx_factory� r% � /tracing.py�__init__ s zTraceConfig.__init__Nc C s | j d | � S)z( Return a new trace_config_ctx instance �trace_request_ctx)r# )r$ r( r% r% r& �trace_config_ctx+ s z TraceConfig.trace_config_ctxc C s� | j j � | j j � | j j � | j j � | j j � | j j � | j j � | j j � | j j � | j
j � | j j � | j j � | j
j � d S)N)r �freezer r r r r r r r r r r! r" )r$ r% r% r& r* 0 s
zTraceConfig.freezec C s | j S)N)r )r$ r% r% r& �on_request_start? s z TraceConfig.on_request_startc C s | j S)N)r )r$ r% r% r& �on_request_endC s zTraceConfig.on_request_endc C s | j S)N)r )r$ r% r% r& �on_request_exceptionG s z TraceConfig.on_request_exceptionc C s | j S)N)r )r$ r% r% r& �on_request_redirectK s z TraceConfig.on_request_redirectc C s | j S)N)r )r$ r% r% r& �on_connection_queued_startO s z&TraceConfig.on_connection_queued_startc C s | j S)N)r )r$ r% r% r& �on_connection_queued_endS s z$TraceConfig.on_connection_queued_endc C s | j S)N)r )r$ r% r% r& �on_connection_create_startW s z&TraceConfig.on_connection_create_startc C s | j S)N)r )r$ r% r% r& �on_connection_create_end[ s z$TraceConfig.on_connection_create_endc C s | j S)N)r )r$ r% r% r& �on_connection_reuseconn_ s z#TraceConfig.on_connection_reuseconnc C s | j S)N)r )r$ r% r% r& �on_dns_resolvehost_startc s z$TraceConfig.on_dns_resolvehost_startc C s | j S)N)r )r$ r% r% r& �on_dns_resolvehost_endg s z"TraceConfig.on_dns_resolvehost_endc C s | j S)N)r! )r$ r% r% r& �on_dns_cache_hitk s z TraceConfig.on_dns_cache_hitc C s | j S)N)r" )r$ r% r% r& �on_dns_cache_misso s z TraceConfig.on_dns_cache_miss)�__name__�
__module__� __qualname__�__doc__r r' r) r* �propertyr+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r% r% r% r& r s" �frozenT�slotsc @ sL e Z d Z d Z e j d e � Z e j d e � Z e j d e
� Z d S)r z1 Parameters sent by the `on_request_start` signal�typeN) r8 r9 r: r; �attr�ib�str�methodr �urlr �headersr% r% r% r& r t s c @ s^ e Z d Z d Z e j d e � Z e j d e � Z e j d e
� Z e j d e � Z
d S)r
z/ Parameters sent by the `on_request_end` signalr? N)r8 r9 r: r; r@ rA rB rC r rD r rE r �responser% r% r% r& r
| s
c @ s^ e Z d Z d Z e j d e � Z e j d e � Z e j d e
� Z e j d e � Z
d S)r z5 Parameters sent by the `on_request_exception` signalr? N)r8 r9 r: r; r@ rA rB rC r rD r rE � Exception� exceptionr% r% r% r& r � s
c @ s^ e Z d Z d Z e j d e � Z e j d e � Z e j d e
� Z e j d e � Z
d S)r z4 Parameters sent by the `on_request_redirect` signalr? N)r8 r9 r: r; r@ rA rB rC r rD r rE r rF r% r% r% r& r � s
c @ s e Z d Z d Z d S)r z; Parameters sent by the `on_connection_queued_start` signalN)r8 r9 r: r; r% r% r% r& r � s c @ s e Z d Z d Z d S)r
z9 Parameters sent by the `on_connection_queued_end` signalN)r8 r9 r: r; r% r% r% r& r
� s c @ s e Z d Z d Z d S)r z; Parameters sent by the `on_connection_create_start` signalN)r8 r9 r: r; r% r% r% r& r � s c @ s e Z d Z d Z d S)r z9 Parameters sent by the `on_connection_create_end` signalN)r8 r9 r: r; r% r% r% r& r � s c @ s e Z d Z d Z d S)r z8 Parameters sent by the `on_connection_reuseconn` signalN)r8 r9 r: r; r% r% r% r& r � s c @ s( e Z d Z d Z e j d e � Z d S)r z9 Parameters sent by the `on_dns_resolvehost_start` signalr? N)r8 r9 r: r; r@ rA rB �hostr% r% r% r& r � s c @ s( e Z d Z d Z e j d e � Z d S)r z7 Parameters sent by the `on_dns_resolvehost_end` signalr? N)r8 r9 r: r; r@ rA rB rI r% r% r% r& r � s c @ s( e Z d Z d Z e j d e � Z d S)r z1 Parameters sent by the `on_dns_cache_hit` signalr? N)r8 r9 r: r; r@ rA rB rI r% r% r% r& r � s c @ s( e Z d Z d Z e j d e � Z d S)r z2 Parameters sent by the `on_dns_cache_miss` signalr? N)r8 r9 r: r; r@ rA rB rI r% r% r% r& r � s c @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d S) �Tracezd Internal class used to keep together the main dependencies used
at the moment of send a signal.c C s | | _ | | _ | | _ d S)N)�
_trace_config�_trace_config_ctx�_session)r$ ZsessionZ trace_configr) r% r% r& r' � s zTrace.__init__c � s0 | j j j | j | j t | | | � � Id HS)N)rK r+ �sendrM rL r )r$ rC rD rE r% r% r& �send_request_start� s zTrace.send_request_startc � s3 | j j j | j | j t | | | | � � Id HS)N)rK r, rN rM rL r
)r$ rC rD rE rF r% r% r& �send_request_end� s zTrace.send_request_endc � s3 | j j j | j | j t | | | | � � Id HS)N)rK r- rN rM rL r )r$ rC rD rE rH r% r% r& �send_request_exception� s z Trace.send_request_exceptionc � s3 | j j j | j | j t | | | | � � Id HS)N)rK r rN rM rL r )r$ rC rD rE rF r% r% r& �send_request_redirect� s zTrace.send_request_redirectc � s' | j j j | j | j t � � Id HS)N)rK r/ rN rM rL r )r$ r% r% r& � send_connection_queued_start� s z"Trace.send_connection_queued_startc � s' | j j j | j | j t � � Id HS)N)rK r0 rN rM rL r
)r$ r% r% r& �send_connection_queued_end� s z Trace.send_connection_queued_endc � s' | j j j | j | j t � � Id HS)N)rK r1 rN rM rL r )r$ r% r% r& � send_connection_create_start� s z"Trace.send_connection_create_startc � s' | j j j | j | j t � � Id HS)N)rK r2 rN rM rL r )r$ r% r% r& �send_connection_create_end s z Trace.send_connection_create_endc � s' | j j j | j | j t � � Id HS)N)rK r3 rN rM rL r )r$ r% r% r& �send_connection_reuseconn s z Trace.send_connection_reuseconnc � s* | j j j | j | j t | � � Id HS)N)rK r4 rN rM rL r )r$ rI r% r% r& �send_dns_resolvehost_start s z Trace.send_dns_resolvehost_startc � s* | j j j | j | j t | � � Id HS)N)rK r5 rN rM rL r )r$ rI r% r% r& �send_dns_resolvehost_end s z Trace.send_dns_resolvehost_endc � s* | j j j | j | j t | � � Id HS)N)rK r6 rN rM rL r )r$ rI r% r% r& �send_dns_cache_hit s zTrace.send_dns_cache_hitc � s* | j j j | j | j t | � � Id HS)N)rK r7 rN rM rL r )r$ rI r% r% r& �send_dns_cache_miss% s zTrace.send_dns_cache_missN)r8 r9 r: r; r' rO rP rQ rR rS rT rU rV rW rX rY rZ r[ r% r% r% r& rJ � s rJ )r r r
r r r
r r r r r r r r ) �typesr r@ Z multidictr Zyarlr Z
client_reqrepr Zsignalsr �__all__r �sr r
r r r r
r r r r r r r rJ r% r% r% r& �<module> s6 ^++ + + +++++++++