Edit File: rq.cpython-37.pyc
B Z~�_ � @ s d dl mZ d dlZd dlmZ d dlmZmZ d dlm Z d dl m Z m Z y4d dl mZ d dlmZ d d lmZ d d lmZ W n ek r� ed ��Y nX d d lmZ er�d d lmZ d dlmZ d dlm Z d dl m Z d dl m Z d dlm Z G dd� de�Z!dd� Z"dd� Z#dS )� )�absolute_importN)�Hub)� Integration� DidNotEnable)� Transaction)�capture_internal_exceptions�event_from_exception)�VERSION)�JobTimeoutException)�Worker)�QueuezRQ not installed)�MYPY)�Any)�Dict)�Callable)�Job)�ExcInfo)�EventProcessorc @ s e Zd ZdZedd� �ZdS )� RqIntegration�rqc s� y t ttt�d�d d� ��} W n& ttfk rF td�t���Y nX | dk rXtd��t j ��fdd�}|t _ t j ��fdd �}|t _ t j � � fd d �}|t _ d S ) N�.� zUnparsable RQ version: {})r � z RQ 0.6 or newer is required.c s� t j}|�t�}|d kr*� | |f|�|�S |j}|d k s<t�|�� ��}|�� |�t t � |��� t j |j�d�pti ddd�}t� � |j|_W d Q R X |j|d|id�� � | |f|�|�} W d Q R X W d Q R X | jr�|�� | S )N�_sentry_trace_headerszrq.taskzunknown RQ task)�op�nameZrq_job)Zcustom_sampling_context)r �current�get_integrationr �client�AssertionErrorZ push_scopeZclear_breadcrumbsZadd_event_processor�_make_event_processor�weakref�refr Zcontinue_from_headers�meta�getr � func_namer Zstart_transactionZis_horse�flush) �self�job�args�kwargs�hubZ integrationr ZscopeZ transaction�rv)�old_perform_job� �K/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/rq.py�sentry_patched_perform_job1 s* &z<RqIntegration.setup_once.<locals>.sentry_patched_perform_jobc s t |� � | |f|�|�S )N)�_capture_exception)r' r( �exc_infor* )�old_handle_exceptionr. r/ � sentry_patched_handle_exceptionZ s zARqIntegration.setup_once.<locals>.sentry_patched_handle_exceptionc s4 t j}|�t�d k r&t|�� �|jd< � | |f|�S )Nr )r r r r �dictZ iter_trace_propagation_headersr# )r' r( r* r+ )�old_enqueue_jobr. r/ �sentry_patched_enqueue_jobc s z<RqIntegration.setup_once.<locals>.sentry_patched_enqueue_job)�tuple�map�int� RQ_VERSION�split� ValueError� TypeErrorr �formatr Z perform_jobZhandle_exceptionr Z enqueue_job)�versionr0 r4 r7 r. )r6 r3 r- r/ � setup_once# s % zRqIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodrA r. r. r. r/ r s r c s � fdd�}|S )Nc s� � � }|d k rJt � �0 | �di �}|j|j|j|j|jd�|d< W d Q R X d|kr�t � �( t|d d t�r|dd|jg| d< W d Q R X | S ) N�extra)Zjob_id�funcr) r* � descriptionzrq-jobr2 r r r Z fingerprint) r � setdefault�idr% r) r* rI � issubclassr )�event�hintr( rG )�weak_jobr. r/ �event_processorr s z._make_event_processor.<locals>.event_processorr. )rO rP r. )rO r/ r p s r c K sJ t j}|�t�d krd S |j}t| |jddd�d�\}}|j||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)rN )r r r r r r �optionsZ capture_event)r2 r* r+ r rM rN r. r. r/ r1 � s r1 )$� __future__r r! Zsentry_sdk.hubr Zsentry_sdk.integrationsr r Zsentry_sdk.tracingr Zsentry_sdk.utilsr r Z rq.versionr r; Z rq.timeoutsr Z rq.workerr Zrq.queuer � ImportErrorZsentry_sdk._typesr �typingr r r Zrq.jobr r r r r r1 r. r. r. r/ �<module> s. P