U
\2 @ s d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZmZmZmZ d d
Zed Z da d d
Z G dd de Z!dd Z"G dd de#Z$G dd de Z%G dd de Z&G dd de e$Z'e' Z(e )e( dS ) N)datetime)contextmanager)warn)with_metaclassScope)Client)exc_info_from_errorevent_from_exceptionlogger
ContextVarc C s | S N )xr r /hub.pyoverload s r Zsentry_current_hubc C s t jj} | sdS | jd S )NFZsend_default_pii)Hubcurrentclientoptions)r r r r _should_send_default_pii( s r c @ s$ e Zd Zdd Zdd Zdd ZdS )
_InitGuardc C s
|| _ d S r
)_client)selfr r r r __init__1 s z_InitGuard.__init__c C s | S r
r r r r r __enter__4 s z_InitGuard.__enter__c C s | j }|d k r| d S r
)r close)r exc_type exc_valuetbcr r r __exit__7 s z_InitGuard.__exit__N__name__
__module__ __qualname__r r r" r r r r r 0 s r c O s4 t | |}tj| t|}|dk r0t|a|S )zsInitializes the SDK and optionally integrations.
This takes the same arguments as the client constructor.
N)r r r bind_clientr weakrefref_initial_client)argskwargsr rvr r r init= s
r. 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)_localgetr
GLOBAL_HUBsetr r- r r r r L s
zHubMeta.currentc C s t S )z%Returns the main instance of the hub.)r2 r r r r mainV s z HubMeta.mainN)r$ r% r& propertyr r5 r r r r r/ K s
r/ c @ s e Zd Zdd Zdd ZdS ) _HubManagerc C s t j| _t| d S r
)r r _oldr0 r3 r hubr r r r ] s z_HubManager.__init__c C s t | j d S r
)r0 r3 r8 )r r r r r r r r" a s z_HubManager.__exit__N)r$ r% r& r r" r r r r r7 \ s r7 c @ s$ e Zd Zdd Zdd Zdd ZdS )
_ScopeManagerc C s" || _ t|j| _|jd | _d S N)_hublen_stack
_original_len_layerr9 r r r r f s z_ScopeManager.__init__c C s | j d }|d k st|S )N )rB AssertionError)r scoper r r r k s
z_ScopeManager.__enter__c C s t | jj}|| jk r,td| j| d S || jkrXtd|| j | jj| jd | jj| jd }| jj| jd d = |d | jd krtd|d | jd n |d | jd krd}t| d S )Nz1Scope popped too soon. Popped %s scopes too many.zLeaked %s scopes: %srC z2Wrong scope found. Meant to pop %s, but popped %s.r zinit() 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.)r? r> r@ rA r errorwarningrB )r r r r Z current_lenZlayerrG r r r r" q s2
z_ScopeManager.__exit__Nr# r r r r r; e s r; c @ s e Zd ZdZdZd)ddZdd Zdd Zd d
Zd d Z e
d
d Z dd Z dd Z
d*ddZd+ddZd,ddZdd Zd-dd Zed d Zed d Zd.d d ZeZd!d" Zed#d$ Zed%d$ Zd/d&d$Zd0d'd(ZdS )1r zThe 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 |tr0|}|jd \}}|d kr4t|}n|}|d krBt }||fg| _d | _g | _d S r< )
isinstancer r@ copyr _last_event_id _old_hubs)r Z
client_or_hubrE r: r Z other_scoper r r r s
z Hub.__init__c C s | j tj t| | S r
)rK appendr r r0 r3 r r r r r s
z
Hub.__enter__c C s | j }t| d S r
)rK popr0 r3 )r r r r oldr r r r" s
z Hub.__exit__c
C s" | | W 5 Q R 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 r callbackr r r run s zHub.runc C s t |tr|}n|jdk r"|j}ntd| jd d }|dk rX|j|}|dk rX|S t}|dk rj| }|dk r||k r|j|dk rd|f }tt |dd t
| 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 namer= r zIntegration %r attempted to run but it was only enabled on init() but not the client that was bound to the current flow. Earlier versions of the SDK would consider these integrations enabled but this is no longer the case. )
stacklevel) rH strZ
identifier
ValueErrorr@ Z integrationsr1 r* r Warningr rG )r Z
name_or_classZintegration_namer r- Zinitial_clientrG r r r get_integration s2
zHub.get_integrationc C s | j d d S )z&Returns the current client on the hub.r= r r@ r r r r r s z
Hub.clientc C s | j S )zReturns the last event ID.)rJ r 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.r= rC NrX )r newtopr r r r' s
zHub.bind_clientc C s: | j d \}}|dk r6||||}|dk r2|| _|S dS )ak Captures an event. The return value is the ID of the event.
The event is a dictionary following the Sentry v7/v8 protocol
specification. Optionally an event hint dict can be passed that
is used by processors to extract additional information from it.
Typically the event hint object would contain exception information.
r= N)r@
capture_eventrJ )r eventhintr rE r- r r r r\ s zHub.capture_eventc C s* | j dkrdS |dkrd}| ||dS )zyCaptures a message. The message is just a string. If no level
is provided the default level is `info`.
Ninfo)messagelevel)r r\ )r r` ra r r r capture_message s
zHub.capture_messagec C sv | j }|dkrdS |dkr$t }nt|}t||jd\}}z| j||dW S tk
rp | t Y nX dS )zCaptures an exception.
The argument passed can be `None` in which case the last exception
will be reported, otherwise an exception object or an `exc_info`
tuple.
N)Zclient_options)r^ ) r sysexc_infor r
r r\ Exception_capture_internal_exception)r rF r rd r] r^ r r r capture_exception s
zHub.capture_exceptionc C s t jd|d dS )zNCapture an exception that is likely caused by a bug in the SDK
itself.z Internal error in sentry_sdk)rd N)r rF )r rd r r r rf 1 s z Hub._capture_internal_exceptionc K s | j d \}}|dkr$td dS t|p,d}|| |sBdS t|pJd}|ddkrht |d< |ddkr~d|d< |}|jd dk r|jd ||}|dk r|j
| n td | |jd
}t |j |kr|j qdS ) zAdds a breadcrumb. The breadcrumbs are a dictionary with the
data as the sentry v7/v8 protocol expects. `hint` is an optional
value that can be used by `before_breadcrumb` to customize the
breadcrumbs that are emitted.
r= Nz*Dropped breadcrumb because no client boundr Z timestamptypedefaultZbefore_breadcrumbz)before breadcrumb dropped breadcrumb (%s)max_breadcrumbs)
r@ r r_ dictupdater1 r Zutcnowr Z _breadcrumbsrL r? popleft)r Zcrumbr^ r, r rE Zoriginal_crumbrj r r r add_breadcrumb6 s,
zHub.add_breadcrumbc C s d S r
r r r r r
push_scope[ s zHub.push_scopec C s d S r
r rO r r r ro ` s c C sX |dk r(| }|| W 5 Q R X dS | jd \}}|t|f}| j| t| S )zPushes a new layer on the scope stack. Returns a context manager
that should be used to pop the scope again. Alternatively a callback
can be provided that is executed in the context of the scope.
Nr= )ro r@ rI rL r; )r rP rE r Z new_layerr r r ro e s
c C s | j }| j std|S )zaPops a scope layer from the stack. Try to use the context manager
`push_scope()` instead.z"stack must have at least one layer)r@ rM rD r4 r r r pop_scope_unsafex s
zHub.pop_scope_unsafec C s d S r
r r r r r configure_scope s zHub.configure_scopec C s d S r
r rO r r r rq s c sB | j d \ |dk r* dk r&| dS t fdd}| S )zReconfigures the scope.r= Nc 3 s d k rV nt V d S r
r r r rE r r inner s z"Hub.configure_scope.<locals>.inner)r@ r )r rP rs r rr r rq s c C s( | j d \}}|dk r$|j||dS dS )zAlias for self.client.flushr= N)timeoutrP )r@ flush)r rt rP r rE r r r ru s z Hub.flush)NN)N)N)N)NN)N)N)NN)r$ r% r& __doc__r@ r r r" rQ rW r6 r rY r' r\ rb rg rf rn r ro rE rp rq ru r r r r r s:
)
%
r )*rc rI r( r
contextlibr warningsr Zsentry_sdk._compatr Zsentry_sdk.scoper Zsentry_sdk.clientr Zsentry_sdk.utilsr r
r r ZUnionZAnyZOptionalZDictZTupleZListCallabler ZContextManagerZ Integrationr0 r* r objectr r. rh r/ r7 r; r r2 r3 r r r r <module> s.
.