관리-도구
편집 파일: rq.cpython-38.pyc
U ��\� � @ sx d dl mZ d dlZd dlmZ d dlmZ d dlmZm Z d dl m Z d dl m Z G dd � d e�Zd d � Zd d � ZdS )� )�absolute_importN)�Hub)� Integration)�capture_internal_exceptions�event_from_exception)�JobTimeoutException)�Workerc @ s e Zd ZdZedd� �ZdS )� RqIntegration�rqc s4 t j��fdd�} | t _t j� � fdd�}|t _d S )Nc s� t j}|�t�}|d kr*� | |f|�|�S |�� �4}|�� |�tt� |��� � | |f|�|�}W 5 Q R X | j r||j � � |S �N) r �current�get_integrationr Z push_scopeZclear_breadcrumbsZadd_event_processor�_make_event_processor�weakref�refZis_horse�client�flush)�self�job�args�kwargs�hubZ integrationZscope�rv)�old_perform_job� �K/opt/alt/python38/lib/python3.8/site-packages/sentry_sdk/integrations/rq.py�sentry_patched_perform_job s z<RqIntegration.setup_once.<locals>.sentry_patched_perform_jobc s t |� � | |f|�|�S r )�_capture_exception)r r �exc_infor )�old_handle_exceptionr r � sentry_patched_handle_exception9 s zARqIntegration.setup_once.<locals>.sentry_patched_handle_exception)r Z perform_jobZhandle_exception)r r r )r r r � setup_once s zRqIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr! r r r r r s r c s � fdd�}|S )Nc s� � � }|d k rft � � |j| d< W 5 Q R X t � �0 | �di �}|j|j|j|j|jd�|d< W 5 Q R X d|kr�t � �( t|d d t�r�dd|jg| d < W 5 Q R X | S ) NZ transaction�extra)Zjob_id�funcr r � descriptionzrq-jobr r r r Z fingerprint) r � func_name� setdefault�idr r r) � issubclassr )�event�hintr r' ��weak_jobr r �event_processorB s" �z._make_event_processor.<locals>.event_processorr )r1 r2 r r0 r r @ s r c K sF t j}|�t�d krd S t| |jjddd�d�\}}|j||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)r/ )r r r r r r �optionsZ capture_event)r r r r. r/ r r r r ] s � r )� __future__r r Zsentry_sdk.hubr Zsentry_sdk.integrationsr Zsentry_sdk.utilsr r Z rq.timeoutsr Z rq.workerr ZAny�Dict�CallableZJobZQueueZExcInfor r r r r r r �<module> s )