�
h 0i� � �> � d dl Z d dlZd dlmZ d dlmZ d dlmZmZm Z d dl
m Z d dl m
Z
d dlmZmZmZmZmZ d dlmZ d d lmZ d d
lmZ d d lm Z d d l m Z d dl m!Z! e!r d dl m"Z"m#Z# d dl$m%Z%m&Z& d dlm'Z' d dl m(Z( G d� de � Z)d� Z*d� Z+y# e $ r ed
� �w xY w)� N)�OP)�continue_trace)�_check_minimum_version� DidNotEnable� Integration)�
ignore_logger)�TransactionSource)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�format_timestamp�
parse_version)�Queue)�JobTimeoutException)�VERSION)�Worker)� JobStatuszRQ not installed)�
TYPE_CHECKING)�Any�Callable)�Event�EventProcessor)�ExcInfo)�Jobc �* � e Zd ZdZde� �Zed� � Zy)�
RqIntegration�rqz auto.queue.c �j ���� t t � } t t | � t j
�t
t �� �fd�� }|t _ t j ��fd�}|t _ t j �t
t �� �fd�� }|t _ t d� y )Nc � �� t j � 5 }|j � |j t t j |� � � t |j j d� xs i t j dt j t j �� }t! � 5 |j" |_ d d d � t j& |d|i�� 5 �| |g|��i |��}d d d � d d d � | j( r"t j* � j- � S # 1 sw Y �pxY w# 1 sw Y �MxY w# 1 sw Y �QxY w)N�_sentry_trace_headerszunknown RQ task)�op�name�source�origin�rq_job)�custom_sampling_context)�
sentry_sdk� new_scope�clear_breadcrumbs�add_event_processor�_make_event_processor�weakref�refr �meta�getr �
QUEUE_TASK_RQr �TASKr r$ r
� func_namer" �start_transaction�is_horse�
get_client�flush)�self�job�args�kwargs�scope� transaction�rv�old_perform_jobs ��I/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/rq.py�sentry_patched_perform_jobz<RqIntegration.setup_once.<locals>.sentry_patched_perform_job1 s �� � �%�%�'�5��'�'�)��)�)�*?�� � �C�@P�*Q�R� ,��H�H�L�L�!8�9�?�R��'�'�*�,�1�1�(�/�/� � � 1�2�'*�}�}�K�$� 3� �1�1� �-5�s�O�� )��s�D�T�D�V�D�B� � (�* �}�}� �%�%�'�-�-�/��I� 3�2��� �� (�'�s< �B E�5D6�!E�(E�6E�6D? �;E�E �E�Ec �� �� t |d� xr |j xr |j dkD }|j t j k( xs |j
}|r
|s t
|� �| |g|��i |��S )N� retries_leftr )�hasattrrB �_statusr �FAILED� is_failed�_capture_exception)r7 r8 �exc_infor: �retry�failed�old_handle_exceptions �r? � sentry_patched_handle_exceptionzARqIntegration.setup_once.<locals>.sentry_patched_handle_exceptionU sx �� � ��^�,� )��$�$�)��$�$�q�(�
�
�[�[�I�$4�$4�4�E��
�
�F��e�"�8�,�'��c�G�H�G��G� G� c � �� t j � }|j �&t |j � � |j
d< �| |fi |��S )Nr )r' �get_current_scope�span�dict� iter_trace_propagation_headersr. )r7 r8 r: r; �old_enqueue_jobs �r? �sentry_patched_enqueue_jobz<RqIntegration.setup_once.<locals>.sentry_patched_enqueue_jobf sQ �� � �0�0�2�E��z�z�%�48��8�8�:�5����0�1� #�4��7��7� 7rM z rq.worker) r �
RQ_VERSIONr r r � perform_jobr �handle_exceptionr � enqueue_jobr )�versionr@ rL rT rS rK r> s @@@r? �
setup_oncezRqIntegration.setup_once) s� �� � �
�+�� �}�g�6� �,�,�� #�M�?� C� �
D� �> 8��� %� 6� 6� � H� #B�� � �+�+�� #�M�?� C� 8�
D� 8� 7����k�"rM N)�__name__�
__module__� __qualname__�
identifierr$ � staticmethodrZ � rM r? r r % s' � ��J��:�,�
'�F��I#� �I#rM r c � � � � fd�}|S )Nc �& �� �� }|��t � 5 | j di � }|j |j |j |j
|j d�}|j rt |j � |d< |j rt |j � |d< ||d< d d d � d|v r=t � 5 t |d d t � rdd |j g| d
< d d d � | S | S # 1 sw Y �LxY w# 1 sw Y | S xY w) N�extra)�job_id�funcr9 r: � description� enqueued_at�
started_atzrq-jobrH r r r � fingerprint) r
�
setdefault�idr2 r9 r: rf rg r
rh �
issubclassr )�event�hintr8 rc r% �weak_jobs �r? �event_processorz._make_event_processor.<locals>.event_processorx s� �� ��j�� �?�,�.� �(�(��"�5�� !�f�f� �M�M� �H�H� !�j�j�#&�?�?� �� �?�?�,<�S�_�_�,M�F�=�)��>�>�+;�C�N�N�+K�F�<�(�"(��h�� /�" �� �,�.� �d�:� .�q� 1�3F�G�,0�2G�� � �+W�E�-�(� /� � �u� �- /�.��$ /� � �s �BC:�)D�:D�Dr` )ro rp s ` r? r+ r+ v s �� ��8 �rM c � � t j � }t | |j ddd��� \ }}t j ||�� y )Nr F)�type�handled)�client_options� mechanism)rn )r' r5 r �options�
capture_event)rH r: �clientrm rn s r? rG rG � sB � �
�
"�
"�
$�F�&�� �~�~� �E�2��K�E�4� � � �U��.rM ),r, r' �sentry_sdk.constsr �sentry_sdk.apir �sentry_sdk.integrationsr r r � sentry_sdk.integrations.loggingr �sentry_sdk.tracingr �sentry_sdk.utilsr
r r r
r �rq.queuer � rq.timeoutsr �
rq.versionr rU � rq.workerr �rq.jobr � ImportError�typingr r r �sentry_sdk._typesr r r r r r+ rG r` rM r? �<module>r� s� �� � � � )� U� U� 9� 0�� �+� �/�0� � � !��$�7�(��N#�K� N#�b �B
/��A � +�
�)�
*�*�+�s � B �B