o Zh1 @sddlZddlZddlmZmZddlZddlmZddlmZm Z m Z ddl m Z m Z mZddlmZddlmZerddlZddlmZmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZedZ edZ!ddl"Zddlm#Z#m$Z$m%Z%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-e)gZ.e,e,e,e*e)e,e*e*e*d Z/Gdd d e0Z1Gd!d"d"e0Z2Gd#d$d$e2Z3Gd%d&d&e2Z4ered-d'd(Z5ed)d(Z5d-d*d(Z5dd+l6m7Z7m8Z8m9Z9m:Z:m;Z;dd,lr?r@rArDrHrFrGrJr isinstancefloatr rBrrCAttributeErrorrEr1rK) r+r;r<r=r>r?r@rArHrDrLcontaining_transactionrBr,r,r-r.s2    z Span.__init__cCs|jdur t||_dSdSr/)r1r%r*r,r,r-init_span_recorders zSpan.init_span_recordercCs|j}|dur t}|_|Sr/)rKLocalAggregatorr+rvr,r,r-_get_local_aggregators zSpan._get_local_aggregatorcCs&d|jj|j|j|j|j|j|jfS)NzS<%s(op=%r, description:%r, trace_id=%r, span_id=%r, parent_span_id=%r, sampled=%r)>) __class__r5r@rAr;r<r=r?r+r,r,r-__repr__sz Span.__repr__cCs8|jptjj}|jd\}}|j}||_|||f|_|S)N)rH sentry_sdkHubcurrent_stackr3rI)r+rH_scopeold_spanr,r,r- __enter__s  zSpan.__enter__cCs6|dur |d|j\}}}|`||||_dS)Ninternal_error) set_statusrIfinishr3)r+tyvaluetbrHrhrir,r,r-__exit__s     z Span.__exit__cCs|jS)zThe ``Transaction`` that this span belongs to. The ``Transaction`` is the root of the span tree, so one could also think of this ``Transaction`` as the "root span".)rJr`r,r,r-rYs zSpan.containing_transactioncKsz|jptjj}|j}|o|jd}||krtS|d|jt d|j |j |j d|}|j o3|j j }|r;|||S)a Start a sub-span from the current span or transaction. Takes the same arguments as the initializer of :py:class:`Span`. The trace id, sampling decision, transaction pointer, and span recorder are inherited from the current span/transaction. instrumenterr?)r;r=rYNr,)rHrcrdreclientoptionsNoOpSpan setdefaultr?r:r;r<rYr1r4)r+rrrQrHrsZconfiguration_instrumenterchildZ span_recorderr,r,r- start_childs$   zSpan.start_childcKstd|jdi|S)zGDEPRECATED: use :py:meth:`sentry_sdk.tracing.Span.start_child` instead.z^Deprecated: use Span.start_child instead of Span.new_span. This will be removed in the future.Nr,)rwarningrxr+rQr,r,r-new_spansz Span.new_spancKs(|tur tdtjt|fi|S)a( Create a Transaction with the given params, then add in data pulled from the ``sentry-trace`` and ``baggage`` headers from the environ (if any) before returning the Transaction. This is different from :py:meth:`~sentry_sdk.tracing.Span.continue_from_headers` in that it assumes header names in the form ``HTTP_HEADER_NAME`` - such as you would get from a WSGI/ASGI environ - rather than the form ``header-name``. :param environ: The ASGI/WSGI environ to pull information from. zXDeprecated: use Transaction.continue_from_environ instead of Span.continue_from_environ.)r:rryrOcontinue_from_headersEnvironHeaders)rPenvironrQr,r,r-continue_from_environs zSpan.continue_from_environcKsp|tur tdt|t}|t|it|t }|dur,||| t di|}d|_ |S)z Create a transaction with the given params (including any data pulled from the ``sentry-trace`` and ``baggage`` headers). :param headers: The dictionary with the HTTP headers to pull information from. zXDeprecated: use Transaction.continue_from_headers instead of Span.continue_from_headers.NFr,) r:rryBaggageZfrom_incoming_headergetBAGGAGE_HEADER_NAMEupdateextract_sentrytrace_dataSENTRY_TRACE_HEADER_NAMEfreezerOr>)rPheadersrQrZsentrytrace_kwargsrLr,r,r-r|0s zSpan.continue_from_headersccs>t|fV|jr|j}|rt|fVdSdSdS)z Creates a generator which returns the span's ``sentry-trace`` and ``baggage`` headers. If the span's containing transaction doesn't yet have a ``baggage`` value, this will cause one to be generated and stored. N)rto_traceparentrY get_baggage serializer)r+rr,r,r- iter_headersZszSpan.iter_headerscKs(td|s dS|jt|ifi|S)a DEPRECATED: Use :py:meth:`sentry_sdk.tracing.Span.continue_from_headers`. Create a ``Transaction`` with the given params, then add in data pulled from the given ``sentry-trace`` header value before returning the ``Transaction``. zwDeprecated: Use Transaction.continue_from_headers(headers, **kwargs) instead of from_traceparent(traceparent, **kwargs)N)rryr|r)rP traceparentrQr,r,r-from_traceparenths zSpan.from_traceparentcCsN|jdurd}n |jdurd}nd}d|j|jf}|dur%|d|f7}|S)NT1F0z%s-%sz-%s)r?r;r<)r+r?rr,r,r-rs  zSpan.to_traceparentcCs|jr|jSdS)zReturns the :py:class:`~sentry_sdk.tracing_utils.Baggage` associated with this ``Span``, if any. (Taken from the root of the span tree.) N)rYrr`r,r,r- to_baggages zSpan.to_baggagecC||j|<dSr/)rFr+keyror,r,r-set_tagz Span.set_tagcCrr/)rGrr,r,r-set_datarz Span.set_datacCs ||_dSr/rDr+ror,r,r-rl zSpan.set_statuscCsT|dt||tj||dkr|ddSd|kr$dkronnI|dkr1|ddS|dkr<|ddS|d krG|d dS|d krR|d dS|d kr]|ddS|dkrh|ddS|ddSd|krydkrnn(|dkr|ddS|dkr|ddS|dkr|ddS|ddS|ddS)Nzhttp.status_codeiokiiZpermission_deniedi not_foundiZresource_exhaustediZfailed_preconditioniZunauthenticatediZalready_existsZinvalid_argumentiXiZdeadline_exceedediZ unimplementedi unavailablerkZ unknown_error)rstrrr ZHTTP_STATUS_CODErlr+Z http_statusr,r,r-set_http_statuss:zSpan.set_http_statuscCs |jdkS)Nrrr`r,r,r- is_successrzSpan.is_successcCs|jdurdS|p|jptjj}z"|r t|trt|}||_nt|j }|j t |dd|_Wn t y?t |_Ynwt||dS)a4Sets the end timestamp of the span. Additionally it also creates a breadcrumb from the span, if the span represents a database or HTTP request. :param hub: The hub to use for this transaction. If not provided, the current hub will be used. :param end_timestamp: Optional timestamp that should be used as timestamp instead of the current time. :return: Always ``None``. The type is ``Optional[str]`` to match the return value of :py:meth:`sentry_sdk.tracing.Transaction.finish`. Ni) microseconds)rErHrcrdrerVrWr rrCrBrrXr"maybe_create_breadcrumbs_from_span)r+rH end_timestampelapsedr,r,r-rms$       z Span.finishc Cs|j|j|j|j|j|j|j|jd}|jr|j|j d<|j dur,|j }|r,||d<|j }|r5||d<|j }|r>||d<|S)z5Returns a JSON-compatible representation of the span.)r;r<r=r>r@rArBrErDN_metrics_summarytagsdata) r;r<r=r>r@rArBrErDrFrKto_jsonrG)r+r]metrics_summaryrrr,r,r-rs,   z Span.to_jsoncCsF|j|j|j|j|jd}|jr|j|d<|jr!|j|d<|S)N)r;r<r=r@rArDdynamic_sampling_context) r;r<r=r@rArDrYrrr\r,r,r-get_trace_contexts  zSpan.get_trace_context) NNNTNNNNNNNNNN)!r5r6r7r8r9rNr.rZr^rarjrqpropertyrYrSENTRYrxr{ classmethodrr|rrrrrrrlrrrmrrr,r,r,r-r:ZsV 0   !  )  $ ( !r:cseZdZdZdZdddeffdd ZddZfd d Zfd d Z e d dZ dfdd Z dddZ ddZfddZfddZddZddZZS)rOzhThe Transaction is the root element that holds all the spans for Sentry performance instrumentation.)namesourceparent_sampled sample_rate _measurements _contexts_profile_baggageNc sj|sd|vrtd|d}tt|jdi|||_||_d|_||_ i|_ i|_ d|_ ||_ dS)aConstructs a new Transaction. :param name: Identifier of the transaction. Will show up in the Sentry UI. :param parent_sampled: Whether the parent transaction was sampled. If True this transaction will be kept, if False it will be discarded. :param baggage: The W3C baggage header value. (see https://www.w3.org/TR/baggage/) :param source: A string describing the source of the transaction name. This will be used to determine the transaction's type. See https://develop.sentry.dev/sdk/event-payloads/transaction/#transaction-annotations for more information. Default "custom". rLz^Deprecated: use Transaction(name=...) to create transactions instead of Span(transaction=...).Nr,)rrypopsuperrOr.rrrrrrrr)r+rrrrrQr_r,r-r.:s   zTransaction.__init__c Cs*d|jj|j|j|j|j|j|j|jfS)NzW<%s(name=%r, op=%r, trace_id=%r, span_id=%r, parent_span_id=%r, sampled=%r, source=%r)>) r_r5rr@r;r<r=r?rr`r,r,r-raeszTransaction.__repr__cs&tt||jdur|j|Sr/)rrOrjrr`rr,r-rjus  zTransaction.__enter__cs2|jdur |j|||tt||||dSr/)rrqrrO)r+rnrorprr,r-rq~s zTransaction.__exit__cCs|S)znThe root element of the span tree. In the case of a transaction it is the transaction itself. r,r`r,r,r-rYs z"Transaction.containing_transactionc s|jdurdS|p|jptjj}|j}|durdS|jdurCtd|j rAt |j rA|j r7|j j dkr7d}nd}|j j|dddS|jsNtdd |_tt||||jsf|jdurdtd dSd d |jjD}d|_i}||j|d |id|jd|ji||j|j|j|d}|jdur|jr|j|d<d|_|j|d<|jdur|j }|r||d<|!|S)aFinishes the transaction and sends it to Sentry. All finished spans in the transaction will also be sent to Sentry. :param hub: The hub to use for this transaction. If not provided, the current hub will be used. :param end_timestamp: Optional timestamp that should be used as timestamp instead of the current time. :return: The event ID if the transaction was sent to Sentry, otherwise None. Nz.Discarding transaction because sampled = FalserZ backpressurerrL)Z data_categoryzCTransaction has no name, falling back to ``.zz1Discarding transaction without sampling decision.cSsg|] }|jdur|qSr/)rEr).0r3r,r,r- s  z&Transaction.finish..tracer)typerLZtransaction_infocontextsrrErBr(Zprofile measurementsr)"rErHrcrdrersr1rdebug transporthas_tracing_enabledrtmonitordownsample_factorZrecord_lost_eventrryrrOrmr?r(rrrrrFrBrvalidrrKrZ capture_event) r+rHrrsreasonZfinished_spansreventrrr,r-rmsb            zTransaction.finishcCs||d|j|<dS)N)rounit)rr+rrorr,r,r-set_measurementszTransaction.set_measurementcCs||j|<dS)aSets a context. Transactions can have multiple contexts and they should follow the format described in the "Contexts Interface" documentation. :param key: The name of the context. :param value: The information about the context. N)rrr,r,r- set_contexts zTransaction.set_contextcs$tt|||dd|idS)zySets the status of the Transaction according to the given HTTP status. :param http_status: The HTTP status code.response status_codeN)rrOrrrrr,r-rszTransaction.set_http_statuscs0tt|}|j|d<|j|d<|j|d<|S)z r)r@rFNZtraces_samplerrZtraces_sample_rateZTracing)rzN[Tracing] Discarding {transaction_description} because of invalid sample rate.)transaction_descriptionz?[Tracing] Discarding {transaction_description} because {reason}z"traces_sampler returned 0 or Falseztraces_sample_rate is set to 0)rrz,[Tracing] Starting {transaction_description}z}[Tracing] Discarding {transaction_description} because it's not included in the random sample (sampling rate = {sample_rate}))rr)rHrcrdrersrtformatr@rrr?rWrcallablerrrryrrrrandom)r+sampling_contextrHrsrtrrr,r,r-_set_initial_sampling_decision#sj         z*Transaction._set_initial_sampling_decisionrr)r5r6r7r8r9TRANSACTION_SOURCE_CUSTOMr.rarjrqrrYrmrrrrrr __classcell__r,r,rr-rO*s(+    c   rOc@seZdZddZeddZejfddZddZ d d Z d d Z d dZ ddZ ddZddZddZddZddZddZddZd+d d!Zd,d#d$Zd%d&Zd'd(Zd)d*ZdS)-rucCs|jjSr/)r_r5r`r,r,r-razNoOpSpan.__repr__cCdSr/r,r`r,r,r-rYzNoOpSpan.containing_transactioncKstSr/)ru)r+rrrQr,r,r-rxszNoOpSpan.start_childcKs|jdi|SNr,)rxrzr,r,r-r{szNoOpSpan.new_spancCdS)Nrr,r`r,r,r-rzNoOpSpan.to_traceparentcCrr/r,r`r,r,r-rrzNoOpSpan.to_baggagecCrr/r,r`r,r,r-rrzNoOpSpan.get_baggagecCstdSr)iterr`r,r,r-rrzNoOpSpan.iter_headerscCrr/r,rr,r,r-rrzNoOpSpan.set_tagcCrr/r,rr,r,r-rrzNoOpSpan.set_datacCrr/r,rr,r,r-rlrzNoOpSpan.set_statuscCrr/r,rr,r,r-rrzNoOpSpan.set_http_statuscCr)NTr,r`r,r,r-rrzNoOpSpan.is_successcCiSr/r,r`r,r,r-rrzNoOpSpan.to_jsoncCrr/r,r`r,r,r-rrzNoOpSpan.get_trace_contextNcCrr/r,)r+rHrr,r,r-rmrzNoOpSpan.finishrcCrr/r,rr,r,r-rrzNoOpSpan.set_measurementcCrr/r,rr,r,r-rrzNoOpSpan.set_contextcCrr/r,r*r,r,r-rZrzNoOpSpan.init_span_recordercCrr/r,)r+rr,r,r-rrz'NoOpSpan._set_initial_sampling_decisionrr)r5r6r7rarrYrrrxr{rrrrrrrlrrrrrmrrrZrr,r,r,r-rus,    rucCrr/r,funcr,r,r-rrrcCrr/r,rr,r,r-rrcCs.tr ddlm}nddlm}|r||S|S)av Decorator to start a child span under the existing current transaction. If there is no current transaction, then nothing will be traced. .. code-block:: :caption: Usage import sentry_sdk @sentry_sdk.trace def my_function(): ... @sentry_sdk.trace async def my_async_function(): ... r)start_child_span_decorator)r Zsentry_sdk.tracing_utils_py2rZsentry_sdk.tracing_utils_py3)rrr,r,r-rs  )rr}rrr)r[r/)>rSrrrrcZsentry_sdk.constsrZsentry_sdk.utilsrrrZsentry_sdk._compatrr r r Zsentry_sdk._typesr typingcollections.abcr rrrrrrrrrrrrrZsentry_sdk.profilerrrrrrrZTRANSACTION_SOURCE_URLZTRANSACTION_SOURCE_ROUTEZTRANSACTION_SOURCE_VIEWZTRANSACTION_SOURCE_COMPONENTZTRANSACTION_SOURCE_TASKZLOW_QUALITY_TRANSACTION_SOURCESZSOURCE_FOR_STYLErMr%r:rOrurZsentry_sdk.tracing_utilsrr}rrrZsentry_sdk.metricsr[r,r,r,r-s~              SfS   "