OwlCyberSecurity - MANAGER
Edit File: server.cpython-312.pyc
� G��g� � � � d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z d dl m Z d dl m Z e rd dlmZmZ d d l mZmZ d dlZd d lmZmZ d d lmZmZ G d � dej4 j6 � Zy# e$ r ed � �w xY w)� N)�OP)� DidNotEnable)� SPAN_ORIGIN)� Transaction�TransactionSource)�event_from_exception)� TYPE_CHECKING)� Awaitable�Callable)�Any�Optional)�HandlerCallDetails�RpcMethodHandler)� AbortError�ServicerContextzgrpcio is not installedc �, � � e Zd Zd� fd� Zd� Zd� Z� xZS )�ServerInterceptorc �L �� |xs | j | _ t �| � � y �N)� _find_name�_find_method_name�super�__init__)�self� find_name� __class__s ��V/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/grpc/aio/server.pyr zServerInterceptor.__init__ s �� �!*�!=�d�o�o�� � ���� c � �� � �K � |� _ ||� � d { ��� ���y �j s#�j st j }�� fd�}n��j s"�j rt j }�fd�}n[�j r"�j st j }�fd�}n-�j r!�j rt j }�fd�} �j �j �� S 7 �ޭw)Nc � � �K � �j |� }|s �| |� � d { ��� S t j t |j � � t j |t j t �� }t j |�� 5 �j | |� � d { ��� cd d d � S 7 ��7 �# t $ r � t $ r0}t |ddd��� \ }}t j ||�� � d }~ww xY w# 1 sw Y y xY w�w)N)�op�name�source�origin)� transaction�grpcF)�type�handled)� mechanism)�hint)r r �continue_from_headers�dict�invocation_metadatar � GRPC_SERVERr �CUSTOMr � sentry_sdk�start_transaction� unary_unaryr � Exceptionr � capture_event) �request�contextr" r% �exc�eventr* �handlerr s ��r �wrappedz4ServerInterceptor.intercept_service.<locals>.wrapped( s� �� �� ��-�-�g�6���!(��'�!:�:�:� *� ?� ?���4�4�6�7��~�~� �,�3�3�&� � � �1�1�k�J� �%,�%8�%8��'�%J� J� K�J� ;�� K��%� � �$� �&:� �/5�%�&H�'� ��t� #�0�0��T�B� �� �� K�J�sX � D �B4�A+D �C=�B8�&B6�'B8�* D �6B8�8C:� +C5�5C:�:C=�=D�D c �V �K � �j | |� 2 3 d { ��� }|��� �7 � 6 y �wr )� unary_stream�r5 r6 �rr9 s �r r: z4ServerInterceptor.intercept_service.<locals>.wrappedG s, �� �� � &� 3� 3�G�W� E� �!��G� �� E�� �)�'�%�'�)�'�)c � �H �K � �j | |� }|� d { ��� S 7 ��wr )� stream_unary)r5 r6 �responser9 s �r r: z4ServerInterceptor.intercept_service.<locals>.wrappedO s# �� �� �"�/�/���A�� %�~�%�~�s �"� �"c �V �K � �j | |� 2 3 d { ��� }|��� �7 � 6 y �wr )� stream_streamr= s �r r: z4ServerInterceptor.intercept_service.<locals>.wrappedW s, �� �� � &� 4� 4�W�g� F� �!��G� �� F�r? )�request_deserializer�response_serializer) �_handler_call_details�request_streaming�response_streamingr& � unary_unary_rpc_method_handler� unary_stream_rpc_method_handler� stream_unary_rpc_method_handler� stream_stream_rpc_method_handlerrE rF )r � continuation�handler_call_details�handler_factoryr: r9 s ` @r �intercept_servicez#ServerInterceptor.intercept_service s� �� �� �%9��"�$�%9�:�:�� �?���(�(��1K�1K� "� A� A�O� �8 �*�*�w�/I�/I� "� B� B�O� � � &� &�w�/I�/I� "� B� B�O� &� � &� &�7�+E�+E� "� C� C�O� � � �!(�!=�!=� '� ;� ;� � �w ;�s �C8�C6�C C8c �. � | j j S r )rG �method)r r6 s r r z ServerInterceptor._find_nameb s � ��)�)�0�0�0r r )�__name__� __module__� __qualname__r rQ r � __classcell__)r s @r r r s �� �� B �H1r r ) r0 �sentry_sdk.constsr �sentry_sdk.integrationsr �#sentry_sdk.integrations.grpc.constsr �sentry_sdk.tracingr r �sentry_sdk.utilsr �typingr �collections.abcr r r r r& r r �grpc.aior r � ImportError�aior � r r �<module>rc sf �� � � 0� ;� =� 1� ��3�$�2��9�4� M1����2�2� M1�� � 2� �0� 1�1�2�s �A2 �2B