tni#XdZddlZddlmZmZddlmZmZmZm Z m Z ddl Z ddl m Z ddlmZddlmZddlmZe e e eefZGd d eZGd d eZd eddfdZddZddZddZddZdddefdZd dZ GddZ!Gdde Z"dS)!ae This integration ingests tracing data from native extensions written in Rust. Using it requires additional setup on the Rust side to accept a `RustTracingLayer` Python object and register it with the `tracing-subscriber` using an adapter from the `pyo3-python-tracing-subscriber` crate. For example: ```rust #[pyfunction] pub fn initialize_tracing(py_impl: Bound<'_, PyAny>) { tracing_subscriber::registry() .with(pyo3_python_tracing_subscriber::PythonCallbackLayerBridge::new(py_impl)) .init(); } ``` Usage in Python would then look like: ``` sentry_sdk.init( dsn=sentry_dsn, integrations=[ RustTracingIntegration( "demo_rust_extension", demo_rust_extension.initialize_tracing, event_type_mapping=event_type_mapping, ) ], ) ``` Each native extension requires its own integration. N)Enumauto)AnyCallableDictTupleOptional) Integration)should_send_default_pii)Span)SENSITIVE_DATA_SUBSTITUTEc"eZdZdZdZdZdZdZdS)RustTracingLevelTRACEDEBUGINFOWARNERRORN)__name__ __module__ __qualname__TraceDebugInfoWarnErrora/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/sentry_sdk/integrations/rust_tracing.pyrr.s' E E D D EEErrc^eZdZeZeZeZeZdS)EventTypeMappingN)rrrrIgnoreExc BreadcrumbEventrrrr!r!6s: TVVF $&&CJ DFFEEErr!levelreturnzsentry_sdk._types.LogLevelStrct|}|tjtjfvrdS|tjkrdS|tjkrdS|tjkrdSdS)Ndebuginfowarningerror)rrrrrr)r&s rtracing_level_to_sentry_levelr-=so U # #E !')9)?@@@w "' ' 'v "' ' 'y "( ( (wvreventDict[str, Any]c(|di}i}i}dD]}||vr ||||<t|dkr||d<i}|dgD]}||||<t|dkr||d<|S)Nmetadata) module_pathfilelinerrust_tracing_locationfieldsrust_tracing_fields)getlen)r.r1contextslocationfieldr6s rextract_contextsr=LsyyR((HHH0.. H  &uoHUO 8}}q,4() Fh++)) %((u  6{{Q*0&' Orc|di}|d}t|d}|d}t|}||||d}tj|dS)Nr1targetr&message)loggerr&r@r:)r8r-r= sentry_sdk capture_event)r.r1rAr&r@r: sentry_events r process_eventrE`syyR((H \\( # #F )(,,w*?*? @ @E',yy';';G&&H //L\*****rc$t|dSN)rE)r.s rprocess_exceptionrHrs%rct|did}|d}tj||dS)Nr1r&r@)r&r@)r-r8rBadd_breadcrumb)r.r&r@s rprocess_breadcrumbrKvsX )%))J*C*C*G*G*P*P Q QEii ""GE7;;;;;;rr1ct|dtjtjtjfvSNr&)rr8rrr)r1s rdefault_span_filterrN}s: HLL11 2 27 rc.t|d}|tjkr tjS|tjtjfvr tjS|tjtj fvr tj Stj SrM) rr8rr!r#rrr$rrr")r1r&s rdefault_event_type_mappingrPs~ X\\'22 3 3E  &&&## #(*:*?@ @ @** #)+;+AB B B&&&&rc eZdZeedfdeddddddfd Zd efd Zd ed dd dfdZ deded dfdZ deddd dfdZ dededdd dfdZ dS)RustTracingLayerNoriginevent_type_mappingBCallable[ [Dict[str, Any]], EventTypeMapping ] span_filter Callable[[Dict[str, Any]], bool]include_tracing_fieldsOptional[bool]c>||_||_||_||_dSrG)rSrTrVrX)selfrSrTrVrXs r__init__zRustTracingLayer.__init__s) "4&&<###rr'c:|jtn|jS)a9 By default, the values of tracing fields are not included in case they contain PII. A user may override that by passing `True` for the `include_tracing_fields` keyword argument of this integration or by setting `send_default_pii` to `True` in their Sentry client options. )rXr )r[s r_include_tracing_fieldsz(RustTracingLayer._include_tracing_fieldss&*2 $ % % %, rr. _span_state TraceStatecntj|}|di}||}|tjkrdS|tjkrt|dS|tjkrt|dS|tj krt|dSdS)Nr1) jsonloadsr8rTr!r"r#rHr$rKr%rE)r[r.r_deserialized_eventr1 event_types ron_eventzRustTracingLayer.on_events!Z..%))*b99,,X66 )0 0 0 F +/ / / 0 1 1 1 1 1 +6 6 6 1 2 2 2 2 2 +1 1 1 , - - - - -2 1rattrsspan_idctj|}|di}||sdS|d}|d}|d}||}n| ||d|}n||}nd}d||jd}t j} | j} | r| jd i|} n | j d i|} |d g} | D][} | r*| | || @| | t\| | _| | fS) Nr1r2namer@z::z function)oprjrSr6r) rbrcr8rVrSrBget_current_scopespan start_child start_spanr^set_datar )r[rgrhr1r2rjr@sentry_span_namekwargsscopeparent_sentry_span sentry_spanr6r<s r on_new_spanzRustTracingLayer.on_new_spans 5!!99Z,,)) 4ll=11 ||F##))I&&  &    $)9"-7777    #  * $k   ,.."Z  58,8BB6BBKK*%*44V44Kh++ G GE++-- G$$UEIIe,<,<====$$U,EFFFF  "K00r span_statecp|dS|\}}||tj_dSrG)finishrBrmrn)r[rhrxrurvs ron_closezRustTracingLayer.on_closesA   F*4'K.@ $&&+++rvaluesc|dS|\}}tj|}|D]K\}}|r|||0||t LdSrG)rbrcitemsr^rqr ) r[rhr|rx_parent_sentry_spanrvdeserialized_valueskeyvalues r on_recordzRustTracingLayer.on_records   F+5(["j00-3355 E EJC++-- E$$S%0000$$S*CDDDD  E Er) rrrrPrNstrr\boolr^rfrwr{rrrrrRrRs7 *:M37 = = =  = 8 =!1 = = = =        .c . . . . . .)1)1s)1|)1)1)1)1VAAA$AAAA E Ec E| EPT E E E E E ErrRc NeZdZdZeedfdedddddd d d f d ZeddZ dS)RustTracingIntegrationa Ingests tracing data from a Rust native extension's `tracing` instrumentation. If a project uses more than one Rust native extension, each one will need its own instance of `RustTracingIntegration` with an initializer function specific to that extension. Since all of the setup for this integration requires instance-specific state which is not available in `setup_once()`, setup instead happens in `__init__()`. N identifier initializerz"Callable[[RustTracingLayer], None]rTrUrVrWrXrYcl||_d|}t|||||_||jdS)Nzauto.function.rust_tracing.)rrR tracing_layer)r[rrrTrVrXrSs rr\zRustTracingIntegration.__init__sP%;z;;- & 5K    D&'''''rr'cdSrGrrrr setup_oncez!RustTracingIntegration.setup_onces r)r'N) rrr__doc__rPrNrr\ staticmethodrrrrrrs  "*:M37(((:( (8(!1(((($   \   rr)r.r/r'r/)r.r/r'N)r1r/r'r!)#rrbenumrrtypingrrrrr rBsentry_sdk.integrationsr sentry_sdk.scoper sentry_sdk.tracingr SentrySpansentry_sdk.utilsr r`rr!rr-r=rErHrKrrNrPrRrrrrrs5@ 77777777777777//////444444111111666666 eHZ0*<= > tt  1P    (++++$<<<<"2t ' ' ' 'gEgEgEgEgEgEgEgET      [      r