OwlCyberSecurity - MANAGER
Edit File: rq.cpython-312.pyc
� G��g� � �> � 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