관리-도구
편집 파일: ariadne.cpython-38.pyc
U �MgB � @ s" d dl mZ d dlZd dlmZmZ d dlmZmZ d dlm Z d dl m Z d dl m Z d dlmZmZmZmZ z ed �ZW n ek r� ed ��Y nX d d lmZ er�d d lmZmZmZmZ d d lm Z m Z m Z m Z d dl m!Z! d dl"m#Z#m$Z$ G dd� de�Z%dd� Z&dd� Z'dd� Z(dS )� )� import_moduleN)� get_client� capture_event)� DidNotEnable� Integration)� ignore_logger)�request_body_within_bounds)�should_send_default_pii)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�package_versionzariadne.graphqlzariadne is not installed)� TYPE_CHECKING)�Any�Dict�List�Optional)� GraphQLError� GraphQLResult� GraphQLSchema� QueryParser)� DocumentNode)�Event�EventProcessorc @ s e Zd ZdZedd� �ZdS )�AriadneIntegration�ariadnec C s: t d�} | d krtd��| dk r(td��td� t� d S )Nr zUnparsable ariadne version.)r � z ariadne 0.20 or newer required.)r r r �_patch_graphql)�version� r �M/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/ariadne.py� setup_once# s z AriadneIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr! r r r r r s r c sj t j�t j� t j�tt���fdd��} tt� �� fdd��}tt���fdd��}| t _|t _|t _d S )Nc s& t |�}t�� �|� � | ||�}|S )N)� _make_request_event_processor� sentry_sdk�get_isolation_scope�add_event_processor)Z context_valueZ query_parser�data�event_processor�result)�old_parse_queryr r �_sentry_patched_parse_query9 s z3_patch_graphql.<locals>._sentry_patched_parse_queryc s� � | f|�|�}t |d �}t�� �|� t� }|�� r|t� �8 | D ],}t||jt j dd�d�\}}t ||d� qDW 5 Q R X |S �N� F)�typeZhandled)Zclient_optionsZ mechanism)�hint) � _make_response_event_processorr( r) r* r � is_activer r �optionsr r% r ) �errors�args�kwargsr- r, �client�error�eventr3 )�old_handle_errorsr r �%_sentry_patched_handle_graphql_errorsB s �� z=_patch_graphql.<locals>._sentry_patched_handle_graphql_errorsc s� � | f|�|�}t |d �}t�� �|� t� }|�� r�t� �> | jpHg D ],}t||j t j dd�d�\}}t ||d� qJW 5 Q R X |S r0 ) r4 r( r) r* r r5 r r7 r r6 r r% r ) r- r8 r9 Z query_resultr, r: r; r< r3 )�old_handle_query_resultr r �#_sentry_patched_handle_query_resultZ s �� z;_patch_graphql.<locals>._sentry_patched_handle_query_result)�ariadne_graphqlZ parse_queryZhandle_graphql_errorsZhandle_query_resultr r )r/ r> r@ r )r= r? r. r r 3 s r c s � fdd�}|S )z*Add request data and api_target to events.c s� t � t�s| S t� �� z t� �d�p&i �dd��}W n( ttfk r\ | Y W 5 Q R � S X t� r�tt � |�r�| � di �}d|d< � |d< n | �di ��d�r�| d d= W 5 Q R X | S )N�headerszContent-Lengthr �request�graphqlZ api_targetr+ ) � isinstance�dictr �int�get� TypeError� ValueErrorr r r � setdefault)r< r3 �content_lengthZ request_info�r+ r r �inner{ s& � � z,_make_request_event_processor.<locals>.innerr )r+ rN r rM r r' w s r' c s � fdd�}|S )z2Add response data to the event's response context.c s> t � �. t� r0� �d�r0| �di �}d� i|d< W 5 Q R X | S )Nr7 �contextsr+ �response)r r rH rK )r< r3 rO �rP r r rN � s �z-_make_response_event_processor.<locals>.innerr )rP rN r rQ r r4 � s r4 ))� importlibr r( r r Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.loggingr Z$sentry_sdk.integrations._wsgi_commonr Zsentry_sdk.scoper Zsentry_sdk.utilsr r r r rA � ImportError�typingr r r r r Z ariadne.typesr r r r Zgraphql.language.astr Zsentry_sdk._typesr r r r r' r4 r r r r �<module> s* D