xjddlmZmZddlmZmZddlmZmZddl m Z ddl m Z m Z ddlmZddlmZmZmZmZ ddlmZmZmZdd lmZdd lmZdd lmZmZm Z dd l!m"Z"m#Z#dd l$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2n#e3$r edwxYwddl4m5Z5e5r ddl4m6Z6m7Z7m8Z8m9Z9d%dZ: d& d'dZ;dad d!de?fd"Z@Gd#d$eZAdS))) capture_event get_client)VERSION EndpointType) DidNotEnable Integration)%register_external_propagation_context)BAGGAGE_HEADER_NAMESENTRY_TRACE_HEADER_NAME)Baggage)Dsncapture_internal_exceptionsevent_from_exceptionlogger)Context get_current get_value)OTLPSpanExporter)set_global_textmap)CarrierTSetterdefault_setter)SpanTracerProvider)BatchSpanProcessor)INVALID_SPAN_IDINVALID_TRACE_ID SpanContextformat_span_idformat_trace_idget_current_spanget_tracer_providerset_tracer_provider)SENTRY_BAGGAGE_KEY)SentryPropagatorz+opentelemetry-distro[otlp] is not installed) TYPE_CHECKING)AnyDictOptionalTuplereturnOptional[Tuple[str, str]]ct}|jtks|jt krdSt |jt|jfS)zC Get the (trace_id, span_id) from opentelemetry if exists. N)r!get_span_contexttrace_idrspan_idrr r)ctxs MC:\Projects\Historical\venv\Lib\site-packages\sentry_sdk/integrations/otlp.pyotel_propagation_contextr36sY    - - / /C |'''3;/+I+It CL ) )>#++F+F GGNdsn Optional[str] collector_urlcHt}t|ts1tjdt}t |d}d}|r|}tjd|nx|rvt |dt}| tj }d| i}tjd|t||}t|}||dS)Nz?[OTLP] No TracerProvider configured by user, creating a new onez&[OTLP] Sending traces to collector at zsentry.python/z X-Sentry-Authz[OTLP] Sending traces to )endpointheaders)r" isinstancerrdebugr#r to_authr get_api_urlr OTLP_TRACES to_headerrradd_span_processor)r5r7tracer_providerr9r:auth otlp_exporterspan_processors r2setup_otlp_traces_exporterrFBs$*++O o~ 6 6- VWWW(**O,,,HG=  HhHHIIII =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_exceptionsrroptions identifierr)rHrIrKrLotlp_integrationeventrR_original_record_exceptions r2 _sentry_patched_record_exceptionzBsetup_capture_exceptions.._sentry_patched_record_exceptionis&<<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) rHrrIrJrKr'rLr'r+N)rrecord_exception_sentry_patched_exception)r[rZs @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. NcarrierrcontextOptional[Context]setterSetter[CarrierT]r+ctt}|dS|t}t |}|}|jsdSt|}||t|tt|}|It|tr6|} | r"||t| dSdSdSdSN)rrSrTrr!r.is_valid_to_traceparentsetr rr$r;r serializer ) rHrarbrdrX current_spancurrent_span_context sentry_tracebaggage baggage_datas r2injectzSentryOTLPPropagator.injects &<<77HH  # F ?!mmG'00 +<<>>#,  F&';<<  74lCCC.88  :gw#?#? ",,..L G 7$7FFFFF     G Gr4)rarrbrcrdrer+N)__name__ __module__ __qualname____doc__rrqr4r2r`r`|sj  (,%3 GG%G# G  GGGGGGr4r` span_contextrct|j}t|j}|jj}|d|d|rdndS)z< Helper method to generate the sentry-trace header. -10)rr0r r/ trace_flagssampled)rwr0r/r}s r2ririsT\122G|455H&.G < < < <'#:33s < <||_||_||_||_dSrg)rFr7rrU)rHrFr7rrUs r2__init__zOTLPIntegration.__init__s++E'* 0"4r4cVtjdttdS)Nz.[OTLP] Setting up trace linking for all events)rr<r r3rvr4r2 setup_oncezOTLPIntegration.setup_onces( EFFF-.FGGGGGr4rVOptional[Dict[str, Any]]c"|jrCtjd|r|dnd}t||j|jr/tjdt ttdS)Nz[OTLP] Setting up OTLP exporterr5)r7z4[OTLP] Setting up propagator for distributed tracing) rFrr<getr7rrr`r^)rHrVr5s r2setup_once_with_optionsz'OTLPIntegration.setup_once_with_optionss  * N L: ; ; ;9@#J7;;u#5#5#5dC &s$:L M M M M   7 LO P P P 355 6 6 6 """""r4)TNTFr+Nrg)rVrr+N) rrrsrtrurWboolr staticmethodrrrvr4r2rTrTs  J,0)-!%#( 5 5$( 5' 5 5 ! 5  5 5 5 5HHH\H 59 #1 #  # # # # # #r4rT)r+r,)NN)r5r6r7r6r+Nr)B sentry_sdkrrsentry_sdk.constsrrsentry_sdk.integrationsrrsentry_sdk.scoper sentry_sdk.tracingr r sentry_sdk.tracing_utilsr sentry_sdk.utilsr rrropentelemetry.contextrrr5opentelemetry.exporter.otlp.proto.http.trace_exporterropentelemetry.propagater!opentelemetry.propagators.textmaprrropentelemetry.sdk.tracerropentelemetry.sdk.trace.exportropentelemetry.tracerrrrr r!r"r#,sentry_sdk.integrations.opentelemetry.constsr$0sentry_sdk.integrations.opentelemetry.propagatorr% ImportErrortypingr&r'r(r)r*r3rFr]r^r`strrirTrvr4r2rs0000000033333333========BBBBBB-,,,,,F WVVVVV:::::: =<<<<<<<AAAAAA                    POOOOOQQQQQQQFFF ,D E EEF! 2111111111111 H H H HCG7 70?7 77774"%%%%:'G'G'G'G'G+'G'G'GT=-=C====.#.#.#.#.#k.#.#.#.#.#s AB B