OwlCyberSecurity - MANAGER
Edit File: hub.cpython-38.pyc
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_metaclass��Scope)�Client)�exc_info_from_error�event_from_exception�logger� ContextVarc C s | S �N� )�xr r �?/opt/alt/python38/lib/python3.8/site-packages/sentry_sdk/hub.py�overload s r Zsentry_current_hubc C s t jj} | sdS | jd S )NFZsend_default_pii)�Hub�current�client�options)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_value�tb�cr 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 �weakref�ref�_initial_client)�args�kwargsr �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)�_local�getr � GLOBAL_HUB�set�r 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���)�_hub�len�_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,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: %srC 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.)r? r> r@ rA r �error�warningrB )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)dd�Zdd� Zdd� Zd d � Zd d � Z e d d� �Z dd� Z dd� Z d*dd�Zd+dd�Zd,dd�Zdd� Zd-dd �Zed d � �Zed d � �Zd.d d �ZeZd!d"� Zed#d$� �Zed%d$� �Zd/d&d$�Zd0d'd(�ZdS )1r 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 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 |t�r|}n|jdk r"|j}ntd��| jd d }|dk rX|j�|�}|dk rX|S t}|dk rj|� }|dk r�||k r�|j�|�dk r�d|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 z�Integration %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 �str� 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 NrY )r �new�topr 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 �event�hintr rE r- r r r r] � s zHub.capture_eventc C s* | j dkrdS |dkrd}| �||d��S )zyCaptures a message. The message is just a string. If no level is provided the default level is `info`. N�info)�message�level)r r] )r ra rb r r r �capture_message s zHub.capture_messagec C sv | j }|dkrdS |dkr$t�� }nt|�}t||jd�\}}z| j||d�W S tk rp | �t�� � Y nX dS )z�Captures 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 �sys�exc_infor r r r] � Exception�_capture_internal_exception)r rF r re 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)re N)r rF )r re r r r rg 1 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 ||�}|dk r�|j � |� n t�d |� |jd }t |j �|kr�|j � � q�dS ) z�Adds 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 � timestamp�type�defaultZbefore_breadcrumbz)before breadcrumb dropped breadcrumb (%s)�max_breadcrumbs) r@ r r` �dict�updater1 r �utcnowr Z _breadcrumbsrL r? �popleft)r Zcrumbr_ r, r rE Zoriginal_crumbrl 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 rr ` s c C sX |dk r(| � � �}||� W 5 Q R X dS | jd \}}|t�|�f}| j�|� t| �S )z�Pushes 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= )rr r@ rI rL r; )r rP rE r Z new_layerr r r rr 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 rt � s c sB | j d \� �|dk r*� dk r&|�� dS t� �fdd��}|� S )zReconfigures the scope.r= Nc 3 s � d k r�V nt � V d S r r r �r rE r r �inner� s z"Hub.configure_scope.<locals>.inner)r@ r )r rP rv r ru r rt � s c C s( | j d \}}|dk r$|j||d�S dS )zAlias for self.client.flushr= N)�timeoutrP )r@ �flush)r rw rP r rE r r r rx � s z Hub.flush)NN)N)N)N)NN)N)N)NN)r$ r% r& �__doc__r@ r r r" rQ rX r6 r rZ r' r] rc rh rg rq r rr rE rs rt rx r r r r r � s: ) % r )*rd rI r( r � contextlibr �warningsr Zsentry_sdk._compatr Zsentry_sdk.scoper Zsentry_sdk.clientr Zsentry_sdk.utilsr r r r �UnionZAny�Optional�DictZTupleZList�Callabler ZContextManagerZ Integrationr0 r* r �objectr r. rj r/ r7 r; r r2 r3 r r r r �<module> s. .