OwlCyberSecurity - MANAGER
Edit File: client.cpython-312.pyc
� G��g� � � � d dl mZmZmZmZ d dlmZmZmZm Z m Z m Z d dl m Z d dlZd dlmZ d dlmZ G d� d� Z G d � d ee� Z G d � d ee� Zy) � )�Callable�Union� AsyncIterable�Any)�UnaryUnaryClientInterceptor� UnaryStreamClientInterceptor�ClientCallDetails�UnaryUnaryCall�UnaryStreamCall�Metadata)�MessageN)�OP)� SPAN_ORIGINc �& � e Zd Zededefd�� Zy)�ClientInterceptor�client_call_details�returnc �r � | j �| j t � �� } nIt | j t � s/| j t j | j � �� } t j � j � D ]! \ }}| j j ||� �# | S )N)�metadata) r �_replacer � isinstance� from_tuple� sentry_sdk�get_current_scope� iter_trace_propagation_headers�add)r �key�values �V/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/grpc/aio/client.py�/_update_client_call_details_metadata_from_scopezAClientInterceptor._update_client_call_details_metadata_from_scope s� � � � '� '� /�"5�">�">�� �">�"S� �� /� 8� 8�(�C� #6�">�">�!�,�,�-@�-I�-I�J� #?� #� � � )� )� +� J� J� L� � � � � (� (� ,� ,�S�%� 8� M�"�"� N)�__name__� __module__� __qualname__� staticmethodr r � r! r r r s$ � ��#� .�#� �#� �#r! r c �< � e Zd Zdeeegef dededeeef fd�Zy)�!SentryUnaryUnaryClientInterceptor� continuationr �requestr c � �� K � |j }t j t j d|j � z t �� 5 }|j dd� |j d|� | j |� } |||� � d { ��� }|j � � d { ��� }|j d|j � |cd d d � S 7 �C7 �-# 1 sw Y y xY w�w)Nzunary unary call to %s��op�name�origin�typez unary unary�method�code) r1 r � start_spanr � GRPC_CLIENT�decoder �set_datar r2 r. )�selfr) r r* r1 �span�response� status_codes r �intercept_unary_unaryz7SentryUnaryUnaryClientInterceptor.intercept_unary_unary( s� � �� � %�+�+�� � "� "��~�~�)�F�M�M�O�;� � � � �M�M�&�-� 0� �M�M�(�F� +�"&�"V�"V�#�#� � *�*=�w� G�G�H� (� � ��/�K� �M�M�&�+�"2�"2� 3��! � � H��/�� � �sC �AC � AC� C� C�$C�%!C� C �C�C�C �C N) r" r# r$ r r r r r r; r& r! r r( r( ' sG � � � � 1�7� ;�^� K�L� � /� � � � �~�w�&� '� r! r( c �B � e Zd Zdeeegef dededeee ef fd�Z y)�"SentryUnaryStreamClientInterceptorr) r r* r c � �X K � |j }t j t j d|j � z t �� 5 }|j dd� |j d|� | j |� } |||� � d { ��� }|cd d d � S 7 �# 1 sw Y y xY w�w)Nzunary stream call to %sr, r0 z unary streamr1 ) r1 r r3 r r4 r5 r r6 r )r7 r) r r* r1 r8 r9 s r �intercept_unary_streamz9SentryUnaryStreamClientInterceptor.intercept_unary_streamF s� � �� � %�+�+�� � "� "��~�~�*�V�]�]�_�<� � � � �M�M�&�.� 1� �M�M�(�F� +�"&�"V�"V�#�#� � *�*=�w� G�G�H� �! � � H�� � �s1 �AB*� AB � B � B � B*� B � B'�#B*N) r" r# r$ r r r r r r r r? r& r! r r= r= C sL � � � � 1�7� ;�_� L�M� � /� � � � �}�S�!�?�2� 3� r! r= )�typingr r r r �grpc.aior r r r r r �google.protobuf.messager r �sentry_sdk.constsr �#sentry_sdk.integrations.grpc.constsr r r( r= r& r! r �<module>rE sM �� 6� 6�� � ,� � � ;�#� #�* �(9�;V� �8 ��3� r!