관리-도구
편집 파일: tracing_utils.cpython-38.pyc
U �Mg�U � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m Z d dl m Z m Z d dlZd dlZd dlmZmZ d dlmZmZmZmZmZmZmZmZm Z m Z m Z d dl m Z e �r d d l m!Z! d d l m"Z" d d l m#Z# d d l m$Z$ d d l m%Z% d dl&m'Z' e�(d�Z)dZ*G dd� de�Z+dd� Z,e j-d3dd��Z.dd� Z/dd � Z0d d � Z1d d � Z2d!d"� Z3d#d$� Z4G d%d&� d&�Z5G d'd(� d(�Z6d)d*� Z7d+d,� Z8d-d.� Z9d4d/d0�Z:d d1l;m<Z<m=Z=m>Z> e �r�d d2l;m?Z? dS )5� N)�Mapping)� timedelta)�wraps)�quote�unquote)�OP�SPANDATA) �capture_internal_exceptions�filename_for_module�Dsn�logger�match_regex_list�qualname_from_function� to_string� is_sentry_url�_is_external_source�_is_in_project_root�_module_in_list)� TYPE_CHECKING)�Any)�Dict)� Generator)�Optional)�Union)� FrameTypez5^[ ]*([0-9a-f]{32})?-?([0-9a-f]{16})?-?([01])?[ ]*$z(([a-zA-Z0-9+/]{4})*([a-zA-Z0-9+/]{2,3})?c @ s. e Zd Zd dd�Zdd� Zdd� Zdd � Zd S ) �EnvironHeaders�HTTP_c C s || _ || _d S �N)�environ�prefix)�selfr r � r! �F/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/tracing_utils.py�__init__= s zEnvironHeaders.__init__c C s | j | j|�dd��� S �N�-�_)r r �replace�upper)r �keyr! r! r" � __getitem__F s zEnvironHeaders.__getitem__c C s t dd� t| �D ��S )Nc s s | ] }d V qdS )� Nr! )�.0r& r! r! r" � <genexpr>L s z)EnvironHeaders.__len__.<locals>.<genexpr>)�sum�iter�r r! r! r" �__len__J s zEnvironHeaders.__len__c c sN | j D ]B}t|t�sq|�dd��� }|�| j�s4q|t| j�d � V qd S r$ )r � isinstance�strr' r( � startswithr �len)r �kr! r! r" �__iter__N s zEnvironHeaders.__iter__N)r )�__name__� __module__� __qualname__r# r* r1 r7 r! r! r! r" r < s � r c C s: | dkr dS t | �d�dk o6| �d�dk p6| �d�dk �S )z} Returns True if either traces_sample_rate or traces_sampler is defined and enable_tracing is set and not false. NFZenable_tracing�traces_sample_rateZtraces_sampler)�bool�get)�optionsr! r! r" �has_tracing_enabled[ s �r? F�manualc c s� t �� jd �dd�r6|r$|d gkr(d }|dkr>d}nd }d }t| |�}i }|d k r\||d< |d k rl||d< |rxd|d < |r�| d k r�| |d < t� � t j|d |d � W 5 Q R X t jtj ||d ��*}|� � D ]\} } |� | | � q�|V W 5 Q R X d S )NZ _experimentsZrecord_sql_paramsFZpyformat�formatz db.paramsz db.paramstyleTzdb.executemanyz db.cursor�query)�message�category�data)�op�name�origin) � sentry_sdk� get_clientr> r= � _format_sqlr �add_breadcrumbZ start_spanr ZDB�items�set_data) �cursorrB Z params_listZ paramstyleZ executemanyZrecord_cursor_reprZ span_originrE �spanr6 �vr! r! r" �record_sql_queriesm s8 �rR c C sh |j tjkr$| j|jdd|jd� n@|j tjkrD| jdd|jd� n |j dkrd| jdd|j|jd� d S )NZredis)rC �typerD rE �http�httplib)rS rD rE � subprocess)rS rD rC rE )rF r ZDB_REDISrL � descriptionZ_tagsZ HTTP_CLIENT�_data)�scoperP r! r! r" �"maybe_create_breadcrumbs_from_span� s � �rZ c C s& z | j jW S tk r Y d S X d S r )�f_code� co_filename� Exception)�framer! r! r" �_get_frame_module_abs_path� s r_ c C s6 t ||�}t|�pt ||�}| o4|p4t||�o4| S r )r r r )�is_sentry_sdk_frame� namespace�in_app_include�in_app_exclude�abs_path� project_root�should_be_includedZshould_be_excludedr! r! r" �_should_be_included� s ��rg c C sT t �� }|�� sdS | jdks(| jdkr,dS |j�dd�}|sBdS | j| j }|j�dd�}|tdd� |k}|svdS |jd }|j�d �}|j�d �}t� � } | dk �rt | �} z| j �d �} W n t k r� d} Y nX | dk o�| � d �} t| | ||| |d �} | �r �q| j} q�d} | dk �rPz | j}W n t k �rF d}Y nX |dk �rb| �tj| j� z| j �d �} W n t k �r� d} Y nX | dk �r�| �tj| � t | �}|dk �r | dk �r�t| |�}n0|dk �r�|� |��r�|�|d��tj�}n|}| �tj|� z | jj }W n t k �r2 d}Y nX |dk �rP| �tj | jj � dS )zB Adds OTel compatible source code information to the span NZenable_db_query_sourceTZ db_query_source_threshold_msr r+ )Z millisecondsre rb rc r8 z sentry_sdk.)r` ra rb rc rd re � ) rI rJ � is_active� timestampZstart_timestampr> r= r �sys� _getframer_ � f_globalsr] r4 rg �f_back�f_linenorN r Z CODE_LINENOZCODE_NAMESPACEr r' �lstrip�os�sepZ CODE_FILEPATHr[ �co_nameZ CODE_FUNCTION)rP �clientZshould_add_query_source�duration� thresholdZ slow_queryre rb rc r^ rd ra r` rf �lineno�filepathZ in_app_pathZ code_functionr! r! r" �add_query_source� s� �� ry c C s� | sdS | � d�r(| �d�r(| dd� } t�| �}|s:dS |�� \}}}d}|r`d�t|d��}|rtd�t|d��}|r�|d k}|||d �S ) zL Given a `sentry-trace` header string, return a dictionary of data. Nz00-z-00� ���z{:032x}� z{:016x}�0)�trace_id�parent_span_id�parent_sampled)r4 �endswith�SENTRY_TRACE_REGEX�match�groupsrA �int)�headerr� r~ r Z sampled_strr� r! r! r" �extract_sentrytrace_data& s& �r� c C sZ d }z0t | d�r2| �|�}t|t�r2|�| jj�}W n tk rL d }Y nX |pXt|�S )N�mogrify) �hasattrr� r2 �bytes�decode� connection�encodingr] r )rO �sqlZreal_sqlr! r! r" rK F s rK c @ sn e Zd ZdZdZddd�Zedd� �Zedd � �Z e j d d � �Z ed d � �Z e j d d � �Z dd� Z dd� Z dS )�PropagationContextzJ The PropagationContext represents the data of a trace in Sentry. �� _trace_id�_span_idr r� �dynamic_sampling_contextNc C s" || _ || _|| _|| _|| _d S r r� )r r~ �span_idr r� r� r! r! r" r# g s zPropagationContext.__init__c C sj d }t |�}|�t�}|r0t� }t�|��� |_|�t�}|rft|�}|d k rf|d kr\t� }|� |� |S r ) �normalize_incoming_datar= �BAGGAGE_HEADER_NAMEr� �Baggage�from_incoming_headerr� �SENTRY_TRACE_HEADER_NAMEr� �update)�cls� incoming_data�propagation_contextZnormalized_dataZbaggage_headerZsentry_trace_headerZsentrytrace_datar! r! r" �from_incoming_data� s � z%PropagationContext.from_incoming_datac C s | j st�� j| _ | j S )z!The trace id of the Sentry trace.)r� �uuid�uuid4�hexr0 r! r! r" r~ � s zPropagationContext.trace_idc C s || _ d S r )r� �r �valuer! r! r" r~ � s c C s | j st�� jdd� | _ | j S )z+The span id of the currently executed span.r| N)r� r� r� r� r0 r! r! r" r� � s zPropagationContext.span_idc C s || _ d S r )r� r� r! r! r" r� � s c C s< |� � D ].\}}zt| ||� W q tk r4 Y qX qdS )zU Updates the PropagationContext with data from the given dictionary. N)rM �setattr�AttributeError)r � other_dictr) r� r! r! r" r� � s zPropagationContext.updatec C s d� | j| j| j| j| j�S )Nzm<PropagationContext _trace_id={} _span_id={} parent_span_id={} parent_sampled={} dynamic_sampling_context={}>)rA r� r� r r� r� r0 r! r! r" �__repr__� s �zPropagationContext.__repr__)NNNNN)r8 r9 r: �__doc__� __slots__r# � classmethodr� �propertyr~ �setterr� r� r� r! r! r! r" r� Z s( � r� c @ sj e Zd ZdZdZdZe�d�Zddd�Z e d d � �Z e d d � �Z e d d� �Z dd� Zdd� Zddd�ZdS )r� zR The W3C Baggage header information (see https://www.w3.org/TR/baggage/). �� sentry_items�third_party_items�mutablezsentry-z^sentry-rh Tc C s || _ || _|| _d S r r� )r r� r� r� r! r! r" r# � s zBaggage.__init__c C s� i }d}d}|r�|� d�D ]z}d|kr(qt� �` |�� }|� d�\}}tj�|�rvt|� d�d �}t|�||< d}n||r�dnd| 7 }W 5 Q R X qt|||�S )zF freeze if incoming header already has sentry baggage rh T�,�=r% r+ F)�splitr �stripr� �SENTRY_PREFIX_REGEXr� r ) r� r� r� r� r� �itemr) �valZ baggage_keyr! r! r" r� � s z Baggage.from_incoming_headerc C s� i }d}d}t �� }|�� r&|jd kr.t|�S |j}|j}|d k rL|j|d< |�d�rb|d |d< |�d�rx|d |d< |�d�r�t|d �j |d< |�d�r�t |d �|d < t|||�S ) Nrh Fr~ � environment�release�dsn� public_keyr; � sample_rate) rI rJ ri Z_propagation_contextr� r> r~ r= r r� r3 )r� rY r� r� r� rt r> r� r! r! r" � from_options� s& zBaggage.from_optionsc C s� t �� }i }|�� s t|�S |jp$i }|j|d< |�d�rF|d |d< |�d�r\|d |d< |�d�rxt|d �j|d< |j r�|j t kr�|j |d< |j dk r�t |j �|d< |jdk r�|jr�d nd |d < |jr�|jjr�|�|jj� t|d d �S )z� Populate fresh baggage entry with sentry_items and make it immutable if this is the head SDK which originates traces. r~ r� r� r� r� � transactionNr� �true�false�sampledF�r� )rI rJ ri r� r> r~ r= r r� rG �source� LOW_QUALITY_TRANSACTION_SOURCESr� r3 r� Z_baggager� r� )r� r� rt r� r> r! r! r" �populate_from_transaction s0 �� z!Baggage.populate_from_transactionc C s d| _ d S )NFr� r0 r! r! r" �freezeI s zBaggage.freezec C s$ i }| j �� D ]\}}|||< q|S r )r� rM )r r� r) r� r! r! r" r� M s z Baggage.dynamic_sampling_contextFc C sl g }| j �� D ]B\}}t� �. tjt|� d tt|�� }|�|� W 5 Q R X q|rb|�| j� d� |�S )Nr� r� ) r� rM r r� � SENTRY_PREFIXr r3 �appendr� �join)r Zinclude_third_partyrM r) r� r� r! r! r" � serializeV s zBaggage.serializeN)rh T)F)r8 r9 r: r� r� r� �re�compiler� r# r� r� r� r� r� r� r� r! r! r! r" r� � s � - r� c C s& | j d }t| |�rdS t||dd�S )zy Returns True if url matches trace_propagation_targets configured in the given client. Otherwise, returns False. �trace_propagation_targetsFT)Zsubstring_matching)r> r r )rt �urlr� r! r! r" �should_propagate_tracee s r� c C sH i }| � � D ]6\}}|�d�r*|dd� }|�dd��� }|||< q |S )z� Normalizes incoming data so the keys are all lowercase with dashes instead of underscores and stripped from known prefixes. r � Nr& r% )rM r4 r' �lower)r� rE r) r� r! r! r" r� r s r� c s� t �� �rFt� �� fdd��}zt �� �|_W q� tk rB Y q�X n:t� �� fdd��}zt �� �|_W n tk r~ Y nX |S )zc Decorator to add child spans for functions. See also ``sentry_sdk.tracing.trace()``. c � sj t � }|d kr.t�dt� �� � | |�I d H S |jtjt� �d�� � | |�I d H W 5 Q R � S Q R X d S �NzbCannot create a child span for %s. Please start a Sentry transaction before calling this function.)rF rG ��get_current_spanr �debugr Z start_childr ZFUNCTION��args�kwargsrP ��funcr! r" �func_with_tracing� s ��z5start_child_span_decorator.<locals>.func_with_tracingc s^ t � }|d kr(t�dt� �� � | |�S |jtjt� �d�� � | |�W 5 Q R � S Q R X d S r� r� r� r� r! r" r� � s � �)�inspect�iscoroutinefunctionr � signature� __signature__r] )r� r� r! r� r" �start_child_span_decorator� s r� c C s | p t �� } | j}|S )zU Returns the currently active span if there is one running, otherwise `None` )rI Zget_current_scoperP )rY Z current_spanr! r! r" r� � s r� )r� r� r� )�Span)Fr@ )N)@� contextlibr� rq r� rk �collections.abcr �datetimer � functoolsr � urllib.parser r r� rI Zsentry_sdk.constsr r Zsentry_sdk.utilsr r r r r r r r r r r �typingr r r r r r �typesr r� r� Zbase64_strippedr r? �contextmanagerrR rZ r_ rg ry r� rK r� r� r� r� r� r� Zsentry_sdk.tracingr� r� r� r� r! r! r! r" �<module> s^ 4 � � � /[ q B