ghddlmZddlZddlmZddlmZddlmZddl m Z m Z m Z m Z mZddlmZerddlmZdd lmZdd lmZmZmZGd d eZd ZdZdZdS))absolute_importN)configure_scope)Hub) Integration)capture_internal_exceptionsexc_info_from_error!single_exception_from_error_tuplewalk_exception_chainevent_hint_with_exc_info)MYPY)Any)Optional)ExcInfoEventHintc(eZdZdZedZdS)SparkWorkerIntegration spark_workerc*ddlm}t|_dSNr)pyspark.daemondaemon_sentry_worker_main worker_main)original_daemons /builddir/build/BUILD/imunify360-venv-2.6.0/opt/imunify360/venv/lib/python3.11/site-packages/sentry_sdk/integrations/spark/spark_worker.py setup_oncez!SparkWorkerIntegration.setup_onces$ 100000&9###N)__name__ __module__ __qualname__ identifier staticmethodrrrrrs2J::\:::rrc |j}|j}ddd}t|}|\}}}g}t|D]B\}}}|tt t fvr&|t|||||C|rQ| t|} dd|id} t| | | dSdS)NsparkF)typehandlederrorvalues)level exception)hint) clientoptionsrr SystemExitEOFErrorConnectionResetErrorappendr reverser _tag_task_context capture_event) exc_infohubr.client_options mechanismexc_type exc_valuetbrvr-events r_capture_exceptionr@$s ZF^N U33I"8,,H&Hi B$8#A#A)R J2FG G G II1i^Y     , '11!2?? %d+++++,,rcddlmt5}|jfd}ddddS#1swxYwYdS)Nr) TaskContextct5tjt}}|||cdddS|didt||ddt| |ddt| |ddt| |j rd|j vrN|dd|j d|dd|j d d |j vr5|d id |j d dddn #1swxYwY|S) NtagsstageId partitionId attemptNumber taskAttemptIdsentry_app_nameapp_nameapplication_idsentry_application_idzcallSite.shortextracallSite) rrcurrentget_integrationrget setdefaultstrrErFrGrH_localProperties)r?r- integration task_contextrBs r process_eventz(_tag_task_context..process_eventJs`-..  !k99:PQQ *00 &,*>           ,,77s<#7#7#9#9::f ((LG=FG==HH)pyspark.taskcontextrBradd_event_processor)scoperWrBs @rr5r5Ds//////   &e  "# # # #  # "# &&&&&&&&&&&&&&&&&&s 6::cLddlm} |j|i|dS#t$rtjtZtj}tj }t5t||dddYdS#1swxYwYYdSYdSwxYwr) pyspark.workerworkermainr0rrOrPrsysr7rr@)argskwargsoriginal_workerr8r7s rrrqs,,,,,,2d-f----- 222 ; & &'= > > J+C|~~H,.. 2 2"8S111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 K J J2s4 AB#2B B#B B#B B#"B#) __future__rr_ sentry_sdkrsentry_sdk.hubrsentry_sdk.integrationsrsentry_sdk.utilsrrr r r sentry_sdk._typesr typingr rrrrrr@r5rr$rrrjs]&&&&&& &&&&&&//////#"""""76666666666:::::[:::,,,@***Z 2 2 2 2 2r