o 0 iv#@srdZddlZddlmZmZddlmZmZmZm Z m Z ddl Z ddl m Z ddlmZddlmZddlmZe e e eefZGd d d eZGd d d eZd dZdeeefdeeeffddZdeeefddfddZdeeefddfddZdeeefddfddZdeeefdefddZdeeefdefddZ GdddZ!Gd d!d!e 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@s eZdZdZdZdZdZdZdS)RustTracingLevelTRACEDEBUGINFOWARNERRORN)__name__ __module__ __qualname__TraceDebugInfoWarnErrorrrp/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/sentry_sdk/integrations/rust_tracing.pyr .s r c@s$eZdZeZeZeZeZdS)EventTypeMappingN)rrrrIgnoreExc BreadcrumbEventrrrrr6s  rcCsJt|}|tjtjfvrdS|tjkrdS|tjkrdS|tjkr#dSdS)Ndebuginfowarningerror)r rrrrr)levelrrrtracing_level_to_sentry_level=s   r'eventreturncCs|di}i}i}dD] }||vr||||<q t|dkr#||d<i}|dgD] }||||<q+t|dkr?||d<|S)Nmetadata) module_pathfilelinerZrust_tracing_locationfieldsZrust_tracing_fields)getlen)r(r*contextslocationfieldr.rrrextract_contextsMs    r4cCsR|di}|d}t|d}|d}t|}||||d}t|dS)Nr*targetr&message)loggerr&r6r1)r/r'r4 sentry_sdkZ capture_event)r(r*r7r&r6r1Z sentry_eventrrr process_eventas   r9cCs t|dSN)r9)r(rrrprocess_exceptionss r;cCs2t|did}|d}tj||ddS)Nr*r&r6)r&r6)r'r/r8Zadd_breadcrumb)r(r&r6rrrprocess_breadcrumbws r<r*cCst|dtjtjtjfvSNr&)r r/rrr)r*rrrdefault_span_filter~s r>cCsPt|d}|tjkrtjS|tjtjfvrtjS|tjtj fvr%tj Stj Sr=) r r/rrrrrr rrr)r*r&rrrdefault_event_type_mappings r?c @seZdZeedfdedeeeefge fdeeeefge fde e fddZ de fd d Z d ed eddfd dZdededefddZdededdfddZdedededdfddZdS)RustTracingLayerNoriginevent_type_mapping span_filterinclude_tracing_fieldscCs||_||_||_||_dSr:)rArBrCrD)selfrArBrCrDrrr__init__s  zRustTracingLayer.__init__r)cCs|jdurtS|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. N)rDr )rErrr_include_tracing_fieldss z(RustTracingLayer._include_tracing_fieldsr( _span_statecCstt|}|di}||}|tjkrdS|tjkr"t|dS|tjkr-t |dS|tj kr8t |dSdS)Nr*) jsonloadsr/rBrrrr;r r<r!r9)rEr(rHZdeserialized_eventr*Z event_typerrron_events          zRustTracingLayer.on_eventattrsspan_idcCst|}|di}||sdS|d}|d}|d}|dur(|}n|dur8|dur8|d|}n |dur?|}nd}d||jd}t} | j} | rZ| jd i|} n| j d i|} |d g} | D]} | rz| | || qj| | t qj| | _| | fS) Nr*r+namer6z::z function)oprNrAr.r) rIrJr/rCrAr8get_current_scopespanZ start_childZ start_spanrGset_datar )rErLrMr*r+rNr6Zsentry_span_namekwargsscopeparent_sentry_span sentry_spanr.r3rrr on_new_spans<       zRustTracingLayer.on_new_span span_statecCs*|durdS|\}}||t_dSr:)finishr8rQrR)rErMrYrVrWrrron_closes zRustTracingLayer.on_closevaluesc CsV|durdS|\}}t|}|D]\}}|r"|||q||tqdSr:)rIrJitemsrGrSr ) rErMr\rYZ_parent_sentry_spanrWZdeserialized_valueskeyvaluerrr on_records zRustTracingLayer.on_record)rrrr?r>strrrrrboolrrFrG TraceStaterKrXr[r`rrrrr@s(  +r@c @sveZdZdZeedfdedeegdfdee ee fge fdee ee fge fde e f dd Zed d d ZdS)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 initializerrBrCrDcCs.||_d|}t|||||_||jdS)Nzauto.function.rust_tracing.)rer@Z tracing_layer)rErerfrBrCrDrArrrrFs  zRustTracingIntegration.__init__r)cCsdSr:rrrrr setup_oncesz!RustTracingIntegration.setup_once)r)N)rrr__doc__r?r>rarr@rrrrbrrF staticmethodrgrrrrrds(  rd)#rhrIenumrrtypingrrrrrr8Zsentry_sdk.integrationsr Zsentry_sdk.scoper Zsentry_sdk.tracingr Z SentrySpanZsentry_sdk.utilsr rcr rr'rar4r9r;r<rbr>r?r@rdrrrrs*     " j