관리-도구
편집 파일: beam.cpython-37.pyc
B T a � @ 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.)�boundz _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) r0 r. )r, r- r/ � setup_once$ s zBeamIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr4 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$ r1 r; � ImportError)r% r* r+ r; )r) r: r. r/ �_inspectX s z$_wrap_inspect_call.<locals>._inspectT)Z apache_beam.typehints.decoratorsr: r r# r" )�clsr) r= r. )r) r: r/ r2 Q s r2 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 �currentrF r r# r" )rG rH r. )rF rG 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 rF r �optionsZ capture_event)�exc_info�hubZ integrationrF �eventrK r. r. r/ �_capture_exception� s rP 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 rI rF Z bind_client�sysrM r rP r )rF rN rM 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@ )rE rF r. r. r/ rD � s rD )'� __future__r rQ rB 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 r2 r$ rP r@ rD r. r. r. r/ �<module> s8 0!