xj[%ddlZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZmZmZmZ ddlZdd lmZn#e$r ed wxYwdd lmZmZmZer+dd lmZdd lm Z m!Z!m"Z"ddl#m$Z$e"de!de fZ%ddde&fdZ'd(dZ(d)dZ) d*d Z*d)d!Z+d+d#Z,Gd$d%eZ-d,d'Z.dS)-N)OP) DidNotEnable Integration) StreamedSpan)Span)has_span_streaming_enabled)AsyncHttpTransport)event_from_exceptionis_internal_tasklogger nullcontextreraise)Taskzasyncio not available) TYPE_CHECKINGOptionalUnion) Coroutine)AnyCallableTypeVar)ExcInfoT.)boundcororreturncJt|ddpt|ddpdS)N __qualname____name__zcoroutine without __name__)getattr)rs PC:\Projects\Historical\venv\Lib\site-packages\sentry_sdk/integrations/asyncio.pyget_namer!$s2nd++ ( 4T * * ( 'wrappedCoroutine[Any, Any, Any]Callable[[T], T]cFtjtj|ddS)N)rr)r#assignedupdated) functoolspartialupdate_wrapper)r#s r _wrap_coroutiner-,s+   -    r"c tjn#t$rYdSwxYwtddrdSddjdfd }|_d_dS) z9Patch loop.close to flush pending events before shutdown.N_sentry_flush_patchedFrcKtj}|sdS t|jt sdS|d{VdS#t$rtj ddYdSwxYw)Nz(Sentry flush failed during loop shutdownTexc_info) sentry_sdk get_client is_active isinstance transportr close_async Exceptionr warning)clients r _flushz patch_loop_close.._flushDs&((!!  F Vf.0BCC $$&& & & & & & & & & & V V V NEPT U U U U U U U VsA% A%% B B c n&#t$rtjddYnwxYwdS#wxYw)Nz/Could not flush Sentry events during loop closeTr1)run_until_completer9r debug)r<loop orig_closesr _patched_closez(patch_loop_close.._patched_closeSs   # #FFHH - - - -    LAD        JLLLLLJJLLLLs&!A AAAA ATrN)asyncioget_running_loop RuntimeErrorrcloser/)rBr<r@rAs @@@r patch_loop_closerH6s ')) t,e44 V V V VJ DJ!%Ds  ''orig_task_factoryr@asyncio.AbstractEventLoopkwargsasyncio.Task[Any]c bd}|r |||fi|}|t|fd|i|}|jr|jd=|S)Nr@)r_source_traceback)rIr@rrKtasks r _create_task_with_factoryrQasi D7  t66v66 |D..t.v..  ! +&r* Kr"c d tj}|tddrdS dfd }d |_||dS#t $rtjd YdSwxYw)N_is_sentry_task_factoryFr@rJrr$rKrrasyncio.Future[Any]ctrt|fi|Stdfd }t||fi|} |t dn#t $rYnwxYw|S)NrrcKd}tj}|t}|r|jnd}d}t |j}tj5|r|rdtj Etj ttj tjd}n8tj tj ttj}|r|n t5 d{V}n5#t $r}|dd}~wt"$rt%t'YnwxYwdddn #1swxYwYdddn #1swxYwY|S)NF)z sentry.opz sentry.origin)name attributes)oprWorigin)r3r4get_integrationAsyncioIntegration task_spansroptionsisolation_scopetracesget_current_span start_spanr!rFUNCTIONrZr StopAsyncIterationr9r_capture_exception)resultr; integrationr]span_ctxis_span_streaming_enableders r _task_with_sentry_span_creationzTpatch_asyncio.._sentry_task_factory.._task_with_sentry_span_creations"#.00$445GHH 7BM[33 BF,Fv~,V,V)/11;;!4)0AACCO+5+<+G+G)1$57[9K9R0&0&,H,","(2'<#%;%-d^^'9'@(((H &.@;==;;;+/ZZZZZZFF1..."#-(;;;#%7%9%9::::; ;;;;;;;;;;;;;;;%;;;;;;;;;;;;;;;4 sg'B3E=E&D%$E&% E/D22"EE&EE& E=&E* *E=-E* .E==FFz (Sentry-wrapped))rr)r rQr-set_namer!AttributeError)r@rrKrkrPrIs ` r _sentry_task_factoryz+patch_asyncio.._sentry_task_factorys  !! 0%tT5;T " "# # # # # # "# J-!4)H)H)J)JNTD   $BBBCCCC!    Ks%A99 BBTzThere is no running asyncio loop so there is nothing Sentry can patch. Please make sure you call sentry_sdk.init() within a running asyncio loop for the AsyncioIntegration to work. See https://docs.sentry.io/platforms/python/integrations/asyncio/)r@rJrr$rKrrrT) rDrEget_task_factoryrrSset_task_factoryrFr r:)r@rnrIs @r patch_asynciorq|sP ')) 1133 $&? G G  F< -< ,< < # < < < < < < |8<4 233333     P       s8A+*A++B  B rctj}tj}|t }|3t ||jddd\}}tj|||S)NrDF)typehandled)client_options mechanism)hint) sysr2r3r4r[r\r r^ capture_event)r2r;rgeventrws r reres}|~~H  " $ $F(();<(U;;   t  T2222 Or"cFeZdZdZdeZd deddfdZed dZdS) r\rDzauto.function.Tr]rNc||_dSN)r])selfr]s r __init__zAsyncioIntegration.__init__s $r"c>ttdSr})rqrHr'r"r setup_oncezAsyncioIntegration.setup_oncesr")TrC) r __module__r identifierrZboolr staticmethodrr'r"r r\r\smJ *j * *F%%4%4%%%%\r"r\argsctj}|sdStjdt |i|}|d|jvr ||jd<dSdS)a Enable AsyncioIntegration with the provided options. This is useful in scenarios where Sentry needs to be initialized before an event loop is set up, but you still want to instrument asyncio once there is an event loop. In that case, you can sentry_sdk.init() early on without the AsyncioIntegration and then, once the event loop has been set up, execute: ```python from sentry_sdk.integrations.asyncio import enable_asyncio_integration async def async_entrypoint(): enable_asyncio_integration() ``` Any arguments provided will be passed to AsyncioIntegration() as is. If AsyncioIntegration has already patched the current event loop, this function won't have any effect. If AsyncioIntegration was provided in sentry_sdk.init(disabled_integrations=[...]), this function will ignore that and the integration will be enabled. NzSetting up integration asynciorD)r3r4r5r r?r\r integrations)rrKr;rgs r enable_asyncio_integrationrs4 " $ $F      L1222$d5f55K+++)4I&&&,+r")r#r$rr%rC) rIrr@rJrr$rKrrrL)rr)rrrKrrN)/r*rxr3sentry_sdk.constsrsentry_sdk.integrationsrrsentry_sdk.tracesrsentry_sdk.tracingrsentry_sdk.tracing_utilsrsentry_sdk.transportr sentry_sdk.utilsr r r r rrD asyncio.tasksr ImportErrortypingrrrcollections.abcrrrrsentry_sdk._typesrrstrr!r-rHrQrqrer\rr'r"r rs  ========******######??????3333330NNN"""""""000 ,. / //02111111111/))))))----------))))))8CH-...A5S(&(&(&(&V % %   6R R R R j"        *5*5*5*5*5*5s  A A