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� �/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__Z
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 Z func_name�
setdefault�idr r r( �
issubclassr )�event�hintr r& ��weak_jobr r �event_processorB s" �z._make_event_processor.<locals>.event_processorr )r/ r0 r r. 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 ZoptionsZ
capture_event)r r r r, r- r r r r ] s �
r )Z
__future__r r Zsentry_sdk.hubr Zsentry_sdk.integrationsr Zsentry_sdk.utilsr r Z rq.timeoutsr Z rq.workerr ZAnyZDict�CallableZJobZQueueZExcInfor r r r r r r �<module> s
)