
    ghz                         S SK r S SKrS SKJr  S SKJrJrJrJrJ	r	  S SK
Jr  \(       a  S SK
Jr  S SK
Jr  S SKJrJrJr   " S S	\5      rS
 rS rS rg)    N)Integration)capture_internal_exceptionsexc_info_from_error!single_exception_from_error_tuplewalk_exception_chainevent_hint_with_exc_info)TYPE_CHECKING)Any)Optional)ExcInfoEventHintc                   (    \ rS rSrSr\S 5       rSrg)SparkWorkerIntegration   spark_workerc                  &    SS K Jn   [        U l        g Nr   )pyspark.daemondaemon_sentry_worker_mainworker_main)original_daemons    m/home/kali/devsecops-assessor/venv/lib/python3.13/site-packages/sentry_sdk/integrations/spark/spark_worker.py
setup_once!SparkWorkerIntegration.setup_once   s     	1&9#     N)__name__
__module____qualname____firstlineno__
identifierstaticmethodr   __static_attributes__r   r   r   r   r      s    J: :r   r   c           
         [         R                  " 5       nSSS.n[        U 5      n U u  p4n/ n[        U 5       HC  u  p4nU[        [
        [        4;  d  M  UR                  [        X4XQR                  U5      5        ME     U(       aA  UR                  5         [        U 5      nSSU0S.n[        5         [         R                  " XS9  g g )NsparkF)typehandlederrorvalues)level	exception)hint)
sentry_sdk
get_clientr   r   
SystemExitEOFErrorConnectionResetErrorappendr   optionsreverser   _tag_task_contextcapture_event)	exc_infoclient	mechanismexc_type	exc_valuetbrvr.   events	            r   _capture_exceptionrA   !   s    ""$F U3I"8,H&H	B $8#ARJ2FGGII1^^Y $B 



'1!2?  2 
r   c                  f   ^ SSK Jm  [        R                  " 5       n U R                  U4S j5       ng )Nr   )TaskContextc                   > [        5          [        R                  " 5       R                  [        5      nTR                  5       nUb  Uc  U sS S S 5        $ U R                  S0 5      R                  S[        UR                  5       5      5        U S   R                  S[        UR                  5       5      5        U S   R                  S[        UR                  5       5      5        U S   R                  S[        UR                  5       5      5        UR                  (       a  SUR                  ;   aD  U S   R                  SUR                  S   5        U S   R                  SUR                  S	   5        S
UR                  ;   a/  U R                  S0 5      R                  SUR                  S
   5        S S S 5        U $ ! , (       d  f       U $ = f)NtagsstageIdpartitionIdattemptNumbertaskAttemptIdsentry_app_nameapp_nameapplication_idsentry_application_idzcallSite.shortextracallSite)r   r/   r0   get_integrationr   get
setdefaultstrrF   rG   rH   rI   _localProperties)r@   r.   integrationtask_contextrC   s       r   process_event(_tag_task_context.<locals>.process_eventE   s    )*$//1AA&K '??,L"l&: +* VR(333|3356 &M$$]C8P8P8R4ST&M$$_c,:T:T:V6WX&M$$_c,:T:T:V6WX,,$(E(EE&M,,"L$A$ABS$T &M,,($556MN
 $|'D'DD$$Wb1<<"L$A$ABR$S7 +> ? +*> s   A GE!G
G)pyspark.taskcontextrC   r/   get_isolation_scopeadd_event_processor)scoperW   rC   s     @r   r7   r7   ?   s,    /**,E
! !r   c                  6   SS K Jn   UR                  " U 0 UD6  g ! [         as    [        R
                  " 5       R                  [        5      bG  [        R                  " 5       n[        5          [        U5        S S S 5         g ! , (       d  f        g = f g f = fr   )pyspark.workerworkermainr1   r/   r0   rP   r   sysr9   r   rA   )argskwargsoriginal_workerr9   s       r   r   r   j   su    ,-d-f- -  "223IJV||~H,."8, /... W-s.    AB-B9B
B	BBB)ra   r/   sentry_sdk.integrationsr   sentry_sdk.utilsr   r   r   r   r   typingr	   r
   r   sentry_sdk._typesr   r   r   r   rA   r7   r   r   r   r   <module>ri      sH    
  /  !66:[ :3<(V
-r   