PATH:
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
spark
/
__pycache__
� o 0iz � � � d dl Z d dlZd dlmZ d dlmZmZmZmZm Z d dl m Z e rd dl m Z d dl m Z d dlmZmZmZ G d� d e� Zd � Zd � Zd � Zy) � 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 � � e Zd ZdZed� � Zy)�SparkWorkerIntegration� spark_workerc �& � dd l m} t | _ y �Nr )�pyspark.daemon�daemon�_sentry_worker_main� worker_main)�original_daemons �Y/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/spark/spark_worker.py� setup_oncez!SparkWorkerIntegration.setup_once s � � 1�&9��#� N)�__name__� __module__� __qualname__� identifier� staticmethodr � r r r r s � � �J��:� �:r r c � � t j � }ddd�}t | � } | \ }}}g }t | � D ]B \ }}}|t t t fvs�|j t ||||j |� � �D |rD|j � t | � }dd|id�}t � t j ||�� y y )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_exceptionr= ! s� � � � "� "� $�F� �U�3�I�"�8�,�H� &��H�i�� �B� $8��#A� ��)�R� �J��2F�G� G� �I�I�1� �i��V�^�^�Y�� � $B� � � � � �'��1��!��2��?����� � ��T�2� r c �b �� ddl m� t j � } | j �fd�� }y )Nr )� TaskContextc � �� t � 5 t j � j t � }�j � }|�|� | cd d d � S | j di � j dt |j � � � | d j dt |j � � � | d j dt |j � � � | d j dt |j � � � |j r�d|j v rD| d j d|j d � | d j d|j d � d |j v r/| j d i � j d |j d � d d d � | S # 1 sw Y | S xY w) N�tags�stageId� partitionId� attemptNumber� taskAttemptId�sentry_app_name�app_name�application_id�sentry_application_idzcallSite.short�extra�callSite) r r+ r, �get_integrationr �get� setdefault�strrB rC rD rE �_localProperties)r< r* � integration� task_contextr? s �r � process_eventz(_tag_task_context.<locals>.process_eventE s� �� � )� *�$�/�/�1�A�A�&��K� '�?�?�,�L��"�l�&:� � +� *� � � �V�R� (� 3� 3��3�|� 3� 3� 5�6� � �&�M� $� $�]�C� �8P�8P�8R�4S� T� �&�M� $� $�_�c�,�:T�:T�:V�6W� X� �&�M� $� $�_�c�,�:T�:T�:V�6W� X��,�,�$� �(E�(E�E��&�M�,�,�"�L�$A�$A�BS�$T�� �&�M�,�,�(�$�5�5�6M�N�� $�|�'D�'D�D��$�$�W�b�1�<�<�"�L�$A�$A�BR�$S��7 +�> � �? +�> � �s �=F5�EF5�5F?)�pyspark.taskcontextr? r+ �get_isolation_scope�add_event_processor)�scoperS r? s @r r3 r3 ? s, �� �/� � *� *� ,�E� � � �!� �!r c �$ � dd l m} |j | i |�� y # t $ rk t j � j t � �At j � }t � 5 t |� d d d � Y y # 1 sw Y Y y xY wY y w xY wr ) �pyspark.worker�worker�mainr- r+ r, rL r �sysr5 r r= )�args�kwargs�original_workerr5 s r r r j st � �,�-� �� � �d�-�f�-�� � -� � � � "� 2� 2�3I� J� V��|�|�~�H�,�.�"�8�,� /�.�.�� W�-�s. � �AB�) A?�5B�?B �B� B�B)r\ r+ �sentry_sdk.integrationsr �sentry_sdk.utilsr r r r r �typingr r r �sentry_sdk._typesr r r r r= r3 r r! r r �<module>rd sH �� � � /�� � !��� �6�6�:�[� :�3�<(�V -r
[+]
..
[-] spark_driver.cpython-312.pyc
[edit]
[-] __init__.cpython-312.pyc
[edit]
[-] spark_worker.cpython-312.pyc
[edit]