o hv#@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__TraceDebugInfoWarnErrorrrf/var/www/html/pro-man-master/venv/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_pathfilelinerrust_tracing_locationfieldsrust_tracing_fields)getlen)r(r*contextslocationfieldr/rrrextract_contextsMs    r6cCsR|di}|d}t|d}|d}t|}||||d}t|dS)Nr*targetr&message)loggerr&r8r3)r1r'r6 sentry_sdk capture_event)r(r*r9r&r8r3 sentry_eventrrr process_eventas   r=cCs t|dSN)r=)r(rrrprocess_exceptionss r?cCs2t|did}|d}tj||ddS)Nr*r&r8)r&r8)r'r1r:add_breadcrumb)r(r&r8rrrprocess_breadcrumbws rAr*cCst|dtjtjtjfvSNr&)r r1rrr)r*rrrdefault_span_filter~s rCcCsPt|d}|tjkrtjS|tjtjfvrtjS|tjtj fvr%tj Stj SrB) r r1rrrrrr rrr)r*r&rrrdefault_event_type_mappings rDc @seZdZeedfdedeeeefge fdeeeefge fde e fddZ de fd d Z d ed eddfd dZdededefddZdededdfddZdedededdfddZdS)RustTracingLayerNoriginevent_type_mapping span_filterinclude_tracing_fieldscCs||_||_||_||_dSr>)rFrGrHrI)selfrFrGrHrIrrr__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)rIr )rJrrr_include_tracing_fieldss z(RustTracingLayer._include_tracing_fieldsr( _span_statecCstt|}|di}||}|tjkrdS|tjkr"t|dS|tjkr-t |dS|tj kr8t |dSdS)Nr*) jsonloadsr1rGrrrr?r rAr!r=)rJr(rMdeserialized_eventr* 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+namer8z::z function)oprUrFr/r) rNrOr1rHrFr:get_current_scopespan start_child start_spanrLset_datar )rJrSrTr*r+rUr8sentry_span_namekwargsscopeparent_sentry_span sentry_spanr/r5rrr on_new_spans<       zRustTracingLayer.on_new_span span_statecCs*|durdS|\}}||t_dSr>)finishr:rXrY)rJrTrcr`rarrron_closes zRustTracingLayer.on_closevaluesc CsV|durdS|\}}t|}|D]\}}|r"|||q||tqdSr>)rNrOitemsrLr\r ) rJrTrfrc_parent_sentry_spanradeserialized_valueskeyvaluerrr on_records zRustTracingLayer.on_record)rrrrDrCstrrrrrboolrrKrL TraceStaterRrbrerlrrrrrEs(  +rEc @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 initializerrGrHrIcCs.||_d|}t|||||_||jdS)Nzauto.function.rust_tracing.)rqrE tracing_layer)rJrqrrrGrHrIrFrrrrKs  zRustTracingIntegration.__init__r)cCsdSr>rrrrr setup_oncesz!RustTracingIntegration.setup_once)r)N)rrr__doc__rDrCrmrrErrrrnrrK staticmethodrtrrrrrps(  rp)#rurNenumrrtypingrrrrrr:sentry_sdk.integrationsr sentry_sdk.scoper sentry_sdk.tracingr SentrySpansentry_sdk.utilsr ror rr'rmr6r=r?rArnrCrDrErprrrrs*     " j