"С( 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_exceptionsevent_from_exception) Integration)
ignore_logger)
TYPE_CHECKING)Any)Iterator)TypeVar)Optional)Callable)Client)ExcInfoTF.)boundz _wrapped_{}_z _inspect_{}
_sentry_used_c ( e Zd ZdZed ZdS )BeamIntegrationbeamc ddl m} m} t d t d g dD ]9}t | t
| t | | :|j fd}||_ d S )Nr )DoFnParDorootzbundle_processor.create)process start_bundle
finish_bundlesetupc t | dd sʉD ]}t || st | }t || }t |t | }t |t
d sEt |t
d s/t
||| t
||t | d| _ | |g|R i | d S )N_sentry_is_patchedFT) getattrhasattr WRAPPED_FUNCformat INSPECT_FUNC USED_FUNCsetattr_wrap_task_callr# )
selffnargskwargs func_name wrapped_func process_func inspect_funcfunction_patchesold_inits
n/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/sentry_sdk/integrations/beam.pysentry_init_pardoz5BeamIntegration.setup_once..sentry_init_pardo6 s 4!5u==
/!1 N NI"2y11 ! #/#6#6y#A#AL $+2y#9#9L#*2|/B/B9/M/M#N#NL"<EBB N7$iL L N L,??? I|/L/LMMM*.' HT2 / / / / / / / / / ) 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 /000PPP) I ##I.."433
> 0 0 0 0 0 00 +r8 N)__name__
__module__ __qualname__
identifier staticmethodr< r8 r6 r r ! s2 J)+ )+ \)+ )+ )+r8 r c b t | sd S fd}t |t d |S )Nc ^ t }t | | rt | | }n?t | }t | t | t | || ddlm} || S # t $ r ddl m
} || cY S w xY w)zG
Inspect function overrides the way Beam gets argspec.
r )get_function_args_defaults)getfullargspec) r& r' r% r$ r* r+ r9 rE ImportError apache_beam.typehints.decoratorsrF )r, r1 r2 rE rF r0 s r6 _inspectz$_wrap_inspect_call.._inspectW s
$**955 4 & & 6"4 66LL"433L D)_\%B%B C C C D, 5 5 5 0 N N N N N N--l;; ; 0 0 0 G G G G G G!>,// / / / 0s =B B,+B,T)r% r* r) )clsr0 rI s ` r6 r: r: Q sN 3 " " t0 0 0 0 00 Hi&&& 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) Exceptionraise_exception
isinstancetypes
GeneratorType_wrap_generator_call)r. r/ genclientfuncs r6 _innerz _wrap_task_call.._inner{ sy $$'''CC $ $ $ F # # # # # $ #u 233 J#C000s ((T)r currentrT r r* r) )rU rV rT s ` @r6 r+ r+ s sW [
F
4[[ 1 1 1 1 1 [ 1 FIt$$$ Mr8 c | t }|dS |j }|dS t | |j ddd \ }}| || dS )z(
Send Beam exception to Sentry.
Nr F)typehandled)client_options mechanism)hint)get_integrationr rT r options
capture_event)exc_infohub integrationrT eventr] s r6 _capture_exceptionre s
%%o66K
ZF
~& ~!e44 KE4
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_clientsysra r re r )rT rb ra s r6 rN rN s
+C
z |~~H $ & & * *8S)))* * * * * * * * * * * * * * * Xs
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))OOOO EE $ $ $ 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 rv s & & & & & &
' ' ' ' ' ' & & & & & & N N N N N N N N / / / / / / 9 9 9 9 9 9 + + + + + + / (((((()))))) A8CH-...A -+ -+ -+ -+ -+k -+ -+ -+` D 0( ( (* $ $ $ $ $r8