o ai!@sdZddlZddlmZmZddlmZmZmZm Z ddl Z ddl m Z ddl mZddlmZddlmZGd d d eZGd d d eZd eddfddZd%ddZd&ddZd&ddZd&ddZdddefddZd'dd ZGd!d"d"ZGd#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)AnyCallableDictOptional) Integration)should_send_default_pii)Span)SENSITIVE_DATA_SUBSTITUTEc@s eZdZdZdZdZdZdZdS)RustTracingLevelTRACEDEBUGINFOWARNERRORN)__name__ __module__ __qualname__TraceDebugInfoWarnErrorrrb/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/sentry_sdk/integrations/rust_tracing.pyr ,s r c@s$eZdZeZeZeZeZdS)EventTypeMappingN)rrrrIgnoreExc BreadcrumbEventrrrrr4s  rlevelreturnzsentry_sdk._types.LogLevelStrcCsJt|}|tjtjfvrdS|tjkrdS|tjkrdS|tjkr#dSdS)Ndebuginfowarningerror)r rrrrr)r!rrrtracing_level_to_sentry_level;s   r'eventDict[str, Any]cCs|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_contextsJs    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_event^s   r=cCs t|dSN)r=)r(rrrprocess_exceptionps r?cCs2t|did}|d}tj||ddS)Nr*r!r8)r!r8)r'r1r:add_breadcrumb)r(r!r8rrrprocess_breadcrumbts rAr*cCst|dtjtjtjfvSNr!)r r1rrr)r*rrrdefault_span_filter{s rCcCsPt|d}|tjkrtjS|tjtjfvrtjS|tjtj fvr%tj Stj SrB) r r1rrrrrrrrr)r*r!rrrdefault_event_type_mappings rDc @seZdZeedfdeddddddfd d Zd efd d Zdeddd dfddZ deded dfddZ deddd dfddZ dededdd dfddZ dS)RustTracingLayerNoriginevent_type_mappingBCallable[ [Dict[str, Any]], EventTypeMapping ] span_filter Callable[[Dict[str, Any]], bool]include_tracing_fieldsOptional[bool]cCs||_||_||_||_dSr>)rFrGrIrK)selfrFrGrIrKrrr__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)rKr )rMrrr_include_tracing_fieldss z(RustTracingLayer._include_tracing_fieldsr( sentry_span SentrySpancCstt|}|di}||}|tjkrdS|tjkr"t|dS|tjkr-t |dS|tj kr8t |dSdS)Nr*) jsonloadsr1rGrrrr?rrAr r=)rMr(rPdeserialized_eventr* event_typerrron_events          zRustTracingLayer.on_eventattrsspan_idzOptional[SentrySpan]c Cst|}|di}||sdS|d}|d}|d}|dur(|}n|dur8|dur8|d|}n |dur?|}nd}d||jd}tjd i|} |d g} | D]} |rh| | || qX| | t qX| | S) Nr*r+namer8z::z function)oprYrFr/r) rRrSr1rIrFr: start_spanrOset_datar __enter__) rMrWrXr*r+rYr8sentry_span_namekwargsrPr/r5rrr on_new_spans4       zRustTracingLayer.on_new_spancCs|durdS|ddddSr>)__exit__)rMrXrPrrron_closeszRustTracingLayer.on_closevaluescCsN|durdSt|}|D]\}}|r|||q||tqdSr>)rRrSitemsrOr]r )rMrXrdrPdeserialized_valueskeyvaluerrr on_records zRustTracingLayer.on_record) rrrrDrCstrrNboolrOrVrarcrirrrrrEs$  %rEc @sDeZdZdZeedfdedddddd d d f d d ZedddZ 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]rGrHrIrJrKrLcCs.||_d|}t|||||_||jdS)Nzauto.function.rust_tracing.)rmrE tracing_layer)rMrmrnrGrIrKrFrrrrNs  zRustTracingIntegration.__init__r"cCsdSr>rrrrr setup_oncesz!RustTracingIntegration.setup_once)r"N) rrr__doc__rDrCrjrN staticmethodrprrrrrls$ rl)r(r)r"r))r(r)r"N)r*r)r"r)!rqrRenumrrtypingrrrrr:sentry_sdk.integrationsrsentry_sdk.scoper sentry_sdk.tracingr rQsentry_sdk.utilsr r rrjr'r6r=r?rArkrCrDrErlrrrrs(           a