tni!ddlmZmZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZmZddlmZddlmZmZ ddlmZdd lmZmZdd lmZdd lmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&dd l'm(Z(m)Z)m*Z*ddl+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2n#e3$r edwxYwddl4m5Z5e5r ddl4m6Z6m7Z7m8Z8m9Z9d$dZ:d%d&dZ;dadd de?fd!Z@Gd"d#eZAdS)() get_client capture_event) Integration DidNotEnable)%register_external_propagation_context)Dsnloggerevent_from_exceptioncapture_internal_exceptions)VERSION EndpointType)Baggage)BAGGAGE_HEADER_NAMESENTRY_TRACE_HEADER_NAME)set_global_textmap)TracerProviderSpan)BatchSpanProcessor)OTLPSpanExporter)get_current_spanget_tracer_providerset_tracer_providerformat_trace_idformat_span_id SpanContextINVALID_SPAN_IDINVALID_TRACE_ID)Context get_current get_value)CarrierTSetterdefault_setter)SentryPropagator)SENTRY_BAGGAGE_KEYz+opentelemetry-distro[otlp] is not installed) TYPE_CHECKING)OptionalDictAnyTuplereturnOptional[Tuple[str, str]]ct}|jtks|jt krdSt |jt|jfS)zC Get the (trace_id, span_id) from opentelemetry if exists. N)rget_span_contexttrace_idrspan_idrrr)ctxs Y/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/sentry_sdk/integrations/otlp.pyotel_propagation_contextr39sY    - - / /C |'''3;/+I+It CL ) )>#++F+F GGNdsn Optional[str]ct}t|ts1tjdt}t |d}d}|rvt |dt}| tj }d| i}tjd|t||}t|}||dS)Nz?[OTLP] No TracerProvider configured by user, creating a new onezsentry.python/z X-Sentry-Authz[OTLP] Sending traces to )endpointheaders)r isinstancerr debugrrto_authr get_api_urlr OTLP_TRACES to_headerrradd_span_processor)r5tracer_providerr8r9auth otlp_exporterspan_processors r2setup_otlp_traces_exporterrEEs)++O o~ 6 6- VWWW(**O,,,HG =3xx : : :;;##L$<=="DNN$4$45 ;;;<<<$hHHHM' 66N&&~66666r4FcntjtrdS d fd }|t_d adS) ze Intercept otel's Span.record_exception to automatically capture those exceptions in Sentry. Nselfr exception BaseExceptionargsr)kwargsr+cTtt}|rq|jrjt 5t |tjtjdd\}}t||dddn #1swxYwY||g|Ri|dS)NF)typehandled)client_options mechanism)hint) rget_integrationOTLPIntegrationcapture_exceptionsr r options identifierr)rGrHrJrKotlp_integrationeventrQ_original_record_exceptions r2 _sentry_patched_record_exceptionzBsetup_capture_exceptions.._sentry_patched_record_exceptiongs&<<77HH  0 0 C 0,.. 0 02#-<<#7'6'AeTT t e$////  0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 #"4DTDDDVDDDDDsABBBT) rGrrHrIrJr)rKr)r+N)rrecord_exception_sentry_patched_exception)rZrYs @r2setup_capture_exceptionsr]]s "&!6  E E!0 E9> EJO E  E E E E E E=D $r4c.eZdZdZdef d d ZdS) SentryOTLPPropagatora We need to override the inject of the older propagator since that is SpanProcessor based. !!! Note regarding baggage: We cannot meaningfully populate a new baggage as a head SDK when we are using OTLP since we don't have any sort of transaction semantic to track state across a group of spans. For incoming baggage, we just pass it on as is so that case is correctly handled. Ncarrierr!contextOptional[Context]setterSetter[CarrierT]r+ctt}|dS|t}t |}|}|jsdSt|}||t|tt|}|It|tr6|} | r"||t| dSdSdSdSN)rrRrSrrr.is_valid_to_traceparentsetrr r%r:r serializer) rGr`rarcrW current_spancurrent_span_context sentry_tracebaggage baggage_datas r2injectzSentryOTLPPropagator.injects &<<77HH  # F ?!mmG'00 +<<>>#,  F&';<<  74lCCC.88  :gw#?#? ",,..L G 7$7FFFFF     G Gr4)r`r!rarbrcrdr+N)__name__ __module__ __qualname____doc__r#rpr4r2r_r_zsj  (,%3 GG%G# G  GGGGGGr4r_ span_contextrct|j}t|j}|jj}|d|d|rdndS)z< Helper method to generate the sentry-trace header. -10)rr0rr/ trace_flagssampled)rvr0r/r|s r2rhrhsT\122G|455H&.G < < < <'#:33s < <rsd00000000========BBBBBB 43333333,,,,,, F::::::<<<<<<<<AAAAAAVVVVVV                      RQQQQQOOOOOOOFFF ,D E EEF! 2111111111111 H H H H77777*"%%%%:'G'G'G'G'G+'G'G'GT=-=C====*#*#*#*#*#k*#*#*#*#*#s AB B