관리-도구
편집 파일: hub.cpython-37.pyc
B T a|V � @ s d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z m Z d dlmZ d d lmZmZmZmZ d d lmZ e�r`d d lmZ d d lmZ d d lmZ d dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d dlm Z d dlm!Z! d dlm"Z" d dlm#Z# d dl$m%Z% d dlm&Z&m'Z'm(Z(m)Z)m*Z* d dl+m,Z, e!d�Z-ndd � Z"ed �Z.d d � Z/d d!� Z0G d"d#� d#e1�Z2d$d%� Z3d d lmZ e�r�G d&d'� d'e,e#e �Z4n d(d)� � Z4G d*d+� d+e5�Z6G d,d-� d-e1�Z7G d.d/� d/ee6��Z8e8� Z9e.�:e9� dS )0� N)�datetime)�contextmanager)�with_metaclass)�Scope)�Client)�Span� Transaction)�Session)�exc_info_from_error�event_from_exception�logger� ContextVar)�MYPY)�Union)�Any)�Optional)�Tuple)�Dict)�List)�Callable)� Generator)�Type)�TypeVar)�overload)�ContextManager)� Integration)�Event�Hint� Breadcrumb�BreadcrumbHint�ExcInfo)�ClientConstructor�Tc C s | S )N� )�xr# r# �?/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/hub.pyr 1 s r Zsentry_current_hubc C sd |r|rt d��|d k r@t�| �}t|�r4||� q`|�|� n |r\t�| �}|jf |� n| }|S )Nz cannot provide scope and kwargs)� TypeError�copy�callableZupdate_from_scopeZupdate_from_kwargs)�baseZ scope_changeZ scope_kwargsZ final_scoper# r# r% � _update_scope9 s r* c C s t jj} | sdS | jd S )NFZsend_default_pii)�Hub�current�client�options)r- r# r# r% �_should_send_default_piiK s r/ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _InitGuardc C s || _ d S )N)�_client)�selfr- r# r# r% �__init__T s z_InitGuard.__init__c C s | S )Nr# )r2 r# r# r% � __enter__X s z_InitGuard.__enter__c C s | j }|d k r|�� d S )N)r1 �close)r2 �exc_type� exc_value�tb�cr# r# r% �__exit__\ s z_InitGuard.__exit__N)�__name__� __module__� __qualname__r3 r4 r: r# r# r# r% r0 S s r0 c O s" t | |�}tj�|� t|�}|S )zsInitializes the SDK and optionally integrations. This takes the same arguments as the client constructor. )r r+ r, � bind_clientr0 )�args�kwargsr- �rvr# r# r% �_initc s rB c @ s e Zd ZdS )�initN)r; r< r= r# r# r# r% rC x s rC c C s t S )N)rB r# r# r# r% �<lambda>� � rD c @ s$ e Zd Zedd� �Zedd� �ZdS )�HubMetac C s( t �d�}|dkr$tt�}t �|� |S )z(Returns the current instance of the hub.N)�_local�getr+ � GLOBAL_HUB�set)�clsrA r# r# r% r, � s zHubMeta.currentc C s t S )z%Returns the main instance of the hub.)rI )rK r# r# r% �main� s z HubMeta.mainN)r; r< r= �propertyr, rL r# r# r# r% rF � s rF c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _ScopeManagerc C s" || _ t|j�| _|jd | _d S )N���)�_hub�len�_stack� _original_len�_layer)r2 �hubr# r# r% r3 � s z_ScopeManager.__init__c C s | j d }|d k st�|S )N� )rT �AssertionError)r2 �scoper# r# r% r4 � s z_ScopeManager.__enter__c C s� t | jj�}|| jk r,t�d| j| � d S || jkrXt�d|| j | jj| jd � � | jj| jd }| jj| jd d �= |d | jd kr�t�d|d | jd � n |d | jd kr�d}t�|� d S )Nz1Scope popped too soon. Popped %s scopes too many.zLeaked %s scopes: %srV z2Wrong scope found. Meant to pop %s, but popped %s.r z�init() called inside of pushed scope. This might be entirely legitimate but usually occurs when initializing the SDK inside a request handler or task/job function. Try to initialize the SDK as early as possible instead.)rQ rP rR rS r �error�warningrT )r2 r6 r7 r8 Z current_lenZlayerrZ r# r# r% r: � s* z_ScopeManager.__exit__N)r; r< r= r3 r4 r: r# r# r# r% rN � s rN c @ s< e Zd ZdZdZer dZdZd:dd�Zdd� Z dd� Z d d � Z d d � Z e d d� �Ze dd� �Zdd� Zdd� Zd;dd�Zd<dd�Zd=dd�Zdd � Zd>d d �Zd?d d �Zd@d!d"�ZedAd#d$��Zed%d$� �ZdBd&d$�Zd'd(� ZedCd)d*��Z ed+d*� �Z dDd,d*�Z dEd.d/�Z d0d1� Z d2d3� Z d4d5� Z dFd6d7�Z!dGd8d9�Z"dS )Hr+ z�The hub wraps the concurrency management of the SDK. Each thread has its own hub but the hub might transfer with the flow of execution if context vars are available. If the hub is used with a with statement it's temporarily activated. Nc C s^ t |t�r0|}|jd \}}|d kr4t�|�}n|}|d krBt� }||fg| _d | _g | _d S )NrO )� isinstancer+ rR r' r �_last_event_id� _old_hubs)r2 Z client_or_hubrX rU r- Z other_scoper# r# r% r3 � s z Hub.__init__c C s | j �tj� t�| � | S )N)r] �appendr+ r, rG rJ )r2 r# r# r% r4 � s z Hub.__enter__c C s | j �� }t�|� d S )N)r] �poprG rJ )r2 r6 r7 r8 �oldr# r# r% r: � s z Hub.__exit__c C s | � |� S Q R X dS )z~Runs a callback in the context of the hub. Alternatively the with statement can be used on the hub directly. Nr# )r2 �callbackr# r# r% �run� s zHub.runc C sT t |t�r|}n|jdk r"|j}ntd��| j}|dk rP|j�|�}|dk rP|S dS )a Returns the integration for this hub by name or class. If there is no client bound or the client does not have that integration then `None` is returned. If the return value is not `None` the hub is guaranteed to have a client attached. NzIntegration has no name)r[ �str� identifier� ValueErrorr- Z integrationsrH )r2 Z name_or_classZintegration_namer- rA r# r# r% �get_integration s zHub.get_integrationc C s | j d d S )z&Returns the current client on the hub.rO r )rR )r2 r# r# r% r- s z Hub.clientc C s | j d d S )z%Returns the current scope on the hub.rO rV )rR )r2 r# r# r% rX " s z Hub.scopec C s | j S )zReturns the last event ID.)r\ )r2 r# r# r% � last_event_id( s zHub.last_event_idc C s | j d }||d f| j d< dS )z Binds a new client to the hub.rO rV N)rR )r2 �new�topr# r# r% r> - s zHub.bind_clientc K sF | j d \}}t|||�}|dk rB|�|||�}|dk r>|| _|S dS )zGCaptures an event. Alias of :py:meth:`sentry_sdk.Client.capture_event`.rO N)rR r* � capture_eventr\ )r2 �event�hintrX � scope_argsr- Z top_scoperA r# r# r% rj 5 s zHub.capture_eventc K s6 | j dkrdS |dkrd}| j||d�fd|i|��S )z�Captures a message. The message is just a string. If no level is provided the default level is `info`. :returns: An `event_id` if the SDK decided to send the event (see :py:meth:`sentry_sdk.Client.capture_event`). N�info)�message�levelrX )r- rj )r2 ro rp rX rm r# r# r% �capture_messageG s zHub.capture_messagec K s~ | j }|dkrdS |dk r$t|�}nt�� }t||jd�\}}y| j|f||d�|��S tk rx | �t�� � Y nX dS )z�Captures an exception. :param error: An exception to catch. If `None`, `sys.exc_info()` will be used. :returns: An `event_id` if the SDK decided to send the event (see :py:meth:`sentry_sdk.Client.capture_event`). N)Zclient_options)rl rX ) r- r �sys�exc_infor r. rj � Exception�_capture_internal_exception)r2 rY rX rm r- rs rk rl r# r# r% �capture_exception\ s zHub.capture_exceptionc C s t jd|d� dS )z� Capture an exception that is likely caused by a bug in the SDK itself. These exceptions do not end up in Sentry and are just logged instead. z Internal error in sentry_sdk)rs N)r rY )r2 rs r# r# r% ru y s z Hub._capture_internal_exceptionc K s� | j d \}}|dkr$t�d� dS t|p,d�}|�|� |sBdS t|pJd�}|�d�dkrht�� |d< |�d�dkr~d|d< |jd dk r�|jd ||�}n|}|dk r�|j � |� n t�d |� |jd }x t |j �|kr�|j � � q�W dS ) a Adds a breadcrumb. :param crumb: Dictionary with the data as the sentry v7/v8 protocol expects. :param hint: An optional value that can be used by `before_breadcrumb` to customize the breadcrumbs that are emitted. rO Nz*Dropped breadcrumb because no client boundr# � timestamp�type�defaultZbefore_breadcrumbz)before breadcrumb dropped breadcrumb (%s)�max_breadcrumbs) rR r rn �dict�updaterH r �utcnowr. Z _breadcrumbsr^ rQ �popleft)r2 Zcrumbrl r@ r- rX Z new_crumbrz r# r# r% �add_breadcrumb� s, zHub.add_breadcrumbc K s� t |t�sd|krdd}t |t�r4t�|� | �|�S d|krdt�|� |�d�}| jf d|i|��S |dk rp|S |�d| � | jj}|dk r�|j f |�S t f |�S )a" Create and start timing a new span whose parent is the currently active span or transaction, if any. The return value is a span instance, typically used as a context manager to start and stop timing in a `with` block. Only spans contained in a transaction are sent to Sentry. Most integrations start a transaction at the appropriate time, for example for every incoming HTTP request. Use `start_transaction` to start a new transaction when one is not already in progress. � transactionzcDeprecated: use start_transaction to start transactions and Transaction.start_child to start spans.�nameNrU ) r[ r r rZ �start_transactionr_ � setdefaultrX �spanZ start_childr )r2 r� r@ Zdeprecation_msgr� r# r# r% � start_span� s zHub.start_spanc K s� |� di �}|dkr*|�d| � tf |�}|�� |jd�}|�|� |j|d� |jr~| jrn| jj d � d�ppd}|j |d � |S ) a� Start and return a transaction. Start an existing transaction if given, otherwise create and start a new transaction with kwargs. This is the entry point to manual tracing instrumentation. A tree structure can be built by adding child spans to the transaction, and child spans to other spans. To start a new child span within the transaction or any span, call the respective `.start_child()` method. Every child span must be finished before the transaction is finished, otherwise the unfinished spans are discarded. When used as context managers, spans and transactions are automatically finished at the end of the `with` block. If not using context managers, call the `.finish()` method. When the transaction is finished, it will be sent to Sentry with all its finished child spans. �custom_sampling_contextNrU )Ztransaction_context�parent_sampled)�sampling_contextZ _experiments� max_spansi� )�maxlen) r_ r� r Zto_jsonr� r| Z _set_initial_sampling_decisionZsampledr- r. rH Zinit_span_recorder)r2 r� r@ r� r� r� r# r# r% r� � s zHub.start_transactionc C s d S )Nr# )r2 ra r# r# r% � push_scope s zHub.push_scopec C s d S )Nr# )r2 ra r# r# r% r� s c C sX |dk r(| � � �}||� W dQ R X dS | jd \}}|t�|�f}| j�|� t| �S )a. Pushes a new layer on the scope stack. :param callback: If provided, this method pushes a scope, calls `callback`, and pops the scope again. :returns: If no `callback` is provided, a context manager that should be used to pop the scope again. NrO )r� rR r' r^ rN )r2 ra rX r- Z new_layerr# r# r% r� # s c C s | j �� }| j std��|S )z{ Pops a scope layer from the stack. Try to use the context manager :py:meth:`push_scope` instead. z"stack must have at least one layer)rR r_ rW )r2 rA r# r# r% �pop_scope_unsafe; s zHub.pop_scope_unsafec C s d S )Nr# )r2 ra r# r# r% �configure_scopeF s zHub.configure_scopec C s d S )Nr# )r2 ra r# r# r% r� M s c sB | j d \� �|dk r*� dk r&|�� dS t� �fdd��}|� S )z� Reconfigures the scope. :param callback: If provided, call the callback with the current scope. :returns: If no callback is provided, returns a context manager that returns the scope. rO Nc 3 s � d k r�V nt � V d S )N)r r# )r- rX r# r% �innerh s z"Hub.configure_scope.<locals>.inner)rR r )r2 ra r� r# )r- rX r% r� T s � applicationc C sJ | � � | jd \}}t|r&|jd nd|r6|jd nd|j|d�|_dS )zStarts a new session.rO �releaseN� environment)r� r� �user� session_mode)� end_sessionrR r r. �_user�_session)r2 r� r- rX r# r# r% � start_sessionr s zHub.start_sessionc C sB | j d \}}|j}d| j_|dk r>|�� |dk r>|�|� dS )z)Ends the current session if there is one.rO N)rR r� rX r5 Zcapture_session)r2 r- rX Zsessionr# r# r% r� � s zHub.end_sessionc C s | � � | jd \}}d|_dS )z�Stops automatic session tracking. This temporarily session tracking for the current scope when called. To resume session tracking call `resume_auto_session_tracking`. rO FN)r� rR � _force_auto_session_tracking)r2 r- rX r# r# r% �stop_auto_session_tracking� s z Hub.stop_auto_session_trackingc C s | j d \}}d|_dS )z�Resumes automatic session tracking for the current scope if disabled earlier. This requires that generally automatic session tracking is enabled. rO N)rR r� )r2 r- rX r# r# r% � resume_auto_session_tracking� s z Hub.resume_auto_session_trackingc C s( | j d \}}|dk r$|j||d�S dS )z> Alias for :py:meth:`sentry_sdk.Client.flush` rO N)�timeoutra )rR �flush)r2 r� ra r- rX r# r# r% r� � s z Hub.flushc c sJ |p | j j}|sdS | jd d }|o.|jd }|s8dS d|�� fV dS )z� Return HTTP headers which allow propagation of trace data. Data taken from the span representing the request, if available, or the current span on the scope if not. NrO r �propagate_tracesz sentry-trace)rX r� rR r. Zto_traceparent)r2 r� r- r� r# r# r% � iter_trace_propagation_headers� s z"Hub.iter_trace_propagation_headers)NN)NN)NN)NN)NN)N)N)N)N)N)N)r� )NN)N)#r; r< r= �__doc__rR r r, rL r3 r4 r: rb rf rM r- rX rg r> rj rq rv ru r r� r� r r� r� r� r� r� r� r� r� r� r# r# r# r% r+ � sT - , 4 r+ );r' rr r � contextlibr Zsentry_sdk._compatr Zsentry_sdk.scoper Zsentry_sdk.clientr Zsentry_sdk.tracingr r Zsentry_sdk.sessionr Zsentry_sdk.utilsr r r r Zsentry_sdk._typesr �typingr r r r r r r r r r r r Zsentry_sdk.integrationsr r r r r r Zsentry_sdk.constsr! r" rG r* r/ �objectr0 rB rC rx rF rN r+ rI rJ r# r# r# r% �<module> sZ 0 ~