관리-도구
편집 파일: arq.cpython-38.pyc
U �Mg� � @ s� d dl Z d dlZd dlmZmZ d dlmZmZ d dlm Z d dl m Z d dl m Z mZ d dlmZmZmZmZmZmZ z<d dlZd dlmZ d d lm Z d d lm Z m Z m Z m Z W n e!k r� ed ��Y nX d d l"m#Z# e#�rDd d l"m$Z$m%Z%m&Z&m'Z' d dl(m)Z)m*Z*m+Z+m,Z, d dl-m.Z. d dl/m0Z0 d dl1m2Z2 d dlm3Z3 e e e fZ4G dd� de�Z5dd� Z6dd� Z7dd� Z8dd � Z9d d � Z:d d � Z;d!d"� Z<d#d$� Z=dS )%� N)�OP� SPANSTATUS)� DidNotEnable� Integration)� ignore_logger)�should_send_default_pii)� Transaction�TRANSACTION_SOURCE_TASK)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�SENSITIVE_DATA_SUBSTITUTE� parse_version�reraise)�VERSION)�ArqRedis)�JobExecutionFailed�Retry�RetryJob�WorkerzArq is not installed)� TYPE_CHECKING)�Any�Dict�Optional�Union)�EventProcessor�Event�ExcInfo�Hint)�CronJob)�Job)�WorkerCoroutine)�Functionc @ s& e Zd ZdZde� �Zedd� �ZdS )�ArqIntegration�arqz auto.queue.c C s� z&t tt�rtt�} ntjd d� } W n ttfk rB d } Y nX | d krZtd�t���| dk rjtd��t � t � t � t d� d S )N� zUnparsable arq version: {})r � zarq 0.23 or newer required.z arq.worker) � isinstance� ARQ_VERSION�strr �version� TypeError� ValueErrorr �format�patch_enqueue_job� patch_run_job�patch_create_workerr )r* � r1 �I/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/arq.py� setup_once- s zArqIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier�origin� staticmethodr3 r1 r1 r1 r2 r# ) s r# c s t j� � fdd�} | t _d S )Nc � sr t �� �t�}|d kr.� | |f|�|�I d H S t jtj|tjd��& � | |f|�|�I d H W 5 Q R � S Q R X d S )N)�op�namer8 )� sentry_sdk� get_client�get_integrationr# Z start_spanr ZQUEUE_SUBMIT_ARQr8 )�self�function�args�kwargs� integration�Zold_enqueue_jobr1 r2 �_sentry_enqueue_jobK s �z.patch_enqueue_job.<locals>._sentry_enqueue_job)r Z enqueue_job)rE r1 rD r2 r. G s r. c s t j� � fdd�} | t _d S )Nc � s� t �� �t�}|d kr(� | ||�I d H S t �� �f}d|_|�� tddtj t tj d�}t � |��, � | ||�I d H W 5 Q R � W 5 Q R � S Q R X W 5 Q R X d S )Nr$ zunknown arq task�ok)r; �statusr: �sourcer8 ) r< r= r> r# Zisolation_scope�_nameZclear_breadcrumbsr r ZQUEUE_TASK_ARQr r8 Zstart_transaction)r? �job_idZscorerC �scope� transaction�Z old_run_jobr1 r2 �_sentry_run_job] s � z&patch_run_job.<locals>._sentry_run_job)r Zrun_job)rN r1 rM r2 r/ Y s r/ c C sp t �� }|jd k r>| d tkr0|j�tj� d S |j�tj� t| t � � j t j dd�d�\}}t j ||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)�hint)r< �get_current_scoperL �ARQ_CONTROL_FLOW_EXCEPTIONSZ set_statusr ZABORTED�INTERNAL_ERRORr r= �optionsr# r7 Z capture_event)�exc_inforK �eventrP r1 r1 r2 �_capture_exceptionu s � rW c s � ��fdd�}|S )Nc s� t � �� t�� }|jd k r2�d |j_�d | d< | �di �}�d |d< �d dk|d< | �d i �}�d t� rv� ntt� r��nt�d d �|d < W 5 Q R X | S ) N�job_namerL �tagsrJ Z arq_task_idZjob_try� Zarq_task_retry�extra)�taskrA rB �retryzarq-job)r r< rQ rL r; � setdefaultr r )rV rP rK rY r[ �rA �ctxrB r1 r2 �event_processor� s � z._make_event_processor.<locals>.event_processorr1 )r` rA rB ra r1 r_ r2 �_make_event_processor� s rb c s � �fdd�}|S )Nc � s� t �� �t�}|d kr,� | f|�|�I d H S t �� �t| d�i�f|�|�� z� | f|�|�I d H }W n, tk r� t� � }t |� t |� Y nX |S )NrX ) r< r= r> r# Zget_isolation_scopeZadd_event_processorrb � Exception�sysrU rW r )r` rA rB rC �resultrU �� coroutiner; r1 r2 �_sentry_coroutine� s �z*_wrap_coroutine.<locals>._sentry_coroutiner1 )r; rg rh r1 rf r2 �_wrap_coroutine� s ri c s* t jj� tt� �� fdd��} | t j_d S )Nc s� | d }t |d�r$dd� |jD �|_t |d�r@dd� |jD �|_d|kr^dd� |d D �|d< d|kr|dd� |d D �|d< � | |�S ) Nr � functionsc S s g | ] }t |��qS r1 ��_get_arq_function��.0�funcr1 r1 r2 � <listcomp>� s zFpatch_create_worker.<locals>._sentry_create_worker.<locals>.<listcomp>� cron_jobsc S s g | ] }t |��qS r1 ��_get_arq_cron_job�rn �cron_jobr1 r1 r2 rp � s c S s g | ] }t |��qS r1 rk rm r1 r1 r2 rp � s c S s g | ] }t |��qS r1 rr rt r1 r1 r2 rp � s )�hasattrrj rq )rA rB Z settings_cls�Zold_create_workerr1 r2 �_sentry_create_worker� s$ � �� � z2patch_create_worker.<locals>._sentry_create_worker)r$ �workerZ create_workerr r# )rx r1 rw r2 r0 � s r0 c C s t j�| �}t|j|j�|_|S �N)r$ ry ro ri r; rg )ro Zarq_funcr1 r1 r2 rl � s rl c C s t | j| j�| _| S rz )ri r; rg )ru r1 r1 r2 rs � s rs )>rd r< Zsentry_sdk.constsr r Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.loggingr Zsentry_sdk.scoper Zsentry_sdk.tracingr r Zsentry_sdk.utilsr r r r r r Z arq.workerr$ Z arq.versionr r( Zarq.connectionsr r r r r � ImportError�typingr r r r r Zsentry_sdk._typesr r r r Zarq.cronr Zarq.jobsr Z arq.typingr! r" rR r# r. r/ rW rb ri r0 rl rs r1 r1 r1 r2 �<module> s@