Edit File: asgi.cpython-37.pyc
B Z~�_! � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m Z ddl m Z ddl mZmZmZmZmZ ddlmZ er�dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlm Z m Z ed�Z dZ dd� Z dd� Z!G dd� d�Z"dS )zf An ASGI middleware. Based on Tom Christie's `sentry-asgi <https://github.com/encode/sentry-asgi>`_. � N)�partial)�MYPY)�Hub�_should_send_default_pii)�_filter_headers)� ContextVar�event_from_exception�transaction_from_function�HAS_REAL_CONTEXTVARS�CONTEXTVARS_ERROR_MESSAGE)� Transaction)�Dict)�Any)�Optional)�Callable)�Literal)�Event�HintZ sentry_asgi_middleware_appliedzgeneric ASGI requestc C s8 | j d k r4t|| j jddd�d�\}}| j||d� d S )N�asgiF)�typeZhandled)Zclient_optionsZ mechanism)�hint)�clientr �optionsZ capture_event)�hub�exc�eventr � r �M/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/asgi.py�_capture_exception( s r c C sB t �| �rt| d�S t �| �r(t�| �S t| dd�}t�|�S dS )z� Try to figure out if an application object supports ASGI3. This is how uvicorn figures out the application version as well. � __await__�__call__N)�inspect�isclass�hasattr� isfunction�asyncio�iscoroutinefunction�getattr)�app�callr r r �_looks_like_asgi35 s r* c @ sR e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�SentryAsgiMiddleware)r( r Fc C s8 |st stdt ��|| _t|�r,| j| _n| j| _dS )ab Instrument an ASGI application with Sentry. Provides HTTP/websocket data to sent events and basic handling for exceptions bubbling up through the middleware. :param unsafe_context_data: Disable errors when a proper contextvars installation could not be found. We do not recommend changing this from the default. zRThe ASGI middleware for Sentry requires Python 3.7+ or the aiocontextvars package.N)r � RuntimeErrorr r( r* � _run_asgi3r � _run_asgi2)�selfr( Zunsafe_context_datar r r �__init__H s z SentryAsgiMiddleware.__init__c s � �fdd�}|S )Nc � s �� �� ���fdd��I d H S )Nc s �� ��� ��S )N)r( r )�receive�scoper/ �sendr r �<lambda>d � z@SentryAsgiMiddleware._run_asgi2.<locals>.inner.<locals>.<lambda>)�_run_app)r1 r3 )r2 r/ )r1 r3 r �innerb s z.SentryAsgiMiddleware._run_asgi2.<locals>.innerr )r/ r2 r7 r )r2 r/ r r. ` s z SentryAsgiMiddleware._run_asgi2c � s �� �� ���fdd��I d H S )Nc s �� �� ��S )N)r( r )r1 r2 r/ r3 r r r4 j r5 z1SentryAsgiMiddleware._run_asgi3.<locals>.<lambda>)r6 )r/ r2 r1 r3 r )r1 r2 r/ r3 r r- h s z SentryAsgiMiddleware._run_asgi3c � sp t �d�}|rPy |� I d H S tk rN } zttj|� |d �W d d }~X Y nX t �d� �zttj�}|�� |�� �,}|�� d|_ t | j |d�}|� |� W d Q R X |d }|dkr�t jt|d �d�|�d �} n t d d �} t| _| �d |� |j| d |id ��H y |� I d H S tk �rF } zt||� |d �W d d }~X Y nX W d Q R X W d Q R X W d t �d� X d S )NFTr )� asgi_scoper )�http� websocket�headersz {}.server)�opz asgi.serverz asgi.typer8 )Zcustom_sampling_context)�_asgi_middleware_applied�get� Exceptionr r �current�setZconfigure_scopeZclear_breadcrumbs�_namer �event_processorZadd_event_processorr Zcontinue_from_headers�dict�format�_DEFAULT_TRANSACTION_NAME�nameZset_tagZstart_transaction) r/ r2 �callbackZ is_recursive_asgi_middlewarer r Z sentry_scope� processor�ty� transactionr r r r6 l s@ 0z SentryAsgiMiddleware._run_appc C s� |� di �}|d }|dkrr|� d�|d< t| �|�� |d< }| �|�|d< | �||dkr`dnd|� d ��|d < |� d �}|r�t� r�d |d i|d< |� dt�tkr�|� d�}|r�t|�|d< ||d< |S )N�requestr )r9 r: �methodr; � query_stringr9 �ws�host�urlr Z REMOTE_ADDRr �envrK �endpoint)r> r � _get_headers� _get_query�_get_urlr rF r ) r/ r r r8 Z request_inforJ r; r rS r r r rC � s& z$SentryAsgiMiddleware.event_processorc C s� |� d|�}|� dd�}|� dd�|� dd� }|rBd|||f S |dk r�|\}}dd dd d �| }||kr|d ||||f S d|||f S |S ) zZ Extract URL from the ASGI scope, without also including the querystring. �scheme�serverNZ root_path� �pathz %s://%s%s�P i� )r9 �httpsrO �wssz %s://%s:%s%s)r> ) r/ r2 Zdefault_schemerP rW rX rZ �portZ default_portr r r rV � s z SentryAsgiMiddleware._get_urlc C s$ |� d�}|sdS tj�|�d��S )zj Extract querystring from the ASGI scope, in the format that the Sentry protocol expects. rN Nzlatin-1)r> �urllib�parse�unquote�decode)r/ r2 �qsr r r rU � s z SentryAsgiMiddleware._get_queryc C sX i }xN|d D ]B\}}|� d�}|� d�}||krH|| d | ||< q|||< qW |S )zf Extract headers from the ASGI scope, in the format that the Sentry protocol expects. r; zlatin-1z, )rb )r/ r2 r; Zraw_keyZ raw_value�key�valuer r r rT � s z!SentryAsgiMiddleware._get_headersN)F) �__name__� __module__� __qualname__� __slots__r0 r. r- r6 rC rV rU rT r r r r r+ E s 0) r+ )#�__doc__r% r! r_ Zsentry_sdk._functoolsr Zsentry_sdk._typesr Zsentry_sdk.hubr r Z$sentry_sdk.integrations._wsgi_commonr Zsentry_sdk.utilsr r r r r Zsentry_sdk.tracingr �typingr r r r Ztyping_extensionsr r r r= rF r r* r+ r r r r �<module> s*