OwlCyberSecurity - MANAGER
Edit File: beam.cpython-312.pyc
� d��g> � �( � d dl Z d dlZd dlmZ d dlZd dlmZ d dlmZ d dl m Z m Z m Z m Z d dlmZ er5d dlmZ d dlmZ d d lmZ d d lmZ d d lmZ ed � Z ed edef �� ZdZdZdZ G d� de� Zd� Z d� Z e e� d� � Z d� Z d� Z y)� N)�wraps)� Integration)� ignore_logger)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�reraise)� TYPE_CHECKING)�Any)�Iterator)�TypeVar)�Callable)�ExcInfo�T�F.)�boundz _wrapped_{}_z _inspect_{}� _sentry_used_c � � e Zd ZdZed� � Zy)�BeamIntegration�beamc �� ��� ddl m} m} t d� t d� g d���D ], }t | t j |� t | |� � �. |j ���fd�}||_ y )Nr )�DoFn�ParDo�rootzbundle_processor.create)�process� start_bundle� finish_bundle�setupc �~ �� t | dd� s��D ]� }t ||� s�t j |� }t ||� }t |t j |� � }t |t d� r�bt |t d� r�tt |||� t ||t |� � �� d| _ � | |g|��i |�� y )N�_sentry_is_patchedFT) �getattr�hasattr� WRAPPED_FUNC�format� INSPECT_FUNC� USED_FUNC�setattr�_wrap_task_callr ) �self�fn�args�kwargs� func_name� wrapped_func� process_func� inspect_func�function_patches�old_inits ���K/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/beam.py�sentry_init_pardoz5BeamIntegration.setup_once.<locals>.sentry_init_pardo7 s� �� � �4�!5�u�=�!1�I�"�2�y�1� �#/�#6�#6�y�#A�L� $+�2�y�#9�L�#*�2�|�/B�/B�9�/M�#N�L�"�<��E�B�7�$�i��L� ��L�,�?� ��I��|�/L�M� "2�" +/��'� �T�2� /�� /�� /� ) �apache_beam.transforms.corer r r r' r% r$ �_wrap_inspect_call�__init__)r r r- r4 r1 r2 s @@r3 � setup_oncezBeamIntegration.setup_once% se �� � <��f� ��/�0�P��)�I� �� �#�#�I�.�"�4��3� � *� �>�>�� 0�0 +��r5 N)�__name__� __module__� __qualname__� identifier� staticmethodr9 � r5 r3 r r " s � ��J��)+� �)+r5 r c �N �� t | �� sy �fd�}t |t d� |S )Nc � �� t j �� }t | |� r t | |� }n/t | �� }t | �t |� � t | ||� ddlm} ||� S # t $ r ddl m } ||� cY S w xY w)zG Inspect function overrides the way Beam gets argspec. r )�get_function_args_defaults)�getfullargspec) r# r$ r" r! r' r( r6 rB � ImportError� apache_beam.typehints.decoratorsrC )r) r. r/ rB rC r- s �r3 �_inspectz$_wrap_inspect_call.<locals>._inspectX s� �� � $�*�*�9�5� � �4� � &�"�4� �6�L�"�4��3�L� �D�)�_�\�%B� C� �D�,� � 5� 0� N�-�l�;� ;��� 0� G�!�,�/� /� 0�s � A. �.B�BT)r" r' r& )�clsr- rF s ` r3 r7 r7 R s* �� � �3� � "��0�0 �H�i��&� �Or5 c �P � � t � � � fd�� }t |t d� |S )z< Wrap task call with a try catch to get exceptions. c � �� �| i |��}t t j � s|S t |� S # t $ r t � Y �<w xY w)N)� Exception�raise_exception� isinstance�types� GeneratorType�_wrap_generator_call)r+ r, �gen�funcs �r3 �_innerz _wrap_task_call.<locals>._innerz sP �� � ���'��'�C� �#�u� 2� 2�3��J�#�C�(�(�� � � � � �s �2 �A�AT)r r' r& )rQ rR s ` r3 r( r( t s. �� � �4�[� )� � )� �F�I�t�$� �Mr5 c � � t j � }t | |j ddd��� \ }}t j ||�� y)z( Send Beam exception to Sentry. r F)�type�handled)�client_options� mechanism)�hintN)� sentry_sdk� get_clientr �options� capture_event)�exc_info�client�eventrX s r3 �_capture_exceptionr` � sD � � � "� "� $�F�&�� �~�~�!�e�4��K�E�4� � � �U��.r5 c � � t j � } t � 5 t | � ddd� t | � y# 1 sw Y �xY w)z Raise an exception. N)�sysr] r r` r )r] s r3 rK rK � s3 � � �|�|�~�H� $� &��8�$� '� �X�� '� &�s � <�Ac # �p K � t | � �� �# t $ r Y yt $ r t � Y � w xY w�w)z4 Wrap the generator to handle any failures. N)�next� StopIterationrJ rK )rP s r3 rO rO � s? � �� � � ��s�)�O� �� � � �� � � � �s$ �6� �6� 3�6�3�6�3�6)!rb rM � functoolsr rY �sentry_sdk.integrationsr � sentry_sdk.integrations.loggingr �sentry_sdk.utilsr r r r �typingr r r r r �sentry_sdk._typesr r r r# r% r& r r7 r( r` rK rO r? r5 r3 �<module>rl s� �� � � � � /� 9�� � !��� � � �)��� �A���8�C��H�-�.�A� � � � � � �-+�k� -+�` �D�, �O�,� /� -� /� � r5