관리-도구
편집 파일: rq.cpython-38.pyc
U �Mg� � @ s8 d dl Z d dlZd dlmZ d dlmZ d dlmZmZ d dl m Z d dl m Z d dl mZmZmZmZmZ z@d dlmZ d d lmZ d d lmZ d d lmZ d d l m Z W n e k r� ed ��Y nX d dl m Z e �rd dl m!Z!m"Z" d dl#m$Z$m%Z% d dl m&Z& d dl m'Z' G dd� de�Z(dd� Z)dd� Z*dS )� N)�OP)�continue_trace)� DidNotEnable� Integration)� ignore_logger)�TRANSACTION_SOURCE_TASK)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�format_timestamp� parse_version)�Queue)�JobTimeoutException)�VERSION)�Worker)� JobStatuszRQ not installed)� TYPE_CHECKING)�Any�Callable)�Event�EventProcessor)�ExcInfo)�Jobc @ s& e Zd ZdZde� �Zedd� �ZdS )� RqIntegration�rqz auto.queue.c s� t t�} | d kr td�t���| dk r.td��tj�tt���fdd��}|t_tj��fdd�}|t_t j � tt� �� fdd ��}|t _ t d � d S ) NzUnparsable RQ version: {})r � z RQ 0.6 or newer is required.c s� t �� ��}|�� |�tt�|��� t|j� d�p6i t j dt t jd�}t� � |j|_W 5 Q R X t j|d|id�� � | |f|�|�}W 5 Q R X W 5 Q R X | jr�t �� �� |S )N�_sentry_trace_headerszunknown RQ task)�op�name�source�origin�rq_job)Zcustom_sampling_context)� sentry_sdkZ new_scopeZclear_breadcrumbsZadd_event_processor�_make_event_processor�weakref�refr �meta�getr Z QUEUE_TASK_RQr r r r � func_namer Zstart_transactionZis_horse� get_client�flush)�self�job�args�kwargs�scopeZ transaction�rv)�old_perform_job� �H/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/rq.py�sentry_patched_perform_job7 s( ��& z<RqIntegration.setup_once.<locals>.sentry_patched_perform_jobc s, |j tjks|jrt|� � | |f|�|�S )N)Z_statusr ZFAILEDZ is_failed�_capture_exception)r+ r, �exc_infor. )�old_handle_exceptionr2 r3 � sentry_patched_handle_exception[ s zARqIntegration.setup_once.<locals>.sentry_patched_handle_exceptionc s2 t �� }|jd k r$t|�� �|jd< � | |f|�S )Nr )r" Zget_current_scope�span�dictZ iter_trace_propagation_headersr& )r+ r, r. r/ )�old_enqueue_jobr2 r3 �sentry_patched_enqueue_jobh s � z<RqIntegration.setup_once.<locals>.sentry_patched_enqueue_jobz rq.worker) r � RQ_VERSIONr �formatr Z perform_jobr r Zhandle_exceptionr Z enqueue_jobr )�versionr4 r8 r< r2 )r; r7 r1 r3 � setup_once) s" zRqIntegration.setup_onceN)�__name__� __module__� __qualname__� identifierr � staticmethodr@ r2 r2 r2 r3 r % s r c s � fdd�}|S )Nc s� � � }|d k rvt � �\ | �di �}|j|j|j|j|jd�}|jrPt|j�|d< |j rdt|j �|d< ||d< W 5 Q R X d|kr�t � �( t |d d t �r�dd |jg| d < W 5 Q R X | S ) N�extra)Zjob_id�funcr- r. � description� enqueued_at� started_atzrq-jobr6 r r r � fingerprint) r � setdefault�idr( r- r. rH rI r rJ � issubclassr )�event�hintr, rF r! ��weak_jobr2 r3 �event_processorz s( �z._make_event_processor.<locals>.event_processorr2 )rR rS r2 rQ r3 r# x s r# c K s4 t �� }t| |jddd�d�\}}t j||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)rP )r" r) r �optionsZ capture_event)r6 r. �clientrO rP r2 r2 r3 r5 � s � r5 )+r$ r" Zsentry_sdk.constsr Zsentry_sdk.apir Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.loggingr Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r r Zrq.queuer Z rq.timeoutsr Z rq.versionr r= Z rq.workerr Zrq.jobr � ImportError�typingr r r Zsentry_sdk._typesr r r r r r# r5 r2 r2 r2 r3 �<module> s0 S!