U
\ @ sx d dl mZ d dlZd dlmZ d dlmZ d dlmZm Z d dl
m Z d dl m
Z
G dd d eZd
d Zd d
ZdS ) )absolute_importN)Hub) Integration)capture_internal_exceptionsevent_from_exception)JobTimeoutException)Workerc @ s e Zd ZdZedd ZdS )
RqIntegrationrqc s4 t jfdd} | 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 currentget_integrationr Z
push_scopeZclear_breadcrumbsZadd_event_processor_make_event_processorweakrefrefZis_horseclientflush)selfjobargskwargshubZ integrationZscoperv)old_perform_job /rq.pysentry_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|krt ( t|d d trdd|jg| d < W 5 Q R X | S )
NZ transactionextra)Zjob_idfuncr r descriptionzrq-jobr r r
r Z fingerprint) r Z func_name
setdefaultidr r r(
issubclassr )eventhintr 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}|td krd S t| |jjdddd\}}|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 ZAnyZDictCallableZJobZQueueZExcInfor r r r r r r <module> s
)