j#dZddlZddlmZmZddlmZmZmZm Z m Z ddl Z ddl m Z ddlmZddlmZddlmZe e e eefZGd d eZGd d eZd ZdeeefdeeeffdZdeeefddfdZdeeefddfdZdeeefddfdZdeeefdefdZdeeefdefdZ GddZ!Gdde 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)AnyCallableDictTupleOptional) Integration)should_send_default_pii)Span)SENSITIVE_DATA_SUBSTITUTEcTeZdZUdZeed<dZeed<dZeed<dZeed<d Z eed <y ) RustTracingLevelTRACETraceDEBUGDebugINFOInfoWARNWarnERRORErrorN) __name__ __module__ __qualname__rstr__annotations__rrrra/var/www/photos/backend/venv/lib/python3.12/site-packages/sentry_sdk/integrations/rust_tracing.pyrr.s4E3E3D#D#E3r rcDeZdZeZeZeZeZy)EventTypeMappingN)rrrrIgnoreExc BreadcrumbEventrr r!r#r#6s VF &CJ FEr r#ct|}|tjtjfvry|tjk(ry|tjk(ry|tj k(ryy)Ndebuginfowarningerror)rrrrrr)levels r!tracing_level_to_sentry_levelr.=se U #E !'')9)?)?@@ "'' ' "'' ' "(( (r eventreturnc|jdi}i}i}dD]}||vs||||<t|dkDr||d<i}|jdgD]}|j|||<t|dkDr||d<|S)Nmetadata) module_pathfilelinerrust_tracing_locationfieldsrust_tracing_fields)getlen)r/r2contextslocationfieldr7s r!extract_contextsr>MsyyR(HHH0. H &uoHUO. 8}q,4() Fh+) %(u ) 6{Q*0&' Or c|jdi}|jd}t|jd}|jd}t|}||||d}tj|y)Nr2targetr-message)loggerr-rAr;)r9r.r> sentry_sdk capture_event)r/r2rBr-rAr; sentry_events r! process_eventrFaspyyR(H \\( #F )(,,w*? @Eii "G&H L\*r ct|yN)rF)r/s r!process_exceptionrIss %r ct|jdijd}|jd}tj||y)Nr2r-rA)r-rA)r.r9rCadd_breadcrumb)r/r-rAs r!process_breadcrumbrLws@ )%))J*C*G*G*P QEii "GE7;r r2ct|jdtjtjtjfvSNr-)rr9rrr)r2s r!default_span_filterrO~s= HLL1 27 r cdt|jd}|tjk(rtjS|tj tj fvrtjS|tjtjfvrtjStjSrN) rr9rr#r%rrr&rrr$)r2r-s r!default_event_type_mappingrQs X\\'2 3E  &&&### #((*:*?*?@ @*** #))+;+A+AB B&&&&&&r c eZdZeedfdedeeeefge fdeeeefge fde e fdZ de fdZ d ed eddfd Zd ed edefdZd ededdfdZd edededdfdZy)RustTracingLayerNoriginevent_type_mapping span_filterinclude_tracing_fieldsc<||_||_||_||_yrH)rTrUrVrW)selfrTrUrVrWs r!__init__zRustTracingLayer.__init__s# "4&&<#r r0cF|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 )rYs r!_include_tracing_fieldsz(RustTracingLayer._include_tracing_fieldss-**2 $ % ,, r r/ _span_statecVtj|}|jdi}|j|}|tj k(ry|tj k(r t|y|tjk(r t|y|tjk(r t|yy)Nr2) jsonloadsr9rUr#r$r%rIr&rLr'rF)rYr/r]deserialized_eventr2 event_types r!on_eventzRustTracingLayer.on_events!ZZ.%))*b9,,X6 )00 0  +// / 0 1 +66 6 1 2 +11 1 , -2r attrsspan_idctj|}|jdi}|j|sy|jd}|jd}|jd}||}n| ||d|}n||}nd}d||jd}t j } | j} | r| jd i|} n| jd i|} |jd g} | D]J} |jr"| j| |j| 5| j| tL| | _| | fS) Nr2r3namerAz::z function)oprgrTr7r) r_r`r9rVrTrCget_current_scopespan start_child start_spanr\set_datar )rYrdrer2r3rgrAsentry_span_namekwargsscopeparent_sentry_span sentry_spanr7r=s r! on_new_spanzRustTracingLayer.on_new_spans^ 5!99Z,)ll=1 ||F#))I&  &   $)9"-b7   # * $kk  ,,."ZZ 8,88B6BK*%**4V4Kh+ GE++-$$UEIIe,<=$$U,EF  G ! "K00r span_statecf|y|\}}|j|tj_yrH)finishrCrjrk)rYrerurrrss r!on_closezRustTracingLayer.on_closes3   *4'K.@ $$&+r valuesc|y|\}}tj|}|jD]>\}}|jr|j ||)|j |t @yrH)r_r`itemsr\rnr ) rYreryru_parent_sentry_spanrsdeserialized_valueskeyvalues r! on_recordzRustTracingLayer.on_recordsp   +5(["jj0-335 EJC++-$$S%0$$S*CD  Er )rrrrQrOrrrrr#boolr rZr\ TraceStatercrtrxrrr r!rSrSs '8K15 = =% #s(^ . .  = tCH~.45 =!) =     .c . .t .)1)1s)1z)1VAAAA E Ec Ez Ed Er rSc eZdZdZeedfdedeegdfdee ee fge fdee ee fge fde e f dZed d Zy) 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 initializerrUrVrWcf||_d|}t|||||_||jy)Nzauto.function.rust_tracing.)rrS tracing_layer)rYrrrUrVrWrTs r!rZzRustTracingIntegration.__init__s?%.zl;- & 5K  D&&'r cyrHrrr r! setup_oncez!RustTracingIntegration.setup_onces r )r0N)rrr__doc__rQrOrrrSrrr#rr rZ staticmethodrrr r!rrs " '8K15((/0$67(% #s(^ . . (tCH~.45(!)($  r r)#rr_enumrrtypingrrrrr rCsentry_sdk.integrationsr sentry_sdk.scoper sentry_sdk.tracingr SentrySpansentry_sdk.utilsr rrr#r.rr>rFrIrLrrOrQrSrrr r!rs$@ 77/416 eHZ0*<= > tt  DcNtCH~(+c3h+D+$T#s(^