Edit File: celery.cpython-37.pyc
B Z~�_3% � @ s| d dl mZ d dlZd dlmZ d dlmZmZ d dlm Z d dl m Z d dl m Z mZ d dlmZ d d lmZ d d lmZ er�d d lmZ d d lmZ d d lmZ d dlmZ d dlmZmZm Z m Z ededef d�Z y(d dl m Z! d dl"m#Z#m$Z$m%Z%m&Z& W n e'k �r" ed��Y nX e$e%e&fZ(G dd� de �Z)dd� Z*dd� Z+d d � Z,d&d d �Z-d d!� Z.d"d#� Z/d$d%� Z0dS )'� )�absolute_importN)�Hub)�capture_internal_exceptions�event_from_exception)� Transaction)�reraise)� Integration� DidNotEnable)� ignore_logger)�MYPY)�wraps)�Any)�TypeVar)�Callable)�Optional)�EventProcessor�Event�Hint�ExcInfo�F.)Zbound)�VERSION)�SoftTimeLimitExceeded�Retry�Ignore�RejectzCelery not installedc @ s& e Zd ZdZddd�Zedd� �ZdS ) �CeleryIntegration�celeryTc C s || _ d S )N)�propagate_traces)�selfr � r �O/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/celery.py�__init__+ s zCeleryIntegration.__init__c st t dk rtd��dd lm m} | j� � fdd�}|| _ddlm} t|j �|_ t � t d� t d� t d � d S ) N)� zCelery 3 or newer required.r c sF t |dd�s.t||j�|_t||j�|_d|_t|� | |f|�|��S )N�_sentry_is_patchedFT)�getattr�_wrap_task_call�__call__�runr# � _wrap_tracer)�name�task�args�kwargs)�old_build_tracerr r �sentry_build_tracer9 s z9CeleryIntegration.setup_once.<locals>.sentry_build_tracer)�Taskzcelery.worker.jobzcelery.app.tracezcelery.redirected) �CELERY_VERSIONr Zcelery.app.traceZapp�traceZ build_tracerZcelery.app.taskr/ �_wrap_apply_async� apply_async�_patch_worker_exitr )r1 r. r/ r )r- r � setup_once/ s z CeleryIntegration.setup_onceN)T)�__name__� __module__� __qualname__� identifierr! � staticmethodr5 r r r r r ( s r c s t � �� fdd��}|S )Nc s� t j}|�t�}|d k r�|jr�|jd| d jd��` t� �H t|� � �}|r~|� d�pXi }|� |� |� di �� |� ||d< W d Q R X � | |�S Q R X n � | |�S d S )Nz celery.submitr )�op� description�headers) r �current�get_integrationr r Z start_spanr) r �dictZ iter_trace_propagation_headers�get�update� setdefault)r+ r, �hubZ integrationr= Z kwarg_headers)�fr r r3 ] s z&_wrap_apply_async.<locals>.apply_async)r )rE r3 r )rE r r2 [ s r2 c s t � �� �fdd��}|S )Nc s� t j}|�t�d kr � | |�S |�� ��}d|_|�� |�t�f| �|�� d }t � �6 t j | d � d�pli ddd�}�j |_ |�d� W d Q R X |d kr�� | |�S |j|d�j t| d �| d d �id �� � | |�S Q R X W d Q R X d S ) Nr r" r= z celery.taskzunknown celery task)r; r) �okZ celery_job� � )r* r+ r, )Zcustom_sampling_context)r r> r? r Z push_scope�_nameZclear_breadcrumbsZadd_event_processor�_make_event_processorr r Zcontinue_from_headersrA r) � set_statusZstart_transaction�list)r+ r, rD �scopeZ transaction)rE r* r r �_inner� s0 z _wrap_tracer.<locals>._inner)r )r* rE rN r )rE r* r r( | s +r( c s t � �� �fdd��}|S )Nc sP y � | |�S t k rJ t�� }t� � t�|� W d Q R X t|� Y nX d S )N)� Exception�sys�exc_infor �_capture_exceptionr )r+ r, rQ )rE r* r r rN � s z _wrap_task_call.<locals>._inner)r )r* rE rN r )rE r* r r% � s r% c s � ���fdd�}|S )Nc s� t � �8 | �di �}�|d< | �di �}�j� �d�|d< W d Q R X d|kr�t � �. t|d d t�r|dd t�d ��g| d < W d Q R X | S ) N�tagsZcelery_task_id�extra)Z task_namer+ r, z celery-jobrQ r r r r) Z fingerprint)r rC r) � issubclassr r$ )�event�hintrS rT )r+ r, r* �uuidr r �event_processor� s z._make_event_processor.<locals>.event_processorr )r* rX r+ r, �requestrY r )r+ r, r* rX r rJ � s rJ c C s� t j}|�t�d krd S t|d t�r4t|d� d S t|d� t| d�r\t|d | j�r\d S |j }t ||j ddd�d�\}}|j ||d � d S ) NrG ZabortedZinternal_error�throwsr F)�typeZhandled)Zclient_optionsZ mechanism)rW ) r r> r? r � isinstance� CELERY_CONTROL_FLOW_EXCEPTIONS� _set_status�hasattrr[ �clientr �optionsZ capture_event)r* rQ rD ra rV rW r r r rR � s rR c C s@ t � �0 | �� � }|jd k r(|j�|� W d Q R X W d Q R X d S )N)r Zconfigure_scope�spanrK )rD �statusrM r r r r_ s r_ c s( ddl m} | j� � fdd�}|| _d S )Nr )�Workerc s@ z � | |�S t � �" tj}|�t�d k r0|�� W d Q R X X d S )N)r r r> r? r �flush)r+ r, rD )� old_workloopr r �sentry_workloop s z+_patch_worker_exit.<locals>.sentry_workloop)Z billiard.poolre Zworkloop)re rh r )rg r r4 s r4 )N)1� __future__r rP Zsentry_sdk.hubr Zsentry_sdk.utilsr r Zsentry_sdk.tracingr Zsentry_sdk._compatr Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.loggingr Zsentry_sdk._typesr Zsentry_sdk._functoolsr �typingr r r r r r r r r r r r0 Zcelery.exceptionsr r r r � ImportErrorr^ r r2 r( r% rJ rR r_ r4 r r r r �<module> s< 3!7