관리-도구
편집 파일: quart.cpython-38.pyc
U �Mg� � @ s� d dl Z d dlZd dlZd dlmZ d dlZd dlmZmZ d dl m Z d dl m Z d dl mZ d dlmZ d dlmZmZmZ d d lmZ er�d d lmZ d d lmZ d d lmZmZ z d dl Z W n e k r� dZ Y nX z@d d l m Z m Z m!Z!m"Z"m#Z#m$Z$ d dl%m&Z&m'Z'm(Z(m)Z)m*Z* W n e k �r< ed��Y n6X zd dl+m,Z, W n" e k �rp d dl-m,Z, Y nX dZ.G dd� de�Z/dd� Z0dd� Z1dd� Z2dd� Z3d d � Z4d d � Z5d d!� Z6dS )"� N)�wraps)� DidNotEnable� Integration)�_filter_headers)�SentryAsgiMiddleware)�should_send_default_pii)�SOURCE_FOR_STYLE)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception)� TYPE_CHECKING)�Any)�Union)�Event�EventProcessor)�has_request_context�has_websocket_context�Request�Quart�request� websocket)�got_background_exception�got_request_exception�got_websocket_exception�request_started�websocket_startedzQuart is not installed)�Scaffold)�endpoint�urlc @ s4 e Zd ZdZde� �ZdZd dd�Zedd� �Zd S ) �QuartIntegration�quartz auto.http.� r c C s" |t krtd|t f ��|| _d S )Nz7Invalid value for transaction_style: %s (must be in %s))�TRANSACTION_STYLE_VALUES� ValueError�transaction_style)�selfr$ � r& �K/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/quart.py�__init__@ s ��zQuartIntegration.__init__c C sB t �t� t�t� t�t� t�t� t�t� t� t � d S �N) r �connect�_request_websocket_startedr r �_capture_exceptionr r �patch_asgi_app�patch_scaffold_router& r& r& r' � setup_onceI s zQuartIntegration.setup_onceN)r ) �__name__� __module__� __qualname__� identifier�originr$ r( � staticmethodr/ r& r& r& r' r : s r c s t j� � fdd�} | t _d S )Nc � sX t �� �t�d kr&�� |||�I d H S t�� fdd�tjd�}|j|_||||�I d H S )Nc s � �f| �|�S r) r& )�a�kw)�old_appr% r& r' �<lambda>a � zApatch_asgi_app.<locals>.sentry_patched_asgi_app.<locals>.<lambda>)Z span_origin)� sentry_sdk� get_client�get_integrationr r r4 Z _run_asgi3�__call__)r% �scopeZreceive�sendZ middleware�r8 )r% r' �sentry_patched_asgi_app[ s �z/patch_asgi_app.<locals>.sentry_patched_asgi_app)r r> )rB r& rA r' r- W s r- c s t j� � fdd�} | t _d S )Nc s �| |�� � fdd�}|S )Nc sB t �� �r:t�� �s:t� �tt� �� fdd���}�|�S �� �S )Nc s* t �� }|jd k r t�� j|j_� | |�S r) )r; �get_isolation_scopeZprofile� threading�current_thread�identZactive_thread_id)�args�kwargsr? ��old_funcr& r' � _sentry_funcy s �zTpatch_scaffold_route.<locals>._sentry_route.<locals>.decorator.<locals>._sentry_func)�inspect� isfunction�asyncio�iscoroutinefunctionr r r )rJ rK �Z old_decoratorrI r' � decoratorr s � z>patch_scaffold_route.<locals>._sentry_route.<locals>.decoratorr& )rG rH rQ �Z old_routerP r' � _sentry_routen s z+patch_scaffold_route.<locals>._sentry_route)r Zroute)rS r& rR r' r. j s r. c C sF z,|j j|j jd�}| j|| t| d� W n tk r@ Y nX d S )N)r r )�source)Zurl_rule�ruler Zset_transaction_namer � Exception)r? r$ r Zname_for_styler& r& r' � _set_transaction_name_and_source� s �� rW c � sj t �� �t�}|d krd S t� r(t�� }t� r6t�� }t t � � |j |� t � � }t | ||�}|�|� d S r) )r; r< r= r r r Z_get_current_objectr r rW Zget_current_scoper$ rC � _make_request_event_processorZadd_event_processor)�apprH � integrationZrequest_websocketr? Z evt_processorr& r& r' r+ � s � r+ c s � fdd�}|S )Nc s~ � d kr | S t � �b | �di �}� j|d< � j|d< � j|d< tt� j��|d< t� rpd� j d i|d< t | � W 5 Q R X | S ) Nr r � query_string�method�headersZ REMOTE_ADDRr �env) r � setdefaultr r[ r\ r �dictr] r Z access_route�_add_user_to_event)�event�hintZ request_info�r r& r' �inner� s z,_make_request_event_processor.<locals>.innerr& )rY r rZ re r& rd r' rX � s rX c � sJ t �� �t�}|d krd S t|t �� jddd�d�\}}t j||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)rc )r; r< r= r r �optionsZ capture_event)Zsender� exceptionrH rZ rb rc r& r& r' r, � s � r, c C sL t d kr d S t j}|d kr d S t� � | �di �}t jj|d< W 5 Q R X d S )N�user�id)� quart_authZ current_userr r_ Z_auth_id)rb ri Z user_infor& r& r' ra � s ra )7rN rL rD � functoolsr r; Zsentry_sdk.integrationsr r Z$sentry_sdk.integrations._wsgi_commonr Z sentry_sdk.integrations.asgir Zsentry_sdk.scoper Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r �typingr r r Zsentry_sdk._typesr r rk � ImportErrorr r r r r r r Z quart.signalsr r r r r Zquart.scaffoldr Zflask.sansio.scaffoldr" r r- r. rW r+ rX r, ra r& r& r& r' �<module> sJ $