관리-도구
편집 파일: gql.cpython-38.pyc
U �Mg � @ s* d dl Z d dlmZmZmZ d dlmZmZ d dlm Z z@d dl Z d dl m Z m Z mZmZ d dlmZmZ d dlmZ W n ek r� ed��Y nX d d lmZ er�d d lmZmZmZmZ d d l m Z m Z ee ee eed f f f Z d Z!G dd� de�Z"dd� Z#dd� Z$dd� Z%dd� Z&dd� Z'dS )� N)�event_from_exception�ensure_integration_enabled� parse_version)� DidNotEnable� Integration)�should_send_default_pii)� print_ast�get_operation_ast� DocumentNode�VariableDefinitionNode)� Transport�AsyncTransport)�TransportQueryErrorzgql is not installed)� TYPE_CHECKING)�Any�Dict�Tuple�Union)�Event�EventProcessor.)� � � c @ s e Zd ZdZedd� �ZdS )�GQLIntegration�gqlc C s@ t tj�} | d ks| tk r6tdd�dd� tD �� ��t� d S )Nz?GQLIntegration is only supported for GQL versions %s and above.�.c s s | ]}t |�V qd S )N)�str)�.0�num� r �I/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/gql.py� <genexpr>( s z,GQLIntegration.setup_once.<locals>.<genexpr>)r r � __version__�MIN_GQL_VERSIONr �join�_patch_execute)Z gql_versionr r r � setup_once! s ��zGQLIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr& r r r r r s r c C sf zBt | �}dt| �i}|d k r>|j|d< |jd k r>|jj|d< |W S ttfk r` t� Y S X d S )N�query� variablesZ operationName)r r Zvariable_definitions�name�value�AttributeError� TypeError�dict)�documentZ operation_ast�datar r r �_data_from_document- s r5 c C s$ z| j W S tk r Y dS X dS )zg The RequestsHTTPTransport allows defining the HTTP method; all other transports use POST. �POSTN)�methodr0 )� transportr r r �_transport_method= s r9 c C s@ | d kr i S dt | �i}z| j|d< W n tk r: Y nX |S )Nr7 �url)r9 r: r0 )r8 Z request_infor r r � _request_info_from_transportI s �r; c s* t jj� tt� �� fdd��} | t j_d S )Nc s� t �� }|�t| |�� z� | |f|�|�W S tk r| } z2t|t �� jddd�d�\}}t �||� |�W 5 d }~X Y nX d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism) � sentry_sdkZget_isolation_scopeZadd_event_processor�_make_gql_event_processorr r Z get_client�optionsZ capture_event)�selfr3 �args�kwargs�scope�e�event�hint�Z real_executer r �sentry_patched_execute^ s � z._patch_execute.<locals>.sentry_patched_execute)r ZClient�executer r )rH r rG r r% Z s r% c s � �fdd�}|S )Nc s� z|d d j }W n ttfk r. d }Y nX | �di �}|�ddit� j��� t� r�t��|d< | �di �}|�di �}|�d |i|d �� | S ) N�exc_infor �requestZ api_target�graphqlr4 �contexts�response�errors)r4 r< ) rO r0 �KeyError� setdefault�updater; r8 r r5 )rE rF rO rK rM rN ��clientr3 r r � processoru s, ��� ��z,_make_gql_event_processor.<locals>.processorr )rT r3 rU r rS r r> s s r> )(r= Zsentry_sdk.utilsr r r Zsentry_sdk.integrationsr r Zsentry_sdk.scoper r rL r r r r Z gql.transportr r Zgql.transport.exceptionsr � ImportError�typingr r r r r Zsentry_sdk._typesr r r Z EventDataTyper# r r5 r9 r; r% r> r r r r �<module> s,