관리-도구
편집 파일: huey.cpython-38.pyc
U �MgJ � @ s� d dl Z d dlmZ d dlZd dlmZmZmZ d dlmZm Z d dl m Z m Z d dl mZ d dlmZmZmZ d dlmZmZmZmZmZ d d lmZ er�d d lmZm Z m Z m Z m Z d d l m!Z!m"Z"m#Z# d d lm$Z$ e d e def d�Z%z4d dl&m'Z'm(Z(m)Z)m*Z*m+Z+ d dl,m-Z-m.Z.m/Z/ W n e0k �r> e d��Y nX e-e.e/fZ1G dd� de �Z2dd� Z3dd� Z4dd� Z5dd � Z6d d � Z7dS ) � N)�datetime)�continue_trace� get_baggage�get_traceparent)�OP� SPANSTATUS)� DidNotEnable� Integration)�should_send_default_pii)�BAGGAGE_HEADER_NAME�SENTRY_TRACE_HEADER_NAME�TRANSACTION_SOURCE_TASK)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�SENSITIVE_DATA_SUBSTITUTE�reraise)� TYPE_CHECKING)�Any�Callable�Optional�Union�TypeVar)�EventProcessor�Event�Hint)�ExcInfo�F.)�bound)�Huey�Result� ResultGroup�Task� PeriodicTask)�CancelExecution� RetryTask�TaskLockedExceptionzHuey is not installedc @ s& e Zd ZdZde� �Zedd� �ZdS )�HueyIntegration�hueyz auto.queue.c C s t � t� d S �N)� patch_enqueue� patch_execute� r, r, �J/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/huey.py� setup_once. s zHueyIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier�origin� staticmethodr. r, r, r, r- r' * s r' c s& t j� tt� �� fdd��} | t _d S )Nc sX t jtj|jtjd��8 t|t�s8t t � t t � i|j d< � | |�W 5 Q R � S Q R X d S )N)�op�namer3 �sentry_headers)� sentry_sdkZ start_spanr ZQUEUE_SUBMIT_HUEYr6 r' r3 � isinstancer# r r r r �kwargs)�self�task�Z old_enqueuer, r- �_sentry_enqueue9 s � � z&patch_enqueue.<locals>._sentry_enqueue)r �enqueuer r' )r> r, r= r- r* 5 s r* c s � fdd�}|S )Nc s� t � �p | �di �}� j|d< � j� jk|d< | �di �}� jt� rJ� jntt� rX� j nt� jpbd� j d�|d< W 5 Q R X | S )N�tagsZ huey_task_idZhuey_task_retry�extrar )r< �argsr: �retryzhuey-job) r � setdefault�idZdefault_retries�retriesr6 r rB r r: )�event�hintr@ rA �r< r, r- �event_processorP s ���z._make_event_processor.<locals>.event_processorr, )r< rJ r, rI r- �_make_event_processorN s rK c C sf t �� }| d tkr&|j�tj� d S |j�tj� t| t � � j t j dd�d�\}}|j ||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)rH )r8 Zget_current_scope� HUEY_CONTROL_FLOW_EXCEPTIONS� transaction� set_statusr ZABORTED�INTERNAL_ERRORr Z get_client�optionsr' r2 Z capture_event)�exc_info�scoperG rH r, r, r- �_capture_exceptionl s � rT c s t t� �� fdd��}|S )Nc s@ z� | |�}W n, t k r: t�� }t|� t|� Y nX |S r) )� Exception�sysrR rT r )rB r: �resultrR ��funcr, r- �_sentry_execute� s z+_wrap_task_execute.<locals>._sentry_execute)r r' )rY rZ r, rX r- �_wrap_task_execute} s r[ c s( t j� tt� �d� fdd� �} | t _d S )Nc s� t �� ��}t� �" d|_|�� |�t|�� W 5 Q R X |j�dd �}t |pNi |j t j t tjd�}|�tj� t|dd�s�t|j�|_d|_t �|��& � | ||�W 5 Q R � W 5 Q R � S Q R X W 5 Q R X d S )Nr( r7 )r6 r5 �sourcer3 �_sentry_is_patchedFT)r8 Zisolation_scoper �_nameZclear_breadcrumbsZadd_event_processorrK r: �popr r6 r ZQUEUE_TASK_HUEYr r' r3 rO r �OK�getattrr[ �executer] Zstart_transaction)r; r< � timestamprS r7 rN �Z old_executer, r- rZ � s& � z&patch_execute.<locals>._sentry_execute)N)r Z_executer r' )rZ r, rd r- r+ � s r+ )8rV r r8 Zsentry_sdk.apir r r Zsentry_sdk.constsr r Zsentry_sdk.integrationsr r Zsentry_sdk.scoper Zsentry_sdk.tracingr r r Zsentry_sdk.utilsr r r r r �typingr r r r r r Zsentry_sdk._typesr r r r r Zhuey.apir r r! r" r# Zhuey.exceptionsr$ r% r& � ImportErrorrM r' r* rK rT r[ r+ r, r, r, r- �<module> s4