Edit File: huey.cpython-311.pyc
� r�d\ � � � d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl m Z d dl m Z d d lmZmZ d d lmZmZ d d lmZmZmZ er3d d lmZmZmZm Z m Z d d lm Z m Z m Z d dlm!Z! e dedef �� � Z" d dl#m$Z$m%Z%m&Z&m'Z' d dl(m)Z)m*Z*m+Z+ n# e,$ r ed� � �w xY we)e*e+fZ- G d� de� � Z.d� Z/d� Z0d� Z1d� Z2d� Z3dS ) � )�absolute_importN)�datetime)�reraise)� TYPE_CHECKING)�Hub)�OP)�_should_send_default_pii)� DidNotEnable� Integration)� Transaction�TRANSACTION_SOURCE_TASK)�capture_internal_exceptions�event_from_exception�SENSITIVE_DATA_SUBSTITUTE)�Any�Callable�Optional�Union�TypeVar)�EventProcessor�Event�Hint)�ExcInfo�F.)�bound)�Huey�Result� ResultGroup�Task)�CancelExecution� RetryTask�TaskLockedExceptionzHuey is not installedc �( � e Zd ZdZed� � � ZdS )�HueyIntegration�hueyc �> � t � � t � � d S �N)� patch_enqueue� patch_execute� � �m/builddir/build/BUILD/cloudlinux-venv-1.0.2/venv/lib/python3.11/site-packages/sentry_sdk/integrations/huey.py� setup_oncezHueyIntegration.setup_once( s � � ���������r+ N)�__name__� __module__� __qualname__� identifier� staticmethodr- r* r+ r, r$ r$ % s2 � � � � � ��J��� � �\�� � r+ r$ c �B �� t j ��fd�} | t _ d S )Nc � �� t j }|� t � � � �| |� � S |� t j |j �� � 5 �| |� � cd d d � � S # 1 swxY w Y d S )N)�op� description)r �current�get_integrationr$ � start_spanr �QUEUE_SUBMIT_HUEY�name)�self�task�hub� old_enqueues �r, �_sentry_enqueuez&patch_enqueue.<locals>._sentry_enqueue3 s� �� ��k�� � � �� /� /� 7� �;�t�T�*�*� *� �^�^�r� 3���^� K� K� +� +� �;�t�T�*�*� +� +� +� +� +� +� +� +� +� +� +� +���� +� +� +� +� +� +s � A3�3A7�:A7)r �enqueue)r@ r? s @r, r( r( / s0 �� ��,�K�+� +� +� +� +� #�D�L�L�Lr+ c � � � � fd�}|S )Nc � �� t � � 5 | � di � � }�j |d< �j �j k |d< | � di � � }�j t � � r�j nt t � � r�j nt �j pd�j z d�|d< d d d � � n # 1 swxY w Y | S )N�tags� huey_task_id�huey_task_retry�extrar )r= �args�kwargs�retryzhuey-job) r � setdefault�id�default_retries�retriesr; r rH r rI )�event�hintrD rG r= s �r, �event_processorz._make_event_processor.<locals>.event_processorB s �� � )� *� *� � ��#�#�F�B�/�/�D�#'�7�D� � �&*�&:�T�\�&I�D�"� #��$�$�W�b�1�1�E� � �+�-�-�/�� � �.�+�-�-�/�$�+�+�.� �.�3�!�t�|�C� !� !�E�*� � � � � � � � � � � � ���� � � � � � s �B B8�8B<�?B<r* )r= rQ s ` r, �_make_event_processorrR @ s$ �� �� � � � �* �r+ c �T � t j }| d t v r!|j j � d� � d S |j j � d� � t | |j r |j j nd t j dd��� � \ }}|� ||�� � d S )Nr �aborted�internal_errorF)�type�handled)�client_options� mechanism)rP ) r r7 � HUEY_CONTROL_FLOW_EXCEPTIONS�scope� transaction� set_statusr �client�optionsr$ r1 � capture_event)�exc_infor> rO rP s r, �_capture_exceptionrb Z s� � � �+�C���{�2�2�2� � � �(�(��3�3�3����I��$�$�%5�6�6�6�&��-0�Z�A�s�z�)�)�T�*�5�%�H�H�� � �K�E�4� ���e�$��'�'�'�'�'r+ c � � � � fd�}|S )Nc �� �� t j }|� t � � � �| i |��S �| i |��}n;# t $ r. t j � � }t |� � t |� Y nw xY w|S r' ) r r7 r8 r$ � Exception�sysra rb r )rH rI r> �resultra �funcs �r, �_sentry_executez+_wrap_task_execute.<locals>._sentry_executem s� �� ��k�� � � �� /� /� 7��4��(��(�(� (� ��T�4�*�6�*�*�F�F��� � � ��|�~�~�H� �x� (� (� (� �X� � � � � ���� � s �: �5A2�1A2r* )rh ri s ` r, �_wrap_task_executerj k s$ �� � � � � � � �r+ c �D �� t j �d�fd� } | t _ d S )Nc �� �� t j }|� t � � � �| ||� � S |� � � 5 }t � � 5 d|_ |� � � |� t |� � � � d d d � � n # 1 swxY w Y t |j dt j t �� � }t |dd� � s t! |j � � |_ d|_ |� |� � 5 �| ||� � cd d d � � cd d d � � S # 1 swxY w Y d d d � � d S # 1 swxY w Y d S )Nr% �ok)r; �statusr5 �source�_sentry_is_patchedFT)r r7 r8 r$ � push_scoper �_name�clear_breadcrumbs�add_event_processorrR r r; r �QUEUE_TASK_HUEYr �getattrrj �executerp �start_transaction)r< r= � timestampr> r[ r\ � old_executes �r, ri z&patch_execute.<locals>._sentry_execute� sI �� ��k�� � � �� /� /� 7� �;�t�T�9�5�5� 5� �^�^� � � :��,�.�.� G� G� $�� ��'�'�)�)�)��)�)�*?��*E�*E�F�F�F� G� G� G� G� G� G� G� G� G� G� G���� G� G� G� G� &��Y���%�.� � � �K� �4�!5�u�=�=� /� 1�$�,� ?� ?�� �*.��'��&�&�{�3�3� :� :�"�{�4��y�9�9� :� :� :� :� :� :� :�# :� :� :� :� :� :� :� :�" :� :� :� :���� :� :� :� :� :�# :� :� :� :� :� :� :� :� :� :� :� :���� :� :� :� :� :� :s[ � E�>B"� E�"B& �&E�)B& �*A0E� E �' E� E �E�E �E�E � E r' )r �_execute)ri rz s @r, r) r) s5 �� ��-�K�:� :� :� :� :� :�6 $�D�M�M�Mr+ )4� __future__r rf r �sentry_sdk._compatr �sentry_sdk._typesr � sentry_sdkr �sentry_sdk.constsr �sentry_sdk.hubr �sentry_sdk.integrationsr r �sentry_sdk.tracingr r �sentry_sdk.utilsr r r �typingr r r r r r r r r r �huey.apir r r r �huey.exceptionsr r! r" � ImportErrorrZ r$ r( rR rb rj r) r* r+ r, �<module>r� st �� &� &� &� &� &� &� � � � � � � � � � � &� &� &� &� &� &� +� +� +� +� +� +� � � � � � � � � � � � � 3� 3� 3� 3� 3� 3� =� =� =� =� =� =� =� =� C� C� C� C� C� C� C� C�� � � � � � � � � � � /�>�>�>�>�>�>�>�>�>�>�>�>�>�>�=�=�=�=�=�=�=�=�=�=�(�(�(�(�(�(����8�C��H�-�.�.�.�A�0�8�8�8�8�8�8�8�8�8�8�8�8�O�O�O�O�O�O�O�O�O�O�O��� 0� 0� 0� �,�.� /� /�/�0���� !0��<O� P� �� � � � �k� � � �#� #� #�"� � �4(� (� (�"� � �( $� $� $� $� $s �?B �B'