o Á¿i+ã@s|UddlZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl mZddlm Z ddlmZd Zd Ze ¡Zejeej<e eƒZedƒZeƒZeeeefed<d diZeeefed<dedejdefdd„ZGdd„deƒZ dedede!eeffdd„Z"dejdedeedededdfdd „Z#defd!d"„Z$deeeffd#d$„Z%d)d%d&„Z&d'd(„Z'dS)*éN)ÚAny)ÚOptional)ÚTypeVar)Ú get_logger)Útelemetry_writer)ÚTELEMETRY_NAMESPACE)ÚWrappingContextzdd.hookÚTracerÚTÚ _FN_PARAMSÚfinish_with_ancestorsÚ_finish_with_ancestorsÚ_API_TO_IMPL_NAMEÚwrapping_contextÚinstanceÚfn_namecCs<|jj›d|›}|tvrtt |j¡j ¡ƒt|<t|S)NÚ.) Ú __class__Ú__name__rÚlistÚinspectÚ signatureÚ__wrapped__Ú parametersÚkeys)rrrÚkey©rú^/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/ddtrace_api/patch.pyÚ_params_for_fnsrc@s(eZdZddd„Zdedefdd„ZdS) ÚDDTraceAPIWrappingContextBaseÚreturnNcsPˆ d¡}ˆjjj}t|t ||¡ˆ d¡fi‡fdd„tˆ||ƒDƒ¤ŽdS)NÚselfÚretvalcs i|]}|dkr|ˆ |¡“qS©r!)Ú get_local)Ú.0Úparamr#rrÚ ,s z@DDTraceAPIWrappingContextBase._handle_return..)r$Ú __frame__Úf_codeÚco_nameÚ_call_on_real_instancerÚgetr)r!Ústubrrr#rÚ_handle_return%s ý üz,DDTraceAPIWrappingContextBase._handle_returnÚvaluecCs2z| ¡W|StytjdddY|Sw)zHAlways return the original value no matter what our instrumentation doesz%Error handling instrumentation returnT)Úexc_info)r.Ú ExceptionÚlogÚdebug)r!r/rrrÚ __return__/s ýýz(DDTraceAPIWrappingContextBase.__return__)r N)rÚ __module__Ú__qualname__r.r r4rrrrr$s rÚargsÚkwargsr cs0dd„‰‡fdd„|Dƒ‡fdd„| ¡DƒfS)zfConvert all ddtrace_api.Span objects in the args/kwargs collections to their held ddtrace.Span objectscSst|tjƒr t|S|S©N)Ú isinstanceÚddtrace_apiÚSpanÚ _STUB_TO_REAL)ÚargrrrÚconvert<sz&_proxy_span_arguments..convertcsg|]}ˆ|ƒ‘qSrr)r%r>©r?rrÚ ?sz)_proxy_span_arguments..csi|] \}}|ˆ|ƒ“qSrr)r%ÚnameÚkwargr@rrr'?sz)_proxy_span_arguments..)Úitems)r7r8rr@rÚ_proxy_span_arguments9s(rEÚoperand_stubÚmethod_nameÚretval_from_apicOs^t||ƒ\}}tt||ƒ|i|¤Ž}|dur|t|<|jj›d|›}tjtj|dddS)a Call `method_name` on the real object corresponding to `operand_stub` with `args` and `kwargs` as arguments. Store the value that will be returned from the API call we're in the middle of, for the purpose of mapping from those Stub objects to their real counterparts. Nré)Ú namespacerBr/) rEÚgetattrr=rrrÚadd_count_metricrÚDD_TRACE_API)rFrGrHr7r8Úretval_from_implÚmetric_namerrrr+Bs r+cCsttddƒS)NÚ__version__Ú)rKr;rrrrÚget_versionTsrRcCsddiS)Nr;Ú*rrrrrÚ_supported_versionsXsrTcCsìttddƒrdS|dur|ttj<ttjjƒ ¡ttjjƒ ¡ttjj ƒ ¡ttjj ƒ ¡ttjjƒ ¡ttjj ƒ ¡ttjjƒ ¡ttjjƒ ¡ttjj ƒ ¡ttjjƒ ¡ttjjƒ ¡ttjjƒ ¡dt_dS)NÚ__datadog_patchFT)rKr;r=Útracerrr Ú start_spanÚwrapÚtraceÚset_tagsÚcurrent_spanÚcurrent_root_spanr<ÚfinishÚset_exc_inforÚ set_tracebackÚ __enter__Ú__exit__rU)rVrrrÚpatch\s" rbcCsøttddƒsdSdt_t tjj¡ ¡t tjj¡ ¡t tjj ¡ ¡t tjj ¡ ¡t tjj¡ ¡t tjj ¡ ¡t tjj¡ ¡t tjj¡ ¡t tjj ¡ ¡t tjj¡ ¡t tjj¡ ¡t tjj¡ ¡dt_dS)NrUF)rKr;rUrÚextractr rWÚunwraprYrZr[r\r<r]r^rr_r`rarrrrÚunpatchrs rer9)(rÚtypingrrrÚweakrefr;ÚddtraceÚddtrace.internal.loggerrÚddtrace.internal.telemetryrÚ$ddtrace.internal.telemetry.constantsrÚ!ddtrace.internal.wrapping.contextrÚ _DD_HOOK_NAMEÚ_TRACER_KEYÚWeakKeyDictionaryr=rVrr2r ÚdictrÚstrrÚ__annotations__rÚ_StubrrÚtuplerEr+rRrTrbrerrrrÚsN ÿÿÿÿÿ þ