관리-도구
편집 파일: beam.cpython-38.pyc
U �Mg> � @ s d dl Z d dlZd dlmZ d dlZd dlmZ d dlmZ d dl m Z m Z m Z m Z d dlmZ er�d dlmZ d dlmZ d d lmZ d d lmZ d d lmZ ed �Zed edef d�ZdZdZdZG dd� de�Zdd� Z dd� Z e e�dd� �Z dd � Z d d � Z dS ) � 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_{}Z _sentry_used_c @ s e Zd ZdZedd� �ZdS )�BeamIntegration�beamc sl ddl m} m} td� td� ddddg� � D ] }t| t�|�t| |�� q0|j�� �fd d �}||_d S ) Nr )�DoFn�ParDo�rootzbundle_processor.create�processZ start_bundleZ finish_bundle�setupc s� t | dd�s�� D ]h}t||�s qt�|�}t ||�}t |t�|��}t |td�st |td�st|||� t||t|�� qd| _�| |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�Zfunction_patchesZold_init� �J/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/beam.py�sentry_init_pardo7 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+ � setup_once% s �zBeamIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr0 r* r* r* r+ r " s r c s* t | � �sd S � fdd�}t|td� |S )Nc s� t �� �}t| |�r t| |�}n&t| � �}t| � t|�� t| ||� zddlm} ||�W S tk r� ddl m } ||� Y S X dS )zG Inspect function overrides the way Beam gets argspec. r )�get_function_args_defaults)�getfullargspecN) r r r r r r! r- r6 � ImportErrorZ apache_beam.typehints.decoratorsr7 )r" r' r( r6 r7 �r& r* r+ �_inspectX s z$_wrap_inspect_call.<locals>._inspectT)r r r )�clsr& r: r* r9 r+ r. R s r. c s$ t � �� fdd��}t|td� |S )z< Wrap task call with a try catch to get exceptions. c sB z� | |�}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��funcr* r+ �_innerz s z _wrap_task_call.<locals>._innerT)r r r )rD rE r* rC r+ r! t s r! c C s4 t �� }t| |jddd�d�\}}t j||d� dS )z( Send Beam exception to Sentry. r F)�typeZhandled)Zclient_optionsZ mechanism)�hintN)� sentry_sdkZ get_clientr �optionsZ capture_event)�exc_info�client�eventrG r* r* r+ �_capture_exception� s � rM c C s. t �� } t� � t| � W 5 Q R X t| � dS )z Raise an exception. N)�sysrJ r rM r )rJ r* r* r+ r= � s r= c c sF zt | �V W q tk r& Y qBY q tk r> t� Y q X q dS )z4 Wrap the generator to handle any failures. N)�next� StopIterationr< r= )rB r* r* r+ rA � s rA )!rN r? � functoolsr rH Zsentry_sdk.integrationsr Z sentry_sdk.integrations.loggingr Zsentry_sdk.utilsr r r r �typingr r r r r Zsentry_sdk._typesr r r r r r r r. r! rM r= rA r* r* r* r+ �<module> s2 0"