관리-도구
편집 파일: strawberry.cpython-38.pyc
U �Mgh< � @ s d dl Z d dlZd dlmZ d dlZd dlmZ d dlmZm Z d dl m Z d dl m Z d dlmZ d dlmZmZmZmZmZmZ zd d l mZ W n ek r� e d ��Y nX zZd dlm mZ d d l m Z d d l m Z d d l m!Z" d dl#m$Z%m&Z' d dl(m)Z)m*Z* W n ek �r* e d��Y nX d dl+m,Z, e,�r�d dl+m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2 d dl3m4Z4m5Z5 d dl(m6Z6 d dl7m8Z8m9Z9m:Z: d dl;m<Z<m=Z= e d� G dd� de�Z>dd� Z?G d d � d e �Z@G d d � d e@�ZAd d!� ZBd"d#� ZCd$d%� ZDd&d'� ZEd(d)� ZFdS )*� N�� isawaitable)�OP)� Integration� DidNotEnable)� ignore_logger)�should_send_default_pii)� TRANSACTION_SOURCE_COMPONENT)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�logger�package_version�_get_installed_modules)�cached_propertyz;strawberry-graphql integration requires Python 3.8 or newer)�Schema)�SchemaExtension)�should_skip_tracing)�SentryTracingExtension�SentryTracingExtensionSync)�async_base_view�sync_base_viewz#strawberry-graphql is not installed)� TYPE_CHECKING)�Any�Callable� Generator�List�Optional�Union)� GraphQLError�GraphQLResolveInfo)�GraphQLHTTPResponse)�ExecutionContext�ExecutionResult�SubscriptionExecutionResult)�Event�EventProcessorzstrawberry.executionc @ s0 e Zd ZdZde� �Zddd�Zedd� �ZdS ) �StrawberryIntegration� strawberryz auto.graphql.Nc C s |dkrt d�|���|| _d S )N)NFTz6Invalid value for async_execution: "{}" (must be bool))� ValueError�format�async_execution)�selfr+ � r- �P/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/strawberry.py�__init__; s ��z StrawberryIntegration.__init__c C sD t d�} | d kr td�| ���| dk r.td��t� t� t� d S )Nzstrawberry-graphqlz)Unparsable strawberry-graphql version: {})r �� � z-strawberry-graphql 0.209.5 or newer required.)r r r* �_patch_schema_init�_patch_execute� _patch_views)�versionr- r- r. � setup_onceE s �z StrawberryIntegration.setup_once)N)�__name__� __module__� __qualname__� identifier�originr/ � staticmethodr6 r- r- r- r. r' 7 s r' c s& t j� t�� �� fdd��} | t _d S )Nc s� t �� �t�}|d kr&� | f|�|�S |�d�p2g }|jd k rF|j}n&t|�}t�d|r\dnd|rfdnd� dd� |D �}|� |r�t nt � ||d< � | f|�|�S ) N� extensionszfAssuming strawberry is running %s. If not, initialize it as StrawberryIntegration(async_execution=%s).�async�sync�False�Truec S s g | ]}|t tfkr|�qS r- )� StrawberrySentryAsyncExtension� StrawberrySentrySyncExtension)�.0� extensionr- r- r. � <listcomp>q s ��zK_patch_schema_init.<locals>._sentry_patched_schema_init.<locals>.<listcomp>) � sentry_sdk� get_client�get_integrationr' �getr+ �_guess_if_using_asyncr �info�append�SentryAsyncExtension�SentrySyncExtension)r, �args�kwargsZ integrationr= Zshould_use_async_extension�Zold_schema_initr- r. �_sentry_patched_schema_init[ s( �� �z7_patch_schema_init.<locals>._sentry_patched_schema_init)r r/ � functools�wraps)rS r- rR r. r2 W s %r2 c @ s^ e Zd Zdd�dd�Zedd� �Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )rN N��execution_contextc C s |r || _ d S �NrV )r, rW r- r- r. r/ � s z SentryAsyncExtension.__init__c C s* | � | jj�}| jjr&d�| jj|�S |S )Nz{}:{})� hash_queryrW �query�operation_namer* )r, Z query_hashr- r- r. �_resource_name� s z#SentryAsyncExtension._resource_namec C s t �|�d���� S )Nzutf-8)�hashlib�md5�encode� hexdigest)r, rZ r- r- r. rY � s z SentryAsyncExtension.hash_queryc c sJ | j j| _d}tj}| j jd kr(d| j _| j j�� �d�rFd}tj}n | j j�� �d�rbd}tj }|}| jr||d� | j�7 }t j d| j|d�d� t � � }|r�|j||tjd �| _nt j||tjd �| _| j�d |� | j�d | j� | j�d | j j� | j�d | j� d V | jj}|�r<| j j�r<| j j|_t|_||_| j�� d S )NrZ � ZmutationZ subscriptionz {}zgraphql.operation)r[ �operation_type)�category�data��op�namer; zgraphql.operation.typezgraphql.operation.namezgraphql.documentzgraphql.resource_name)rW r[ Z_operation_namer Z GRAPHQL_QUERYrZ �strip� startswithZGRAPHQL_MUTATIONZGRAPHQL_SUBSCRIPTIONr* rG Zadd_breadcrumbZget_current_span� start_childr' r; � graphql_spanZ start_span�set_datar\ Zcontaining_transactionrg r �sourcerf �finish)r, rb rf � description�spanZ transactionr- r- r. � on_operation� sV ��� � z!SentryAsyncExtension.on_operationc c s, | j jtjdtjd�| _d V | j�� d S )NZ validationre )rk rj r ZGRAPHQL_VALIDATEr' r; Zvalidation_spanrn �r, r- r- r. � on_validate� s �z SentryAsyncExtension.on_validatec c s, | j jtjdtjd�| _d V | j�� d S )NZparsingre )rk rj r Z GRAPHQL_PARSEr' r; Z parsing_spanrn rr r- r- r. �on_parse� s �z SentryAsyncExtension.on_parsec C s t ||�S rX )� strawberry_should_skip_tracing)r, �_nextrL r- r- r. r � s z(SentryAsyncExtension.should_skip_tracingc � s( |||f|�|�}t |�r$|I d H }|S rX r )r, rv �rootrL rP rQ �resultr- r- r. �_resolve� s z SentryAsyncExtension._resolvec � s� | � ||�r(| j|||f|�|�I d H S d�|j|j�}| jjtjd�|�t j d��r}|� d|j� |� d|jj � |� d|� |� dd� tt|j�� ��� | j|||f|�|�I d H W 5 Q R � S Q R X d S � Nz{}.{}z resolving {}re zgraphql.field_namezgraphql.parent_typezgraphql.field_pathz graphql.path�.)r ry r* � parent_type� field_namerk rj r �GRAPHQL_RESOLVEr' r; rl rg �join�map�str�path�as_list�r, rv rw rL rP rQ Z field_pathrp r- r- r. �resolve� s � z SentryAsyncExtension.resolve) r7 r8 r9 r/ r r\ rY rq rs rt r ry r� r- r- r- r. rN � s � : rN c @ s e Zd Zdd� ZdS )rO c O s� | � ||�r |||f|�|�S d�|j|j�}| jjtjd�|�tj d��h}|� d|j� |� d|jj � |� d|� |� dd� t t|j�� ��� |||f|�|�W 5 Q R � S Q R X d S rz )r r* r| r} rk rj r r~ r' r; rl rg r r� r� r� r� r� r- r- r. r� s � zSentrySyncExtension.resolveN)r7 r8 r9 r� r- r- r- r. rO s rO c s> t j� t j�� fdd�} tt���fdd��}| t _|t _d S )Nc � sP � | |�I d H }t �� �t�d kr&|S d|krLt �� }t|d �}|�|� |S �NrW )rG rH rI r' �get_isolation_scope� _make_request_event_processor�add_event_processor�rP rQ rx �scope�event_processor)�old_execute_asyncr- r. � _sentry_patched_execute_async) s z5_patch_execute.<locals>._sentry_patched_execute_asyncc s4 � | |�}d|kr0t �� }t|d �}|�|� |S r� )rG r� r� r� r� )�old_execute_syncr- r. � _sentry_patched_execute_sync7 s z4_patch_execute.<locals>._sentry_patched_execute_sync)�strawberry_schema�executeZ execute_syncr r' )r� r� r- )r� r� r. r3 $ s r3 c sP t jj�tjj�� �fdd�} � �fdd�}tt�dd� �� | t j_|tj_d S )Nc s �| ||� � | ||� d S rX r- �r, �errors� response_data)� _sentry_patched_handle_errors� old_async_view_handle_errorsr- r. �(_sentry_patched_async_view_handle_errorsL s z>_patch_views.<locals>._sentry_patched_async_view_handle_errorsc s �| ||� � | ||� d S rX r- r� )r� �old_sync_view_handle_errorsr- r. �'_sentry_patched_sync_view_handle_errorsQ s z=_patch_views.<locals>._sentry_patched_sync_view_handle_errorsc S sp |sd S t �� }t|�}|�|� t� �> |D ]2}t|t �� jtj dd�d�\}}t j ||d� q.W 5 Q R X d S )NF)�typeZhandled)Zclient_optionsZ mechanism)�hint) rG r� � _make_response_event_processorr� r r rH �optionsr' r: Z capture_event)r, r� r� r� r� �error�eventr� r- r- r. r� V s �� z3_patch_views.<locals>._sentry_patched_handle_errors)r ZAsyncBaseHTTPViewZ_handle_errorsr ZSyncBaseHTTPViewr r' )r� r� r- )r� r� r� r. r4 G s ��r4 c s � fdd�}|S )Nc s� t � �� t� r`| �di �}d|d< |�d�s�d� ji}� jrF� j|d< � jrV� j|d< ||d< n(z| d d= W n ttfk r� Y nX W 5 Q R X | S )N�request�graphqlZ api_targetrd rZ � variablesZ operationName) r r � setdefaultrJ rZ r� r[ �KeyError� TypeError)r� r� Z request_datard rV r- r. �innerw s z,_make_request_event_processor.<locals>.innerr- )rW r� r- rV r. r� t s r� c s � fdd�}|S )Nc s4 t � �$ t� r&| �di �}d� i|d< W 5 Q R X | S )N�contextsrd �response)r r r� )r� r� r� �r� r- r. r� � s z-_make_response_event_processor.<locals>.innerr- )r� r� r- r� r. r� � s r� c C s2 t | kr dS t| krdS tddddhtt� �@ �S )NTFZ starletteZstarliteZlitestarZfastapi)rB rC �bool�setr )r= r- r- r. rK � s �rK )GrT r] �inspectr rG Zsentry_sdk.constsr Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.loggingr Zsentry_sdk.scoper Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r r r r � ImportErrorZstrawberry.schema.schemaZschemar� r( r Zstrawberry.extensionsr Z#strawberry.extensions.tracing.utilsr ru Z strawberry.extensions.tracingr rB r rC Zstrawberry.httpr r �typingr r r r r r r r� r r r! Zstrawberry.typesr" r# r$ Zsentry_sdk._typesr% r& r' r2 rN rO r3 r4 r� r� rK r- r- r- r. �<module> sP - #-