File: //opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/__pycache__/tracing.cpython-37.pyc
B
T a�d � @ sz d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dlm Z d dl
Z
d dl m Z m
Z
mZ d dlmZ d dlmZ er�d dlmZ n d dlmZ er�d dlZd dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ d d
lm Z d dlm Z e � d�Z G dd� de�Z G dd� de!�Z"G dd� de!�Z#G dd� de#�Z$dd� Z%dd� Z&d d � Z'ej(d d � �Z)d d!� Z*dS )"� N)�datetime� timedelta)�Real)�capture_internal_exceptions�logger� to_string)�PY2)�MYPY)�Mapping)� Generator)�Optional)�Any)�Dict)�List)�Tuple)�SamplingContextz5^[ ]*([0-9a-f]{32})?-?([0-9a-f]{16})?-?([01])?[ ]*$c @ s. e Zd Zd dd�Zdd� Zdd� Zdd � Zd
S ) �EnvironHeaders�HTTP_c C s || _ || _d S )N)�environ�prefix)�selfr r � r �C/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/tracing.py�__init__0 s zEnvironHeaders.__init__c C s | j | j|�dd��� S )N�-�_)r r �replace�upper)r �keyr r r � __getitem__9 s zEnvironHeaders.__getitem__c C s t dd� t| �D ��S )Nc s s | ]
}d V qdS )� Nr )�.0r r r r � <genexpr>? s z)EnvironHeaders.__len__.<locals>.<genexpr>)�sum�iter)r r r r �__len__= s zEnvironHeaders.__len__c c sR xL| j D ]B}t|t�sq|�dd��� }|�| j�s6q|t| j�d � V qW d S )Nr r )r �
isinstance�strr r �
startswithr �len)r �kr r r �__iter__A s
zEnvironHeaders.__iter__N)r )�__name__�
__module__� __qualname__r r r% r+ r r r r r / s
r c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�
_SpanRecorderz5Limits the number of spans recorded in a transaction.)�maxlen�spansc C s |d | _ g | _d S )Nr )r0 r1 )r r0 r r r r S s
z_SpanRecorder.__init__c C s( t | j�| jkrd |_n | j�|� d S )N)r) r1 r0 �_span_recorder�append)r �spanr r r �add] s z_SpanRecorder.addN)r, r- r. �__doc__� __slots__r r5 r r r r r/ N s
r/ c
@ s� e Zd ZdZdd� Zd.dd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z e dd� �Z
e dd� �Zdd� Ze dd� �Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd/d(d)�Zd*d+� Zd,d-� ZdS )0�Span)�trace_id�span_id�parent_span_id�same_process_as_parent�sampled�op� description�start_timestamp�_start_timestamp_monotonic�status� timestamp�_tags�_datar2 �hub�_context_manager_state�_containing_transactionc K s d|krt �t�S t �| �S )z_
Backwards-compatible implementation of Span and Transaction
creation.
� transaction)�object�__new__� Transaction)�cls�kwargsr r r rK | s
z Span.__new__NTc C s� |p t �� j| _|p$t �� jdd � | _|| _|| _|| _|| _|| _ | | _
|| _ i | _ i | _
t�� | _yt�� | _W n tk
r� Y nX d | _d | _d | _d S )N� )�uuidZuuid4�hexr9 r: r; r< r= r> r? rB rF rD rE r �utcnowr@ �time� perf_counterrA �AttributeErrorrC r2 rH ) r r9 r: r; r< r= r> r? rF rB rI r r r r � s&
z
Span.__init__c C s$ | j d krt|�| _ | j �| � d S )N)r2 r/ r5 )r r0 r r r �init_span_recorder� s
zSpan.init_span_recorderc C s& d| j j| j| j| j| j| j| jf S )NzS<%s(op=%r, description:%r, trace_id=%r, span_id=%r, parent_span_id=%r, sampled=%r)>)� __class__r, r> r? r9 r: r; r= )r r r r �__repr__� s z
Span.__repr__c C s8 | j p tjj}|jd \}}|j}| |_|||f| _| S )N���)rF �
sentry_sdk�Hub�currentZ_stackr4 rG )r rF r �scope�old_spanr r r � __enter__� s zSpan.__enter__c C s6 |d k r| � d� | j\}}}| `| �|� ||_d S )N�internal_error)�
set_statusrG �finishr4 )r �ty�value�tbrF r] r^ r r r �__exit__� s
z
Span.__exit__c K s` |� d| j� tf | jd| jd�|��}t| t�r:| |_n| 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.
r= N)r9 r: r; )
�
setdefaultr= r8 r9 r: r& rL rH r2 r5 )r rN �rvZrecorderr r r � start_child� s
zSpan.start_childc K s t �d� | jf |�S )z$Deprecated: use start_child instead.z:Deprecated: use Span.start_child instead of Span.new_span.)r �warningri )r rN r r r �new_span� s
z
Span.new_spanc K s$ | t krt�d� tjt|�f|�S )aJ
Create a Transaction with the given params, then add in data pulled from
the 'sentry-trace' header in the environ (if any) before returning the
Transaction.
If the 'sentry-trace' header is malformed or missing, just create and
return a Transaction instance with the given params.
zXDeprecated: use Transaction.continue_from_environ instead of Span.continue_from_environ.)r8 r rj rL �continue_from_headersr )rM r rN r r r �continue_from_environ� s zSpan.continue_from_environc K sB | t krt�d� tj|�d�f|�}|dkr8tf |�}d|_|S )a3
Create a Transaction with the given params, then add in data pulled from
the 'sentry-trace' header (if any) before returning the Transaction.
If the 'sentry-trace' header is malformed or missing, just create and
return a Transaction instance with the given params.
zXDeprecated: use Transaction.continue_from_headers instead of Span.continue_from_headers.z sentry-traceNF)r8 r rj rL �from_traceparent�getr< )rM �headersrN rI r r r rl s
zSpan.continue_from_headersc c s d| � � fV d S )Nz sentry-trace)�to_traceparent)r r r r � iter_headers* s zSpan.iter_headersc K s� | t krt�d� |sdS |�d�r:|�d�r:|dd� }t�t|��}|dkrTdS |�� \}}}|dk rzd� t
|d��}|dk r�d � t
|d��}|r�|d
k}nd}t f |||d �|��S ) a-
Create a Transaction with the given params, then add in data pulled from
the given 'sentry-trace' header value before returning the Transaction.
If the header value is malformed or missing, just create and return a
Transaction instance with the given params.
zNDeprecated: use Transaction.from_traceparent instead of Span.from_traceparent.Nz00-z-00� ���z{:032x}rO z{:016x}�0)r9 r; �parent_sampled) r8 r rj r( �endswith� _traceparent_header_format_re�matchr' �groups�format�intrL )rM Z traceparentrN ry r9 r; Z sampled_strrv r r r rn . s.
zSpan.from_traceparentc C s2 d}| j dkrd}| j dkr d}d| j| j|f S )N� T�1Fru z%s-%s-%s)r= r9 r: )r r= r r r rq _ s
zSpan.to_traceparentc C s || j |< d S )N)rD )r r rd r r r �set_tagh s z Span.set_tagc C s || j |< d S )N)rE )r r rd r r r �set_datal s z
Span.set_datac C s
|| _ d S )N)rB )r rd r r r ra p s zSpan.set_statusc C s6 | � dt|�� |dk r&| �d� �n d| kr:dk r�n n�|dkrR| �d� nn|dkrf| �d� nZ|d krz| �d
� nF|d kr�| �d � n2|d
kr�| �d� n |dkr�| �d� n
| �d� npd| kr�dk �r(n nL|dkr�| �d� n6|dk�r| �d� n |dk�r | �d� n
| �d� n
| �d� d S )Nzhttp.status_codei� �oki� i� Zpermission_deniedi� Z not_foundi� Zresource_exhaustedi� Zfailed_preconditioni� Zunauthenticatedi� Zalready_existsZinvalid_argumentiX i� Zdeadline_exceededi� Z
unimplementedi� Z unavailabler` Z
unknown_error)r r' ra )r Z http_statusr r r �set_http_statust s4
zSpan.set_http_statusc C s
| j dkS )Nr� )rB )r r r r �
is_success� s zSpan.is_successc C sr | j d k rd S |p | jp tjj}y$t�� | j }| jt |d� | _ W n t
k
rb t � � | _ Y nX t
|| � d S )N)�seconds)rC rF rZ r[ r\ rS rT rA r@ r rU r rR �#_maybe_create_breadcrumbs_from_span)r rF Zduration_secondsr r r rb � s
z Span.finishc C s` | j | j| j| j| j| j| j| jd�}| jr8| j| j d< | j }|rJ||d< | j
}|r\||d< |S )N)r9 r: r; r< r> r? r@ rC rB �tags�data) r9 r: r; r< r> r? r@ rC rB rD rE )r rh r� r� r r r �to_json� s"
z Span.to_jsonc C s. | j | j| j| j| jd�}| jr*| j|d< |S )N)r9 r: r; r> r? rB )r9 r: r; r> r? rB )r rh r r r �get_trace_context� s
zSpan.get_trace_context)
NNNTNNNNNN)N)r, r- r. r7 rK r rV rX r_ rf ri rk � classmethodrm rl rr rn rq r r� ra r� r� rb r� r� r r r r r8 e s>
1 !
r8 c sD e Zd ZdZddd�Zdd� Zddd �Z� fd
d �Zd d
� Z� Z S )rL )�namerv r} Nc K s> |s d|kr t �d� |�d�}tj| f|� || _|| _d S )NrI z^Deprecated: use Transaction(name=...) to create transactions instead of Span(transaction=...).)r rj �popr8 r r� rv )r r� rv rN r r r r � s
zTransaction.__init__c C s& d| j j| j| j| j| j| j| jf S )NzL<%s(name=%r, op=%r, trace_id=%r, span_id=%r, parent_span_id=%r, sampled=%r)>)rW r, r� r> r9 r: r; r= )r r r r rX � s zTransaction.__repr__c
s� � j d k rd S � jd kr&t�d� d S |p6� jp6tjj}|j}|d krJd S � j s`t�
d� d� _ t � � |� � j
s�� j
d kr�t�
d� d S � fdd�� jjD �}|�d� j d� �� i� j� j � j|d ��S )
Nz.Discarding transaction because sampled = FalsezCTransaction has no name, falling back to `<unlabeled transaction>`.z<unlabeled transaction>z1Discarding transaction without sampling decision.c s& g | ] }|� k r|j d k r|�� �qS )N)rC r� )r! r4 )r r r �
<listcomp> s z&Transaction.finish.<locals>.<listcomp>rI �trace)�typerI Zcontextsr� rC r@ r1 )rC r2 r �debugrF rZ r[ r\ �clientr� rj r8 rb r= r1 Z
capture_eventr� rD r@ )r rF r� Zfinished_spansr )r r rb � s8
zTransaction.finishc s& t t| ��� }| j|d< | j|d< |S )Nr� r= )�superrL r� r� r= )r rh )rW r r r� + s
zTransaction.to_jsonc C sB | j p tjj}|j}|r |jp i }dj| jr:d| j d nd| jd�}|rRt |�s\d| _
dS | j
dk rjdS t |� d��r�|d |�n|d dk r�|d n|d
}t
|�s�t�d j|d �� d| _
dS |s�t�d
j|t |� d��r�dndd�� d| _
dS t�� t|�k | _
| j
�r&t�dj|d �� nt�dj|t|�d�� dS )aP
Sets the transaction's sampling decision, according to the following
precedence rules:
1. If a sampling decision is passed to `start_transaction`
(`start_transaction(name: "my transaction", sampled: True)`), that
decision will be used, regardlesss of anything else
2. If `traces_sampler` is defined, its decision will be used. It can
choose to keep or ignore any parent sampling decision, or use the
sampling context data to make its own decision or to choose a sample
rate for the transaction.
3. If `traces_sampler` is not defined, but there's a parent sampling
decision, the parent sampling decision will be used.
4. If `traces_sampler` is not defined and there's no parent sampling
decision, `traces_sample_rate` will be used.
z{op}transaction <{name}>�<z> r} )r> r� FN�traces_samplerrv �traces_sample_ratezN[Tracing] Discarding {transaction_description} because of invalid sample rate.)�transaction_descriptionz?[Tracing] Discarding {transaction_description} because {reason}z"traces_sampler returned 0 or Falsez traces_sample_rate is set to 0)r� �reasonz,[Tracing] Starting {transaction_description}z}[Tracing] Discarding {transaction_description} because it's not included in the random sample (sampling rate = {sample_rate}))r� � sample_rate)rF rZ r[ r\ r� �optionsr{ r> r� �has_tracing_enabledr= �callablero �_is_valid_sample_rater rj r� �random�float)r Zsampling_contextrF r� r� r� r� r r r � _set_initial_sampling_decision4 sH
z*Transaction._set_initial_sampling_decision)r} N)N)
r, r- r. r7 r rX rb r� r� �
__classcell__r r )rW r rL � s
3 rL c C s t | �d�dk p | �d�dk �S )zf
Returns True if either traces_sample_rate or traces_sampler is
defined, False otherwise.
r� Nr� )�boolro )r� r r r r� � s r� c C sb t | t�rt�| �r0t�dj| t| �d�� dS t| �} | dk sH| dkr^t�dj| d�� dS dS ) z�
Checks the given sample rate to make sure it is valid type and value (a
boolean or a number between 0 and 1, inclusive).
z}[Tracing] Given sample rate is invalid. Sample rate must be a boolean or a number between 0 and 1. Got {rate} of type {type}.)�rater� Fr r zX[Tracing] Given sample rate is invalid. Sample rate must be between 0 and 1. Got {rate}.)r� T) r& r �math�isnanr rj r{ r� r� )r� r r r r� � s
r� c C sZ d }y0t | d�r2| �|�}t|t�r2|�| jj�}W n tk
rL d }Y nX |pXt|�S )N�mogrify) �hasattrr� r& �bytes�decodeZ
connection�encoding� Exceptionr )�cursorZsqlZreal_sqlr r r � _format_sql� s
r� c
c s� | j r:| j jd �dd�r:|r(|d gkr,d }|dkrBd}nd }d }t||�}i }|d k r`||d< |d k rp||d< |r|d|d < t� � | j|d
|d � W d Q R X | jd |d
��.}x |�� D ]\}} |�|| � q�W |V W d Q R X d S )NZ _experimentsZrecord_sql_paramsFZpyformatr{ z db.paramsz
db.paramstyleTzdb.executemany�query)�message�categoryr� Zdb)r> r? ) r� r� ro r� r �add_breadcrumbZ
start_span�itemsr� )
rF r� r� Z params_listZ
paramstyleZ executemanyr� r4 r* �vr r r �record_sql_queries� s,
r� c C sd |j dkr"| j|jdd|jd� n>|j dkr@| jdd|jd� n |j dkr`| jdd|j|jd� d S )NZredis)r� r� r� r� �httpZhttplib)r� r� r� �
subprocess)r� r� r� r� )r> r� r? rD rE )rF r4 r r r r� s
r� )+�rerP �
contextlibr� r� rS r r �numbersr rZ Zsentry_sdk.utilsr r r Zsentry_sdk._compatr Zsentry_sdk._typesr � collectionsr
�collections.abc�typingr r r
r r r r �compilerx r rJ r/ r8 rL r� r� r� �contextmanagerr� r� r r r r �<module> sJ r E
+