관리-도구
편집 파일: 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 +