OwlCyberSecurity - MANAGER
Edit File: huey.cpython-312.pyc
� d��gC � � � d dl Z d dlmZ d dlZd dlmZmZmZ d dlmZm Z d dl m Z m Z d dl mZ d dlmZmZmZ d dlmZmZmZmZmZ d d lmZ er-d d lmZm Z m Z m Z m Z d d l m!Z!m"Z"m#Z# d d lm$Z$ e d e def �� Z% d dl&m'Z'm(Z(m)Z)m*Z*m+Z+ d dl,m-Z-m.Z.m/Z/ e-e.e/fZ1 G d� de � Z2d� Z3d� Z4d� Z5d� Z6d� Z7y# e0$ r e d� �w xY w)� N)�datetime)�continue_trace� get_baggage�get_traceparent)�OP� SPANSTATUS)� DidNotEnable� Integration)�should_send_default_pii)�BAGGAGE_HEADER_NAME�SENTRY_TRACE_HEADER_NAME�TransactionSource)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�SENSITIVE_DATA_SUBSTITUTE�reraise)� TYPE_CHECKING)�Any�Callable�Optional�Union�TypeVar)�EventProcessor�Event�Hint)�ExcInfo�F.)�bound)�Huey�Result� ResultGroup�Task� PeriodicTask)�CancelExecution� RetryTask�TaskLockedExceptionzHuey is not installedc �* � e Zd ZdZde� �Zed� � Zy)�HueyIntegration�hueyz auto.queue.c �, � t � t � y �N)� patch_enqueue� patch_execute� � �K/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/huey.py� setup_oncezHueyIntegration.setup_once. s � � ���r0 N)�__name__� __module__� __qualname__� identifier�origin� staticmethodr2 r/ r0 r1 r) r) * s% � ��J��:�,� '�F��� �r0 r) c �l �� t j �t t �� �fd�� } | t _ y )Nc �4 �� t j t j |j t j �� 5 t |t � s+t t � t t � i|j d< �| |� cd d d � S # 1 sw Y y xY w)N)�op�namer7 �sentry_headers)� sentry_sdk� start_spanr �QUEUE_SUBMIT_HUEYr< r) r7 � isinstancer$ r r r r �kwargs)�self�task� old_enqueues �r1 �_sentry_enqueuez&patch_enqueue.<locals>._sentry_enqueue9 sv �� � � "� "��#�#����"�)�)� � �d�L�1� (�� �,�o�.?�1�� � � ,�-� �t�T�*� � � �s � AB�B)r �enqueuer r) )rF rE s @r1 r- r- 5 s. �� ��,�,�K� ���=�+� >�+�" #�D�Lr0 c � � � � fd�}|S )Nc � �� t � 5 | j di � }�j |d< �j �j kD |d< | j di � }�j t � r �j nt t � r �j nt �j xs d�j z d�|d< d d d � | S # 1 sw Y | S xY w)N�tags� huey_task_id�huey_task_retry�extrar )rD �argsrB �retryzhuey-job) r � setdefault�id�default_retries�retriesr< r rN r rB )�event�hintrJ rM rD s �r1 �event_processorz._make_event_processor.<locals>.event_processorP s� �� � )� *��#�#�F�B�/�D�#'�7�7�D� � �&*�&:�&:�T�\�\�&I�D�"� #��$�$�W�b�1�E� � � � /�0� �I�I�2� /�0� �K�K�2� �.�.�3�!�t�|�|�C� !�E�*� � +�( � �) +�( � �s �B3C � Cr/ )rD rV s ` r1 �_make_event_processorrW N s �� ��2 �r0 c � � t j � }| d t v r*|j j t j � y |j j t j � t | t j � j t j dd��� \ }}|j ||�� y )Nr F)�type�handled)�client_options� mechanism)rU )r> �get_current_scope� HUEY_CONTROL_FLOW_EXCEPTIONS� transaction� set_statusr �ABORTED�INTERNAL_ERRORr � get_client�optionsr) r6 � capture_event)�exc_info�scoperT rU s r1 �_capture_exceptionrh l s� � � � (� (� *�E���{�2�2� ���$�$�Z�%7�%7�8�� ��� � ��!:�!:�;�&��!�,�,�.�6�6�*�5�5�%�H��K�E�4� ����D��)r0 c �8 � � t t � � � fd�� }|S )Nc � �� �| i |��}|S # t $ r+ t j � }t |� t |� Y S w xY wr, )� Exception�sysrf rh r )rN rB �resultrf �funcs �r1 �_sentry_executez+_wrap_task_execute.<locals>._sentry_execute� sO �� � ��4�*�6�*�F� � �� � ��|�|�~�H� �x� (� �X� �� � �s � �0A� A)r r) )rn ro s ` r1 �_wrap_task_executerp } s& �� � ���6� � 7� � �r0 c �n �� t j �t t �� d�fd� � } | t _ y )Nc �� �� t j � 5 }t � 5 d|_ |j � |j t |� � d d d � |j j dd � }t |xs i |j t j t j t j �� }|j# t$ j& � t) |dd� s!t+ |j, � |_ d|_ t j0 |� 5 �| ||� cd d d � cd d d � S # 1 sw Y ��xY w# 1 sw Y nxY w d d d � y # 1 sw Y y xY w)Nr* r= )r< r; �sourcer7 �_sentry_is_patchedFT)r> �isolation_scoper �_name�clear_breadcrumbs�add_event_processorrW rB �popr r< r �QUEUE_TASK_HUEYr �TASKr) r7 r` r �OK�getattrrp �executert �start_transaction)rC rD � timestamprg r= r_ � old_executes �r1 ro z&patch_execute.<locals>._sentry_execute� s �� � � '� '� )�U�,�.� $�� ��'�'�)��)�)�*?��*E�F� /� "�[�[�_�_�-=�t� D�N�(� �$�"��Y�Y��%�%�(�-�-�&�-�-� �K� � "� "�:�=�=� 1��4�!5�u�=� 1�$�,�,� ?�� �*.��'��-�-�k�:�"�4��y�9� ;�:�+ *� )�.�.��( ;�:��:�+ *� )� )�s; � E �2D>�CE �! E �+ E �>E �E � E �E � E)r, )r �_executer r) )ro r� s @r1 r. r. � s. �� ��-�-�K� ���=�:� >�:�4 $�D�Mr0 )8rl r r> �sentry_sdk.apir r r �sentry_sdk.constsr r �sentry_sdk.integrationsr r �sentry_sdk.scoper �sentry_sdk.tracingr r r �sentry_sdk.utilsr r r r r �typingr r r r r r �sentry_sdk._typesr r r r r �huey.apir r! r" r# r$ �huey.exceptionsr% r& r' � ImportErrorr^ r) r- rW rh rp r. r/ r0 r1 �<module>r� s� �� � � � G� G� ,� =� 4�� � � � !��>�>�=�=�(���8�C��H�-�.�A�0�F�F�O�O� !0��<O� P� ��k� �#�2�<*�"�$ $��Y � 0� �.� /�/�0�s �=B5 �5C