\ @ s d d l m Z d d l Z d d l m Z d d l m Z d d l m Z m Z d d l
m Z d d l m
Z
Gd d d e Z d
d Z d d
Z d S) )absolute_importN)Hub) Integration)capture_internal_exceptionsevent_from_exception)JobTimeoutException)Workerc @ s( e Z d Z d Z e d d Z d S)
RqIntegrationrqc sL t j f d d } | t _ t j f d d } | t _ d S)Nc s t j } | j t } | d k r7 | | | | S| j B } | j | j t t j | | | | | } Wd QRX| j
r | j 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 | | | | | S)N)_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 f d d } | S)Nc s } | d k r t | j | d <Wd QRXt N | j d i } d | j d | j d | j d | j d | j i | d <Wd QRXd | k r t 2 t | d d
t r d d | j g | d
<Wd QRX| S)NZ transactionextraZjob_idfuncr r descriptionzrq-jobr r r
r Z fingerprint) r func_name
setdefaultidr r r'
issubclassr )eventhintr r% )weak_jobr r event_processorB s
z._make_event_processor.<locals>.event_processorr )r. r/ r )r. r r
@ s r
c K si t j } | j t d k r" d St | d | j j d d d d d i \ } } | j | d | d S)NZclient_optionsZ mechanismtyper
ZhandledFr- )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 r r
r r r r r <module> s
)