Edit File: beam.cpython-37.opt-1.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_bundleZsetupc 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__Z identifier� staticmethodr2 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" r/ r8 � ImportError)r# r( r) r8 )r' r7 r, r- �_inspectX s z$_wrap_inspect_call.<locals>._inspectT)Z apache_beam.typehints.decoratorsr7 r r! r )�clsr' r: r, )r' r7 r- r0 Q s r0 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 �currentrC r r! r )rD rE r, )rC rD 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 rC r �optionsZ capture_event)�exc_info�hubZ integrationrC �eventrH r, r, r- �_capture_exception� s rM 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 rF rC Z bind_client�sysrJ r rM r )rC rK rJ 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= )rB rC r, r, r- rA � s rA )'Z __future__r rN r? 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 r0 r" rM r= rA r, r, r, r- �<module> s8 0!