o i#@sdZddlZddlmZmZddlmZmZmZm Z m Z ddl Z ddl m Z ddlmZddlmZddlmZe e e eefZGd d d eZGd d d eZd eddfddZd%ddZd&ddZd&ddZd&ddZdddefddZd'dd Z Gd!d"d"Z!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__TraceDebugInfoWarnErrorrra/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/sentry_sdk/integrations/rust_tracing.pyr .s r c@s$eZdZeZeZeZeZdS)EventTypeMappingN)rrrrIgnoreExc BreadcrumbEventrrrrr6s  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+contextslocationfieldr0rrrextract_contextsLs    r7cCsR|di}|d}t|d}|d}t|}||||d}t|dS)Nr+targetr"message)loggerr"r9r4)r2r(r7 sentry_sdk capture_event)r)r+r:r"r9r4 sentry_eventrrr process_event`s   r>cCs t|dSN)r>)r)rrrprocess_exceptionrs r@cCs2t|did}|d}tj||ddS)Nr+r"r9)r"r9)r(r2r;add_breadcrumb)r)r"r9rrrprocess_breadcrumbvs rBr+cCst|dtjtjtjfvSNr")r r2rrr)r+rrrdefault_span_filter}s rDcCsPt|d}|tjkrtjS|tjtjfvrtjS|tjtj fvr%tj Stj SrC) r r2rrrrrr rrr)r+r"rrrdefault_event_type_mappings rEc @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?)rGrHrJrL)selfrGrHrJrLrrr__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)rLr )rNrrr_include_tracing_fieldss z(RustTracingLayer._include_tracing_fieldsr) _span_state TraceStatecCstt|}|di}||}|tjkrdS|tjkr"t|dS|tjkr-t |dS|tj kr8t |dSdS)Nr+) jsonloadsr2rHrrrr@r rBr!r>)rNr)rQdeserialized_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,namer9z::z function)oprZrGr0r) rSrTr2rJrGr;get_current_scopespan start_child start_spanrPset_datar )rNrXrYr+r,rZr9sentry_span_namekwargsscopeparent_sentry_span sentry_spanr0r6rrr on_new_spans<       zRustTracingLayer.on_new_span span_statecCs*|durdS|\}}||t_dSr?)finishr;r]r^)rNrYrhrerfrrron_closes zRustTracingLayer.on_closevaluesc CsV|durdS|\}}t|}|D]\}}|r"|||q||tqdSr?)rSrTitemsrPrar ) rNrYrkrh_parent_sentry_spanrfdeserialized_valueskeyvaluerrr on_records zRustTracingLayer.on_record) rrrrErDstrrOboolrPrWrgrjrqrrrrrFs$  +rFc @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]rHrIrJrKrLrMcCs.||_d|}t|||||_||jdS)Nzauto.function.rust_tracing.)rurF tracing_layer)rNrurvrHrJrLrGrrrrOs  zRustTracingIntegration.__init__r#cCsdSr?rrrrr setup_oncesz!RustTracingIntegration.setup_once)r#N) rrr__doc__rErDrrrO staticmethodrxrrrrrts$ rt)r)r*r#r*)r)r*r#N)r+r*r#r)#ryrSenumrrtypingrrrrrr;sentry_sdk.integrationsr sentry_sdk.scoper sentry_sdk.tracingr SentrySpansentry_sdk.utilsr rRr rrrr(r7r>r@rBrsrDrErFrtrrrrs*           j