\j!dZddlZddlmZmZddlmZmZmZm Z ddl Z ddl m Z ddl mZddlmZddlmZGd d eZGd d eZd eddfdZddZddZddZddZdddefdZddZGddZGdde Z y) 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)AnyCallableDictOptional) Integration)should_send_default_pii)Span)SENSITIVE_DATA_SUBSTITUTEc eZdZdZdZdZdZdZy)RustTracingLevelTRACEDEBUGINFOWARNERRORN)__name__ __module__ __qualname__TraceDebugInfoWarnErrorN/root/env/lib/python3.12/site-packages/sentry_sdk/integrations/rust_tracing.pyrr,s E E D D ErrcDeZdZeZeZeZeZy)EventTypeMappingN)rrrrIgnoreExc BreadcrumbEventrrrr r 4s VF &CJ FErr levelreturnzsentry_sdk._types.LogLevelStrct|}|tjtjfvry|tjk(ry|tjk(ry|tj k(ryy)Ndebuginfowarningerror)rrrrrr)r%s rtracing_level_to_sentry_levelr,;se U #E !'')9)?)?@@ "'' ' "'' ' "(( (rDict[str, Any]c|jdi}i}i}dD]}||vs||||<t|dkDr||d<i}|jdgD]}|j|||<t|dkDr||d<|S)Nmetadata) module_pathfilelinerrust_tracing_locationfieldsrust_tracing_fields)getlen)eventr/contextslocationfieldr4s rextract_contextsr<JsyyR(HHH0. H &uoHUO. 8}q,4() Fh+) %(u ) 6{Q*0&' Orc|jdi}|jd}t|jd}|jd}t|}||||d}tj|y)Nr/targetr%message)loggerr%r?r9)r6r,r< sentry_sdk capture_event)r8r/r@r%r?r9 sentry_events r process_eventrD^spyyR(H \\( #F )(,,w*? @E',yy';G&H /L\*rct|yN)rD)r8s rprocess_exceptionrGps %rct|jdijd}|jd}tj||y)Nr/r%r?)r%r?)r,r6rAadd_breadcrumb)r8r%r?s rprocess_breadcrumbrJts@ )%))J*C*G*G*P QEii "GE7;rr/ct|jdtjtjtjfvSNr%)rr6rrr)r/s rdefault_span_filterrM{s= HLL1 27 rcdt|jd}|tjk(rtjS|tj tj fvrtjS|tjtjfvrtjStjSrL) rr6rr r"rrr#rrr!)r/r%s rdefault_event_type_mappingrOs X\\'2 3E  &&&### #((*:*?*?@ @*** #))+;+A+AB B&&&&&&rc eZdZeedfdeddddddfd Zd efd Zd ed dd dfdZ deded dfdZ ded dd dfdZ deded dd dfdZ y)RustTracingLayerNoriginevent_type_mappingBCallable[ [Dict[str, Any]], EventTypeMapping ] span_filter Callable[[Dict[str, Any]], bool]include_tracing_fieldsOptional[bool]c<||_||_||_||_yrF)rRrSrUrW)selfrRrSrUrWs r__init__zRustTracingLayer.__init__s# "4&&<#rr&cF|j tS|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. )rWr )rZs r_include_tracing_fieldsz(RustTracingLayer._include_tracing_fieldss-**2 $ % ,, rr8 sentry_span SentrySpancVtj|}|jdi}|j|}|tj k(ry|tj k(r t|y|tjk(r t|y|tjk(r t|yy)Nr/) jsonloadsr6rSr r!r"rGr#rJr$rD)rZr8r^deserialized_eventr/ event_types ron_eventzRustTracingLayer.on_events!ZZ.%))*b9,,X6 )00 0  +// / 0 1 +66 6 1 2 +11 1 , -2rattrsspan_idzOptional[SentrySpan]c:tj|}|jdi}|j|sy|jd}|jd}|jd}||}n| ||d|}n||}nd}d||jd}t j d i|} |jd g} | D]J} |jr"| j| |j| 5| j| tL| j| S) Nr/r0namer?z::z function)oprirRr4r) rarbr6rUrRrA start_spanr]set_datar __enter__) rZrfrgr/r0rir?sentry_span_namekwargsr^r4r;s r on_new_spanzRustTracingLayer.on_new_spans+ 5!99Z,)ll=1 ||F#))I&  &   $)9"-b7   # * $kk  !++5f5 h+ GE++-$$UEIIe,<=$$U,EF  G rc0|y|jdddyrF)__exit__)rZrgr^s ron_closezRustTracingLayer.on_closes   T4.rvaluesc|ytj|}|jD]>\}}|jr|j ||)|j |t @yrF)rarbitemsr]rmr )rZrgrur^deserialized_valueskeyvalues r on_recordzRustTracingLayer.on_recordse   "jj0-335 EJC++-$$S%0$$S*CD  Er) rrrrOrMstrr[boolr]rerqrtr{rrrrQrQs *:M37 = =  = 8 =!1 =     .c . . .##s#7M#J//,/4/ E Ec E EQU ErrQc FeZdZdZeedfdedddddd d d f d Zedd Z y)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]rSrTrUrVrWrXcf||_d|}t|||||_||jy)Nzauto.function.rust_tracing.)rrQ tracing_layer)rZrrrSrUrWrRs rr[zRustTracingIntegration.__init__s?%.zl;- & 5K  D&&'rcyrFrrrr setup_oncez!RustTracingIntegration.setup_onces r)r&N) rrr__doc__rOrMr|r[ staticmethodrrrrrrs[ "*:M37((:( (8(!1($  rr)r8r-r&r-)r8r-r&N)r/r-r&r )!rraenumrrtypingrrrrrAsentry_sdk.integrationsr sentry_sdk.scoper sentry_sdk.tracingr r_sentry_sdk.utilsr rr r|r,r<rDrGrJr}rMrOrQrrrrrs@ 00/416tt  1P (+$<"2t '^E^EB  [  r