OwlCyberSecurity - MANAGER
Edit File: client.cpython-312.pyc
� G��g6 � � � d dl Z d dlmZ d dlmZ d dlmZ d dlmZ erd dlm Z m Z m Z m Z m Z d dlZd dlmZmZ d dlmZ d d lmZ d d lmZ G d � d ej0 ej2 � Zy# e$ r ed � �w xY w)� N)�OP)� DidNotEnable)� SPAN_ORIGIN)� TYPE_CHECKING)�Any�Callable�Iterator�Iterable�Union)�ClientCallDetails�Call)� _UnaryOutcome)�UnaryStreamCall)�Messagezgrpcio is not installedc �, � e Zd ZdZd� Zd� Zed� � Zy)�ClientInterceptorFc �t � |j }t j t j d|z t �� 5 }|j dd� |j d|� | j |� } |||� }|j d|j � j � |cd d d � S # 1 sw Y y xY w)Nzunary unary call to %s��op�name�origin�typez unary unary�method�code) r � sentry_sdk� start_spanr � GRPC_CLIENTr �set_data�/_update_client_call_details_metadata_from_scoper r ��self� continuation�client_call_details�requestr �span�responses �R/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/grpc/client.py�intercept_unary_unaryz'ClientInterceptor.intercept_unary_unary s� � �$�+�+�� � "� "��~�~�)�F�2� � � � �M�M�&�-� 0� �M�M�(�F� +�"&�"V�"V�#�#� � $�$7��A�H� �M�M�&�(�-�-�/�"6�"6� 7�� � � �s �A*B.�.B7c � � |j }t j t j d|z t �� 5 }|j dd� |j d|� | j |� } |||� }|cd d d � S # 1 sw Y y xY w)Nzunary stream call to %sr r z unary streamr )r r r r r r r r r s r' �intercept_unary_streamz(ClientInterceptor.intercept_unary_stream/ s� � �$�+�+�� � "� "��~�~�*�V�3� � � � �M�M�&�.� 1� �M�M�(�F� +�"&�"V�"V�#�#� � $�#�W��H� �% � � �s �A B�B c �t � | j rt | j � ng }t j � j � D ] \ }}|j ||f� � t j j | j | j || j | j | j �� } | S )N)r �timeout�metadata� credentials�wait_for_ready� compression)r- �listr �get_current_scope� iter_trace_propagation_headers�append�grpc� _interceptor�_ClientCallDetailsr r, r. r/ r0 )r# r- �key�values r' r zAClientInterceptor._update_client_call_details_metadata_from_scopeG s� � � 3F�2N�2N�D�$�-�-� .�TV� � � )� )� +� J� J� L� � � � �O�O�S�%�L� )� M� #� /� /� B� B�&�-�-�'�/�/� �+�7�7�.�=�=�+�7�7� C� �� #�"� N)�__name__� __module__� __qualname__�_is_interceptedr( r* � staticmethodr � r: r' r r s' � � �O� �* �0 �#� �#r: r )r �sentry_sdk.constsr �sentry_sdk.integrationsr �#sentry_sdk.integrations.grpc.constsr �typingr r r r r r r5 r r �grpc._interceptorr �grpc.aio._interceptorr �google.protobuf.messager � ImportError�UnaryUnaryClientInterceptor� UnaryStreamClientInterceptorr r@ r: r' �<module>rK sk �� � � 0� ;� ��?�?�2��,�/�5�/� G#��$�$�d�&G�&G�G#�� � 2� �0� 1�1�2�s � A- �-A;