관리-도구
편집 파일: rq.cpython-37.pyc
B T aZ � @ s d dl mZ d dlZd dlmZ d dlmZmZ d dlm Z d dl m Z d dl m Z mZ y4d dlmZ d d lmZ d d lmZ d d lmZ W n ek r� ed ��Y nX d d lmZ er�d dlm Z m Z m Z d dlm Z d dl m Z d dl!m"Z" G dd� de�Z#dd� Z$dd� Z%dS )� )�absolute_importN)�Hub)� DidNotEnable� Integration)� ignore_logger)� Transaction)�capture_internal_exceptions�event_from_exception)�Queue)�JobTimeoutException)�VERSION)�WorkerzRQ not installed)�MYPY)�Any�Callable�Dict)�EventProcessor)�ExcInfo)�Jobc @ 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 _ td � 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_job/ s* &z<RqIntegration.setup_once.<locals>.sentry_patched_perform_jobc s |j rt|� � | |f|�|�S )N)Z is_failed�_capture_exception)r( r) �exc_infor+ )�old_handle_exceptionr/ r0 � sentry_patched_handle_exceptionX 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/ r0 �sentry_patched_enqueue_jobc s z<RqIntegration.setup_once.<locals>.sentry_patched_enqueue_jobz rq.worker)�tuple�map�int� RQ_VERSION�split� ValueError� TypeErrorr �formatr Z perform_jobZhandle_exceptionr Z enqueue_jobr )�versionr1 r5 r8 r/ )r7 r4 r. r0 � setup_once! s % zRqIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodrB r/ r/ r/ r0 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-jobr3 r r r Z fingerprint) r � setdefault�idr& r* r+ rJ � issubclassr )�event�hintr) rH )�weak_jobr/ r0 �event_processort s z._make_event_processor.<locals>.event_processorr/ )rP rQ r/ )rP r0 r! r 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)rO )r r r r r r �optionsZ capture_event)r3 r+ r, r rN rO r/ r/ r0 r2 � s r2 )&� __future__r r" Zsentry_sdk.hubr Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.loggingr Zsentry_sdk.tracingr Zsentry_sdk.utilsr r Zrq.queuer Z rq.timeoutsr Z rq.versionr r< Z rq.workerr � ImportErrorZsentry_sdk._typesr �typingr r r r r Zrq.jobr r r! r2 r/ r/ r/ r0 �<module> s, T