관리-도구
편집 파일: asgi.cpython-38.pyc
U �Mg�1 � @ sZ d Z ddlZddlZddlmZ ddlmZ ddlZddlm Z ddl m Z ddl m Z mZmZ ddlmZmZ dd lmZ dd lmZmZmZmZmZ dd lm Z m Z m Z m Z m Z m!Z!m"Z" dd lm#Z# dd l$m%Z% e%�r&ddl$m&Z& ddl$m'Z' ddl$m(Z( ddl$m)Z) ddl$m*Z* ddl+m,Z,m-Z- e d�Z.dZ/dZ0d dd�Z1dd� Z2G d d � d �Z3dS ) ze An ASGI middleware. Based on Tom Christie's `sentry-asgi <https://github.com/encode/sentry-asgi>`. � N)�deepcopy)�partial)�continue_trace)�OP)� _get_headers�_get_request_data�_get_url)� DEFAULT_HTTP_METHODS_TO_CAPTURE� nullcontext)� track_session)�SOURCE_FOR_STYLE�TRANSACTION_SOURCE_ROUTE�TRANSACTION_SOURCE_URL� TRANSACTION_SOURCE_COMPONENT�TRANSACTION_SOURCE_CUSTOM)� ContextVar�event_from_exception�HAS_REAL_CONTEXTVARS�CONTEXTVARS_ERROR_MESSAGE�logger�transaction_from_function�_get_installed_modules)� Transaction)� TYPE_CHECKING)�Any)�Callable)�Dict)�Optional)�Tuple)�Event�HintZ sentry_asgi_middleware_appliedzgeneric ASGI request)�endpoint�url�asgic C s0 t | t�� j|dd�d�\}}tj||d� d S )NF)�typeZhandled)Zclient_optionsZ mechanism)�hint)r � sentry_sdkZ get_client�optionsZ capture_event)�exc�mechanism_type�eventr% � r+ �J/opt/hc_python/lib/python3.8/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_asgi3J s r9 c @ sL e Zd ZdZddddefdd�Zdd � Zd d � Zd d � Zdd� Z dd� Z dS )�SentryAsgiMiddleware)r7 r/ �transaction_styler) � span_origin�http_methods_to_captureFr! r# Zmanualc C s� |st stdt ��|tkr,td|tf ��|dko<dt� k}|rLt�d� || _|| _ || _ || _ || _ 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.z7Invalid value for transaction_style: %s (must be in %s)r# Z starlettez�The Sentry Python SDK can now automatically support ASGI frameworks like Starlette and FastAPI. Please remove 'SentryAsgiMiddleware' from your project. See https://docs.sentry.io/platforms/python/guides/asgi/ for more information.N)r � RuntimeErrorr �TRANSACTION_STYLE_VALUES� ValueErrorr r �warningr; r) r< r7 r= r9 � _run_asgi3r/ � _run_asgi2)�selfr7 Zunsafe_context_datar; r) r< r= Z0asgi_middleware_while_using_starlette_or_fastapir+ r+ r, �__init__d s4 ������ z SentryAsgiMiddleware.__init__c s � �fdd�}|S )Nc � s �j � | |dd�I d H S )N� �� asgi_version��_run_app)�receive�send��scoperD r+ r, �inner� s z.SentryAsgiMiddleware._run_asgi2.<locals>.innerr+ )rD rN rO r+ rM r, rC � s z SentryAsgiMiddleware._run_asgi2c � s | j |||dd�I d H S )N� rG rI )rD rN rK rL r+ r+ r, rB � s z SentryAsgiMiddleware._run_asgi3c � s� t �d�}|d dk}|s |r�z:|dkr@| �|�|� �I d H W S | �||� �I d H W S W n6 tk r� } zt|| jd� |d �W 5 d }~X Y nX t �d� �zt�� ��}t |dd���� |� � d |_ t | j |d �} |�| � |d } | �| j|�\} } |�d d ��� } d �| | jk�r�| d k�rPtt|�d�| �| | | jd��t�d�� n ttj| | | jd��t�d�� �� d| � t�d�j �j � �d k �r�tj �d|id�nt � �� t�d�� z�� �fdd�}|dk�r| �|�||�I d H W W 5 Q R � W 5 Q R � W 5 Q R � W ��S | �|||�I d H W W 5 Q R � W 5 Q R � W 5 Q R � W �`S W n8 tk �r� } zt|| jd� |d �W 5 d }~X Y nX W 5 Q R X W 5 Q R X W 5 Q R X W 5 t �d� X d S )NFr$ ZlifespanrF )r) T�request)Z session_moder# )� asgi_scope�method� )�httpZ websocketz {}.server)�op�name�source�originz1[ASGI] Created transaction (continuing trace): %sz$[ASGI] Created transaction (new): %sz asgi.typezB[ASGI] Set transaction name and source on transaction: '%s' / '%s'rR )Zcustom_sampling_contextz [ASGI] Started transaction: %sc � s> �d k r0| � d�dko d| k}|r0��| d � � | �I d H S )Nr$ zhttp.response.start�status)�getZset_http_status)r* Zis_http_response�rL � transactionr+ r, �_sentry_wrapped_send� s �z;SentryAsgiMiddleware._run_app.<locals>._sentry_wrapped_send)!�_asgi_middleware_appliedr[ r7 � Exceptionr- r) �setr& Zisolation_scoper Zclear_breadcrumbs�_namer �event_processorZadd_event_processor� _get_transaction_name_and_sourcer; �upperr= r r �formatr< r �debugr r Z HTTP_SERVERZset_tagrW rX Zstart_transactionr )rD rN rK rL rH Z is_recursive_asgi_middlewareZ is_lifespanr( Z sentry_scope� processor�tyZtransaction_nameZtransaction_sourcerS r^ r+ r\ r, rJ � s� �� ��� � � ��� �4 �8:z SentryAsgiMiddleware._run_appc C s� |� di �}|�t|�� t|�|d< |d tkoH|d � d�tttfk}|s�| �| j |�\}}||d< d|i|d< t � d|d |d d � |S )NrQ r] Ztransaction_inforX zF[ASGI] Set transaction name and source in event_processor: '%s' / '%s') r[ �updater r �_DEFAULT_TRANSACTION_NAMEr r r rd r; r rg )rD r* r% rR Z request_dataZ already_setrW rX r+ r+ r, rc s4 ���� � �z$SentryAsgiMiddleware.event_processorc C s� d }t | }|�d�}|dkrZ|�d�}|r:t|�p6d}q�t||dkrJdndd d�}t}nN|dkr�|�d�}|r�t|d d �}|d k r�|}n t||dkr�dndd d�}t}|d kr�t}t}||fS ||fS ) Nr$ r! rT rU �ws)�hostr" �route�path)r r[ r r r r6 rk r ) rD r; rR rW rX ri r! rn ro r+ r+ r, rd - s, z5SentryAsgiMiddleware._get_transaction_name_and_sourceN) �__name__� __module__� __qualname__� __slots__r rE rC rB rJ rc rd r+ r+ r+ r, r: Z s � 3g#r: )r# )4�__doc__r4 r0 �copyr � functoolsr r& Zsentry_sdk.apir Zsentry_sdk.constsr Z$sentry_sdk.integrations._asgi_commonr r r Z$sentry_sdk.integrations._wsgi_commonr r Zsentry_sdk.sessionsr Zsentry_sdk.tracingr r r r r Zsentry_sdk.utilsr r r r r r r r �typingr r r r r r Zsentry_sdk._typesr r r_ rk r? r- r9 r: r+ r+ r+ r, �<module> s6 $