OwlCyberSecurity - MANAGER
Edit File: ariadne.cpython-312.pyc
� G��g� � �0 � d dl mZ d dlZd dlmZmZ d dlmZmZmZ d dl m Z d dl m Z d dl mZ d dlmZmZmZmZ ed � Zd d lmZ er&d d lmZmZmZmZ d d l m Z m Z m Z m Z d dl!m"Z" d dl#m$Z$m%Z% G d� de� Z&d� Z'd� Z(d� Z)y# e$ r ed � �w xY w)� )� import_moduleN)� get_client� capture_event)�_check_minimum_version� 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 � � e Zd ZdZed� � Zy)�AriadneIntegration�ariadnec �d � t d� } t t | � t d� t � y )Nr )r r r r �_patch_graphql)�versions �N/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/ariadne.py� setup_oncez AriadneIntegration.setup_once# s' � � "�)�,�� � 1�7�;��i� ��� N)�__name__� __module__� __qualname__� identifier� staticmethodr# � r$ r" r r s � ��J��� �r$ r c �< ���� t j �t j �t j �t t �� �fd�� } t t �� �fd�� }t t �� �fd�� }| t _ |t _ |t _ y )Nc �x �� t |� }t j � j |� �| ||� }|S )N)� _make_request_event_processor� sentry_sdk�get_isolation_scope�add_event_processor)� context_value� query_parser�data�event_processor�result�old_parse_querys �r" �_sentry_patched_parse_queryz3_patch_graphql.<locals>._sentry_patched_parse_query4 s9 �� � 8��=���&�&�(�<�<�_�M� � � �d�C��� r$ c �| �� � | g|��i |��}t |d � }t j � j |� t � }|j � rVt � 5 | D ]; }t ||j t j dd��� \ }}t ||�� �= d d d � |S |S # 1 sw Y |S xY w�N� F)�type�handled)�client_options� mechanism)�hint) � _make_response_event_processorr. r/ r0 r � is_activer r �optionsr r( r ) �errors�args�kwargsr5 r4 �client�error�eventr? �old_handle_errorss �r" �%_sentry_patched_handle_graphql_errorsz=_patch_graphql.<locals>._sentry_patched_handle_graphql_errors= s� �� � #�6�;�D�;�F�;��8����C���&�&�(�<�<�_�M�� �� � � � �,�.� #�E�"6� �'-�~�~�$6�$A�$A�',�#�#�K�E�4� "�%�d�3� $� /� � �v� � /� � �s �#AB1�1B;c � �� � | g|��i |��}t |d � }t j � j |� t � }|j � rdt � 5 | j xs g D ]; }t ||j t j dd��� \ }}t ||�� �= d d d � |S |S # 1 sw Y |S xY wr9 ) r@ r. r/ r0 r rA r rC r rB r r( r ) r5 rD rE � query_resultr4 rF rG rH r? �old_handle_query_results �r" �#_sentry_patched_handle_query_resultz;_patch_graphql.<locals>._sentry_patched_handle_query_resultU s� �� � /�v�G��G��G� �8� �a� �I���&�&�(�<�<�_�M�� �� � � � �,�.� #�]�]� 0�b� 0�E�"6� �'-�~�~�$6�$A�$A�',�#�#�K�E�4� "�%�d�3� 1� /� ��|�� /� ��s �#AB?�?C )�ariadne_graphql� parse_query�handle_graphql_errors�handle_query_resultr r )r7 rJ rN rI rM r6 s @@@r" r r . s� �� �%�1�1�O�'�=�=�� -� A� A�� � 2�O�D�� E�� � 2�4E�F�� G��. � 2�4K�L� � M� �. #>�O� �,Q�O�)�*M�O�'r$ c � � � � fd�}|S )z*Add request data and api_target to events.c �� �� t �t � s| S t � 5 t �j d� xs i j dd� � }t � r1t t � |� r | j di � }d|d< �|d< n'| j di � j d� r| d d= d d d � | S # t t f$ r | cY cd d d � S w xY w# 1 sw Y | S xY w)N�headerszContent-Lengthr �request�graphql� api_targetr3 ) � isinstance�dictr �int�get� TypeError� ValueErrorr r r � setdefault)rH r? �content_length� request_infor3 s �r" �innerz,_make_request_event_processor.<locals>.innerv s� �� ��$��%��L� (� *� �!$��X�X�i�(�.�B�3�3�4D�a�H�"�� '�(�-G�� �n�.� %� /� /� �2� >� �-6� �\�*�'+� �V�$����9�b�)�-�-�f�5��)�$�V�,�! +�$ � �� �z�*� � � � +� *� �� +�$ � �s/ �C�.B:�A"C�:C� C�C�C�C$r* )r3 rb s ` r" r- r- r s �� ��2 �Lr$ c � � � � fd�}|S )z2Add response data to the event's response context.c � �� t � 5 t � r*�j d� r| j di � }d�i|d< d d d � | S # 1 sw Y | S xY w)NrC �contextsr3 �response)r r r\ r_ )rH r? re rf s �r" rb z-_make_response_event_processor.<locals>.inner� sV �� � (� *�&�(�X�\�\�(�-C� �+�+�J��;���H�(���$� +� � � +� � �s �5A � Ar* )rf rb s ` r" r@ r@ � s �� � � �Lr$ )*� importlibr r. r r �sentry_sdk.integrationsr r r � sentry_sdk.integrations.loggingr �$sentry_sdk.integrations._wsgi_commonr �sentry_sdk.scoper �sentry_sdk.utilsr r r r rO � ImportError�typingr r r r r � ariadne.typesr r r r �graphql.language.astr �sentry_sdk._typesr r r r r- r@ r* r$ r" �<module>rr s� �� #� � 0� U� U� 9� K� 4�� �3� $�$5�6�O� !��0�0�U�U�1�7� �� � AN�H �@��} � 3� �1� 2�2�3�s �B �B