관리-도구
편집 파일: graphene.cpython-38.pyc
U �Mg9 � @ s d dl mZ d dlZd dlmZ d dlmZmZ d dlm Z d dl m Z m Z m Z mZ zd dlmZ W n ek r� ed��Y nX d d lmZ er�d d lmZ d d lmZmZmZ d d lmZ d d l m Z d dl m Z d dl m!Z! G dd� de�Z"dd� Z#dd� Z$edd� �Z%dS )� )�contextmanagerN)�OP)� DidNotEnable� Integration)�should_send_default_pii)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�package_version)�schemazgraphene is not installed)� TYPE_CHECKING)� Generator)�Any�Dict�Union)�Source)�ExecutionResult)� GraphQLSchema)�Eventc @ s e Zd ZdZedd� �ZdS )�GrapheneIntegration�graphenec C s2 t d�} | d krtd��| dk r(td��t� d S )Nr z Unparsable graphene version.)� r z graphene 3.3 or newer required.)r r �_patch_graphql)�version� r �N/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/graphene.py� setup_once! s z GrapheneIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr r r r r r s r c s> t j�t j� tt���fdd��} � fdd�}| t _|t _d S )Nc s� t �� }|�t� t| ||�� � | |f|�|�}W 5 Q R X t� �H t �� }|jpTg D ].}t||j t j dd�d�\}} t j || d� qVW 5 Q R X |S �NF)�typeZhandled)Zclient_optionsZ mechanism)�hint) � sentry_sdk�get_isolation_scope�add_event_processor�_event_processor� graphql_spanr � get_client�errorsr �optionsr r � capture_event) r �source�args�kwargs�scope�result�client�error�eventr$ )�old_graphql_syncr r � _sentry_patched_graphql_sync4 s �� z4_patch_graphql.<locals>._sentry_patched_graphql_syncc � s� t �� �t�}|d kr.� | |f|�|�I d H S t �� }|�t� t| ||�� � | |f|�|�I d H }W 5 Q R X t� �H t �� }|j p�g D ].}t ||j tj dd�d�\} } t j | | d� q�W 5 Q R X |S r" )r% r* Zget_integrationr r& r' r( r) r r+ r r, r r- ) r r. r/ r0 Z integrationr1 r2 r3 r4 r5 r$ )�old_graphql_asyncr r � _sentry_patched_graphql_asyncL s& "�� z5_patch_graphql.<locals>._sentry_patched_graphql_async)�graphene_schemaZ graphql_sync�graphqlr r )r7 r9 r )r8 r6 r r / s r c C s< t � r | �di �}d|d< n | �di ��d�r8| d d= | S )N�requestr; Z api_target�data)r � setdefault�get)r5 r$ Z request_infor r r r( k s r( c c s� |� d�}d}tj}|�� �d�r.d}tj}n|�� �d�rFd}tj}tj||d�dd�d� t� � }|j r~|j j ||d �}ntj ||d �}|� d |� |� d |� |� d |� z d V W 5 |�� X d S ) N�operation_name�queryZmutationZ subscription)r@ �operation_typezgraphql.operation)r= �category)Zcrumb)�op�namezgraphql.documentzgraphql.operation.namezgraphql.operation.type)r? r Z GRAPHQL_QUERY�strip� startswithZGRAPHQL_MUTATIONZGRAPHQL_SUBSCRIPTIONr% Zadd_breadcrumbZget_current_scope�spanZ start_childZ start_span�set_data�finish)r r. r0 r@ rB rD r1 Z _graphql_spanr r r r) w s4 ��� r) )&� contextlibr r% Zsentry_sdk.constsr Zsentry_sdk.integrationsr r Zsentry_sdk.scoper Zsentry_sdk.utilsr r r r Zgraphene.typesr r: � ImportError�typingr �collections.abcr r r r Zgraphene.language.sourcer Zgraphql.executionr Z graphql.typer Zsentry_sdk._typesr r r r( r) r r r r �<module> s, <