Edit File: beam.cpython-37.pyc
B Z~�_ � @ s0 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 er�d d lmZ d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ ed�Z ededef d�Z dZ dZ dZ G dd� de�Z!dd� Z"d d � Z#d d � Z$d d!� Z%d"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.)Zboundz _wrapped_{}_z _inspect_{}Z _sentry_used_c @ s e Zd ZdZedd� �ZdS )�BeamIntegration�beamc sp ddl m} m} td� td� ddddg� x$� D ] }t| t�|�t| |�� q2W |j�� �fd d �}||_d S ) Nr )�DoFn�ParDo�rootzbundle_processor.create�processZ start_bundleZ finish_bundle�setupc s� t | dd�s�xp� D ]h}t||�s"qt�|�}t ||�}t |t�|��}t |td�st |td�st|||� t||t|�� qW d| _�| |f|�|� 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_funcZ inspect_func)�function_patches�old_init� �M/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/beam.py�sentry_init_pardo6 s z5BeamIntegration.setup_once.<locals>.sentry_init_pardo) �apache_beam.transforms.corer r r r" r r �_wrap_inspect_call�__init__)r r r( r/ r- )r+ r, r. � setup_once$ s zBeamIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr3 r- r- r- r. r ! s r c s8 ddl m� t| � �sd S � �fdd�}t|td� |S )Nr )�getfullargspecc sx t �� �}t| |�r t| |�}n&t| � �}t| � t|�� t| ||� yddlm} ||�S tk rr �|�S X dS )zG Inspect function overrides the way Beam gets argspec. r )�get_function_args_defaultsN) r r r r r" r# r0 r: � ImportError)r$ r) r* r: )r( r9 r- r. �_inspectX s z$_wrap_inspect_call.<locals>._inspectT)Z apache_beam.typehints.decoratorsr9 r r" r! )�clsr( r< r- )r( r9 r. r1 Q s r1 c s. t jj� t��� �fdd��}t|td� |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 sF y�| |�}W n t k r* t� � Y nX t|tj�s<|S t|� �S )N)� Exception�raise_exception� isinstance�types� GeneratorType�_wrap_generator_call)r& r' �gen)�client�funcr- r. �_innerz s z _wrap_task_call.<locals>._innerT)r �currentrE r r" r! )rF rG r- )rE rF r. r# r s r# c C sT |� t�}|dkrdS |j}|dkr(dS t| |jddd�d�\}}|j||d� dS )z( Send Beam exception to Sentry. Nr F)�typeZhandled)Zclient_optionsZ mechanism)�hint)Zget_integrationr rE r �optionsZ capture_event)�exc_info�hubZ integrationrE �eventrJ r- r- r. �_capture_exception� s rO c C sJ t j}|jdkr|�| � t�� }t� � t||� W dQ R X t|� dS )zI Raise an exception. If the client is not in the hub, rebind it. N) r rH rE Z bind_client�sysrL r rO r )rE rM rL r- r- r. r? � s r? c c sJ xDyt | �V W q tk r& P Y q tk r@ t|� Y qX qW dS )z4 Wrap the generator to handle any failures. N)�next� StopIterationr> r? )rD rE r- r- r. rC � s rC )'� __future__r rP rA Zsentry_sdk._functoolsr Zsentry_sdk.hubr Zsentry_sdk._compatr Zsentry_sdk.utilsr r Zsentry_sdk.integrationsr Z sentry_sdk.integrations.loggingr Zsentry_sdk._typesr �typingr r r r r Zsentry_sdk.clientr r r r r r r! r r1 r# rO r? rC r- r- r- r. �<module> s8 0!