U
'7`8 @ s d dl mZ d dlmZmZmZmZmZ d dlZd dl m
Z
d dl m Z ddl
mZ ddlmZ erd d lmZ dd
lmZ ed d d
ZG dd dee ZdZG dd dZejd d d dG dd dZejd d d dG dd dZejd d d dG dd dZ ejd d d dG dd dZ ejd d d dG d d d Z ejd d d dG d d d Z ejd d d dG d d! d!Z ejd d d dG d"d# d#Z!ejd d d dG d$d% d%Z"ejd d d dG d&d' d'Z#ejd d d dG d(d) d)Z$ejd d d dG d*d+ d+Z%ejd d d dG d,d- d-Z&ejd d d dG d.d/ d/Z'ejd d d dG d0d1 d1Z(G d2d3 d3Z)dS )4 )SimpleNamespace)
TYPE_CHECKING AwaitableOptionalTypeTypeVarN) CIMultiDict)URL )ClientResponse)Signal)Protocol)
ClientSession_ParamT_contraT)Z
contravariantc @ s$ e Zd Zeeeed dddZdS )_SignalCallbackN) _SignalCallback__client_session!_SignalCallback__trace_config_ctx_SignalCallback__paramsreturnc C s d S N )selfr r r r r B/opt/alt/python38/lib64/python3.8/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 ) TraceConfigTraceRequestStartParamsTraceRequestEndParamsTraceRequestExceptionParams TraceConnectionQueuedStartParams TraceConnectionQueuedEndParams TraceConnectionCreateStartParams TraceConnectionCreateEndParams TraceConnectionReuseconnParams TraceDnsResolveHostStartParams TraceDnsResolveHostEndParamsTraceDnsCacheHitParamsTraceDnsCacheMissParamsTraceRequestRedirectParamsTraceRequestChunkSentParams TraceResponseChunkReceivedParamsc @ sZ e Zd ZdZefee ddddZd9ee edddZdd d
d Z e
d d d
dZ e
dd ddZ e
dd ddZ
e
dd ddZe
dd ddZe
dd d d Ze
d d d d Ze
d!d d"d#Ze
d$d d%d&Ze
d'd d(d)Ze
d*d d+d,Ze
d-d d.d/Ze
d0d d1d2Ze
d3d d4d5Ze
d6d d7d8ZdS ):r zJFirst-class used to trace requests launched via ClientSession
objects.N)trace_config_ctx_factoryr c 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\ zTraceConfig.__init__)trace_request_ctxr c C s | j |dS )z( Return a new trace_config_ctx instance )r? )r= )r r? r r r trace_config_ctxg s z TraceConfig.trace_config_ctxr c C s | j | j | j | j | j | j | j | j | j | j
| j | j | j
| j | j d S r )r. freezer/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r r r r rB m s
zTraceConfig.freezez0Signal[_SignalCallback[TraceRequestStartParams]]c C s | j S r )r. rC r r r on_request_start~ s z TraceConfig.on_request_startz4Signal[_SignalCallback[TraceRequestChunkSentParams]]c C s | j S r )r/ rC r r r on_request_chunk_sent s z!TraceConfig.on_request_chunk_sentz9Signal[_SignalCallback[TraceResponseChunkReceivedParams]]c C s | j S r )r0 rC r r r on_response_chunk_received s z&TraceConfig.on_response_chunk_receivedz.Signal[_SignalCallback[TraceRequestEndParams]]c C s | j S r )r1 rC r r r on_request_end s zTraceConfig.on_request_endz4Signal[_SignalCallback[TraceRequestExceptionParams]]c C s | j S r )r2 rC r r r on_request_exception s z TraceConfig.on_request_exceptionz3Signal[_SignalCallback[TraceRequestRedirectParams]]c C s | j S r )r3 rC r r r on_request_redirect s z TraceConfig.on_request_redirectz9Signal[_SignalCallback[TraceConnectionQueuedStartParams]]c C s | j S r )r4 rC r r r on_connection_queued_start s z&TraceConfig.on_connection_queued_startz7Signal[_SignalCallback[TraceConnectionQueuedEndParams]]c C s | j S r )r5 rC r r r on_connection_queued_end s z$TraceConfig.on_connection_queued_endz9Signal[_SignalCallback[TraceConnectionCreateStartParams]]c C s | j S r )r6 rC r r r on_connection_create_start s z&TraceConfig.on_connection_create_startz7Signal[_SignalCallback[TraceConnectionCreateEndParams]]c C s | j S r )r7 rC r r r on_connection_create_end s z$TraceConfig.on_connection_create_endz7Signal[_SignalCallback[TraceConnectionReuseconnParams]]c C s | j S r )r8 rC r r r on_connection_reuseconn s z#TraceConfig.on_connection_reuseconnz7Signal[_SignalCallback[TraceDnsResolveHostStartParams]]c C s | j S r )r9 rC r r r on_dns_resolvehost_start s z$TraceConfig.on_dns_resolvehost_startz5Signal[_SignalCallback[TraceDnsResolveHostEndParams]]c C s | j S r )r: rC r r r on_dns_resolvehost_end s z"TraceConfig.on_dns_resolvehost_endz/Signal[_SignalCallback[TraceDnsCacheHitParams]]c C s | j S r )r; rC r r r on_dns_cache_hit s z TraceConfig.on_dns_cache_hitz0Signal[_SignalCallback[TraceDnsCacheMissParams]]c C s | j S r )r< rC r r r on_dns_cache_miss s z TraceConfig.on_dns_cache_miss)N)r r r __doc__r r r> r r@ rB propertyrD rE rF rG rH rI rJ rK rL rM rN rO rP rQ rR r r r r r 0 sj 4 r )Z auto_attribsfrozenslotsc @ s* e Zd ZU dZeed< eed< ded< dS )r z1 Parameters sent by the `on_request_start` signalmethodurlCIMultiDict[str]headersN)r r r rS str__annotations__r r r r r r s
r c @ s* e Zd ZU dZeed< eed< eed< dS )r+ z6 Parameters sent by the `on_request_chunk_sent` signalrW rX chunkNr r r rS r[ r\ r bytesr r r r r+ s
r+ c @ s* e Zd ZU dZeed< eed< eed< dS )r, z; Parameters sent by the `on_response_chunk_received` signalrW rX r] Nr^ r r r r r, s
r, c @ s2 e Zd ZU dZeed< eed< ded< eed< dS )r z/ Parameters sent by the `on_request_end` signalrW rX rY rZ responseNr r r rS r[ r\ r r r r r r r s
r c @ s2 e Zd ZU dZeed< eed< ded< eed< dS )r z5 Parameters sent by the `on_request_exception` signalrW rX rY rZ exceptionN)r r r rS r[ r\ r
BaseExceptionr r r r r s
r c @ s2 e Zd ZU dZeed< eed< ded< eed< dS )r* z4 Parameters sent by the `on_request_redirect` signalrW rX rY rZ r` Nra r r r r r* s
r* c @ s e Zd ZdZdS )r! z; Parameters sent by the `on_connection_queued_start` signalNr r r rS r r r r r!
s r! c @ s e Zd ZdZdS )r" z9 Parameters sent by the `on_connection_queued_end` signalNrd r r r r r" s r" c @ s e Zd ZdZdS )r# z; Parameters sent by the `on_connection_create_start` signalNrd r r r r r# s r# c @ s e Zd ZdZdS )r$ z9 Parameters sent by the `on_connection_create_end` signalNrd r r r r r$ s r$ c @ s e Zd ZdZdS )r% z8 Parameters sent by the `on_connection_reuseconn` signalNrd r r r r r% s r% c @ s e Zd ZU dZeed< dS )r& z9 Parameters sent by the `on_dns_resolvehost_start` signalhostNr r r rS r[ r\ r r r r r& # s
r& c @ s e Zd ZU dZeed< dS )r' z7 Parameters sent by the `on_dns_resolvehost_end` signalre Nrf r r r r r' * s
r' c @ s e Zd ZU dZeed< dS )r( z1 Parameters sent by the `on_dns_cache_hit` signalre Nrf r r r r r( 1 s
r( c @ s e Zd ZU dZeed< dS )r) z2 Parameters sent by the `on_dns_cache_miss` signalre Nrf r r r r r) 8 s
r) c @ s( e Zd ZdZdeeddddZeedddd d
Z eee
dd d d
Z eee
dd ddZ eede
ddddZeededdddZeede
ddddZddddZdddd Zddd d Zddd d Zddd!d"Zedd#d$d%Zedd#d&d'Zedd#d(d)Zedd#d*d+ZdS ),TracezcInternal class used to keep together the main dependencies used
at the moment of send a signal.r N)session trace_configr@ r c C s || _ || _|| _d S r )
_trace_config_trace_config_ctx_session)r rh ri r@ r r r r> C s zTrace.__init__rY )rW rX rZ r c s$ | j j| j| jt|||I d H S r )rj rD sendrl rk r )r rW rX rZ r r r send_request_startM s
zTrace.send_request_start)rW rX r] r c s$ | j j| j| jt|||I d H S r )rj rE rm rl rk r+ r rW rX r] r r r send_request_chunk_sentV s
z Trace.send_request_chunk_sentc s$ | j j| j| jt|||I d H S r )rj rF rm rl rk r, ro r r r send_response_chunk_received_ s
z"Trace.send_response_chunk_received)rW rX rZ r` r c s& | j j| j| jt||||I d H S r )rj rG rm rl rk r r rW rX rZ r` r r r send_request_endh s
zTrace.send_request_end)rW rX rZ rb r c s& | j j| j| jt||||I d H S r )rj rH rm rl rk r )r rW rX rZ rb r r r send_request_exceptionu s
z Trace.send_request_exceptionc s& | j j| j| jt||||I d H S r )rj r3 rm rl rk r* rr r r r send_request_redirect s
zTrace.send_request_redirectrA c s | j j| j| jt I d H S r )rj rJ rm rl rk r! rC r r r send_connection_queued_start s
z"Trace.send_connection_queued_startc s | j j| j| jt I d H S r )rj rK rm rl rk r" rC r r r send_connection_queued_end s
z Trace.send_connection_queued_endc s | j j| j| jt I d H S r )rj rL rm rl rk r# rC r r r send_connection_create_start s
z"Trace.send_connection_create_startc s | j j| j| jt I d H S r )rj rM rm rl rk r$ rC r r r send_connection_create_end s
z Trace.send_connection_create_endc s | j j| j| jt I d H S r )rj rN rm rl rk r% rC r r r send_connection_reuseconn s
z Trace.send_connection_reuseconn)re r c s | j j| j| jt|I d H S r )rj rO rm rl rk r& r re r r r send_dns_resolvehost_start s
z Trace.send_dns_resolvehost_startc s | j j| j| jt|I d H S r )rj rP rm rl rk r' r{ r r r send_dns_resolvehost_end s
z Trace.send_dns_resolvehost_endc s | j j| j| jt|I d H S r )rj rQ rm rl rk r( r{ r r r send_dns_cache_hit s
zTrace.send_dns_cache_hitc s | j j| j| jt|I d H S r )rj rR rm rl rk r) r{ r r r send_dns_cache_miss s
zTrace.send_dns_cache_miss)r r r rS r r r> r[ r rn r_ rp rq r rs rc rt ru rv rw rx ry rz r| r} r~ r r r r r rg ? s`
rg )*typesr typingr r r r r attrZ multidictr Zyarlr Z
client_reqrepr Zsignalsr Ztyping_extensionsr
Zclientr r r __all__r sr r+ r, r r r* r! r" r# r$ r% r& r' r( r) rg r r r r <module> sZ
"