Edit File: beam.cpython-311.pyc
� ��Fh � �X � d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm Z d dl m Z m Z d dl mZ d dlmZ d d lmZ erJd d lmZ d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ ed� � Z ededef �� � Z dZ dZ dZ G d� de� � Z!d� Z"d� Z#d � Z$d � Z%d � Z&dS ) � )�absolute_importN)�wraps)�Hub)�reraise)�capture_internal_exceptions�event_from_exception)� Integration)� ignore_logger)�MYPY)�Any)�Iterator)�TypeVar)�Optional)�Callable)�Client)�ExcInfo�T�F.)�boundz _wrapped_{}_z _inspect_{}� _sentry_used_c �( � e Zd ZdZed� � � ZdS )�BeamIntegration�beamc � ��� ddl m} m} t d� � t d� � g d���D ]9}t | t � |� � t | |� � � � �:|j ���fd�}||_ d S )Nr )�DoFn�ParDo�rootzbundle_processor.create)�process� start_bundle� finish_bundle�setupc �� �� t | dd� � sʉD ]�}t ||� � s�t � |� � }t ||� � }t |t � |� � � � }t |t d� � sEt |t d� � s/t |||� � t ||t |� � � � ��d| _ � | |g|�R i |�� d S )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 ���|/builddir/build/BUILD/imunify360-venv-2.5.3/opt/imunify360/venv/lib/python3.11/site-packages/sentry_sdk/integrations/beam.py�sentry_init_pardoz5BeamIntegration.setup_once.<locals>.sentry_init_pardo6 s �� � �4�!5�u�=�=� /�!1� N� N�I�"�2�y�1�1� !� �#/�#6�#6�y�#A�#A�L� $+�2�y�#9�#9�L�#*�2�|�/B�/B�9�/M�/M�#N�#N�L�"�<��E�B�B� N�7�$�i��L� L� N� ��L�,�?�?�?� ��I��|�/L�/L�M�M�M��*.��'� �H�T�2� /�� /� /� /�� /� /� /� /� /� ) �apache_beam.transforms.corer r r r* r( r' �_wrap_inspect_call�__init__)r r r0 r7 r4 r5 s @@r6 � setup_oncezBeamIntegration.setup_once$ s� ��� � <�;�;�;�;�;�;�;��f� � � ��/�0�0�0�P�P�P��)� � �I� �� �#�#�I�.�.�"�4��3�3� � � � � �>�� 0� 0� 0� 0� 0� 0�0 +����r8 N)�__name__� __module__� __qualname__� identifier� staticmethodr< � r8 r6 r r ! s2 � � � � � ��J��)+� )+� �\�)+� )+� )+r8 r c �r ��� ddl m� t | �� � sd S ��fd�}t |t d� � |S )Nr )�getfullargspecc �R �� t � �� � }t | |� � rt | |� � }n?t | �� � }t | �t |� � � � t | ||� � ddlm} ||� � S # t $ r �|� � cY S w xY w)zG Inspect function overrides the way Beam gets argspec. r )�get_function_args_defaults) r& r' r% r$ r* r+ r9 rF � ImportError)r, r1 r2 rF r0 rD s ��r6 �_inspectz$_wrap_inspect_call.<locals>._inspectX s� �� � $�*�*�9�5�5� � �4� � &� &� 6�"�4� �6�6�L�L�"�4��3�3�L� �D�)�_�\�%B�%B� C� C� C� �D�,� � 5� 5� 5� 0� N� N� N� N� N� N�-�-�l�;�;� ;��� 0� 0� 0�!�>�,�/�/� /� /� /� 0���s �=B �B&�%B&T)� apache_beam.typehints.decoratorsrD r% r* r) )�clsr0 rH rD s ` @r6 r: r: Q sd ��� �?�?�?�?�?�?� �3� � "� "� ��t�0� 0� 0� 0� 0� 0�, �H�i��&�&�&� �Or8 c � � �� t j j �t � � � �� fd�� � }t |t d� � |S )z~ Wrap task call with a try catch to get exceptions. Pass the client on to raise_exception so it can get rebinded. c � �� �| i |��}n # t $ r t �� � Y nw xY wt |t j � � s|S t |�� � S )N)� Exception�raise_exception� isinstance�types� GeneratorType�_wrap_generator_call)r. r/ �gen�client�funcs ��r6 �_innerz _wrap_task_call.<locals>._innerz sy �� � $��$��'��'�'�C�C��� $� $� $� �F� #� #� #� #� #� $���� �#�u� 2�3�3� ��J�#�C��0�0�0s � �(�(T)r �currentrT r r* r) )rU rV rT s ` @r6 r+ r+ r sW ��� � �[� �F� �4�[�[� 1� 1� 1� 1� 1� �[� 1� �F�I�t�$�$�$� �Mr8 c � � |� t � � }|�dS |j }|�dS t | |j ddd��� � \ }}|� ||�� � dS )z( Send Beam exception to Sentry. Nr F)�type�handled)�client_options� mechanism)�hint)�get_integrationr rT r �options� capture_event)�exc_info�hub� integrationrT �eventr] s r6 �_capture_exceptionre � s � � �%�%�o�6�6�K����� �Z�F� �~���&�� �~�!�e�4�4�� � �K�E�4� ���e�$��'�'�'�'�'r8 c � � t j }|j �|� | � � t j � � }t � � 5 t ||� � ddd� � n # 1 swxY w Y t |� dS )zI Raise an exception. If the client is not in the hub, rebind it. N) r rW rT � bind_client�sysra r re r )rT rb ra s r6 rN rN � s� � � �+�C� �z�� ���� � � ��|�~�~�H� $� &� &� *� *��8�S�)�)�)�*� *� *� *� *� *� *� *� *� *� *���� *� *� *� *� �X����s � A'�'A+�.A+c # � K � t | � � V � n+# t $ r Y dS t $ r t |� � Y nw xY w�?)z4 Wrap the generator to handle any failures. TN)�next� StopIterationrM rN )rS rT s r6 rR rR � sn � � � � $� $��s�)�)�O�O�O�O�� � � � � �E�E�� $� $� $� �F� #� #� #� #� #� $���� $s � � ?�?�?)'� __future__r rh rP �sentry_sdk._functoolsr �sentry_sdk.hubr �sentry_sdk._compatr �sentry_sdk.utilsr r �sentry_sdk.integrationsr � sentry_sdk.integrations.loggingr �sentry_sdk._typesr �typingr r r r r �sentry_sdk.clientr r r r r& r( r) r r: r+ re rN rR rB r8 r6 �<module>rv s� �� &� &� &� &� &� &� � � � � � � � � '� '� '� '� '� '� � � � � � � &� &� &� &� &� &� N� N� N� N� N� N� N� N� /� /� /� /� /� /� 9� 9� 9� 9� 9� 9� "� "� "� "� "� "�� /������� � � � � � � � � � � � � � � � � � � � � � � � �(�(�(�(�(�(�)�)�)�)�)�)���� � �A����8�C��H�-�.�.�.�A� � � � � � �-+� -+� -+� -+� -+�k� -+� -+� -+�` � � �B� � �0(� (� (�* � � � $� $� $� $� $r8