Edit File: asgi.cpython-37.opt-1.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)�inspectZisclass�hasattrZ isfunction�asyncioZiscoroutinefunction�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)r. r0 )r/ r, )r. r0 r �innerb s z.SentryAsgiMiddleware._run_asgi2.<locals>.innerr )r, r/ r4 r )r/ r, r r+ ` s z SentryAsgiMiddleware._run_asgi2c � s �� �� ���fdd��I d H S )Nc s �� �� ��S )N)r% r )r. r/ r, r0 r r r1 j r2 z1SentryAsgiMiddleware._run_asgi3.<locals>.<lambda>)r3 )r, r/ r. r0 r )r. r/ r, r0 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.typer5 )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, r/ �callbackZ is_recursive_asgi_middlewarer r Z sentry_scopeZ processor�ty� transactionr r r r3 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 )NZrequestr )r6 r7 �methodr8 � query_stringr6 �ws�hostZurlr Z REMOTE_ADDRr �envrG �endpoint)r; r � _get_headers� _get_query�_get_urlr rC r ) r, r r r5 Z request_inforF r8 r rM r r r r@ � 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� )r6 ZhttpsrJ Zwssz %s://%s:%s%s)r; ) r, r/ Zdefault_schemerK rQ rR rT ZportZ default_portr r r rP � 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. rI Nzlatin-1)r; �urllib�parseZunquote�decode)r, r/ Zqsr r r rO � 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. r8 zlatin-1z, )rX )r, r/ r8 Zraw_keyZ raw_value�key�valuer r r rN � s z!SentryAsgiMiddleware._get_headersN)F) �__name__� __module__� __qualname__� __slots__r- r+ r* r3 r@ rP rO rN r r r r r( E s 0) r( )#�__doc__r# r! rV 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: rC r r' r( r r r r �<module> s*