Edit File: celery.cpython-37.opt-1.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__Z 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 )�opZ 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 )rC r3 r )rC 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_headersr? r) � set_statusZstart_transaction�list)r+ r, rB �scopeZ transaction)rC r* r r �_inner� s0 z _wrap_tracer.<locals>._inner)r )r* rC rL r )rC 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, rO )rC r* r r rL � s z _wrap_task_call.<locals>._inner)r )r* rC rL r )rC 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-jobrO r r r r) Z fingerprint)r rA r) � issubclassr r$ )�event�hintrQ rR )r+ r, r* �uuidr r �event_processor� s z._make_event_processor.<locals>.event_processorr )r* rV r+ r, ZrequestrW r )r+ r, r* rV r rH � s rH 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 ) NrE ZabortedZinternal_error�throwsr F)�typeZhandled)Zclient_optionsZ mechanism)rU ) r r< r= r � isinstance� CELERY_CONTROL_FLOW_EXCEPTIONS� _set_status�hasattrrX �clientr �optionsZ capture_event)r* rO rB r^ rT rU r r r rP � s rP 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�spanrI )rB �statusrK 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, rB )� old_workloopr r �sentry_workloop s z+_patch_worker_exit.<locals>.sentry_workloop)Z billiard.poolrb Zworkloop)rb re r )rd r r4 s r4 )N)1Z __future__r rN 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% rH rP r\ r4 r r r r �<module> s< 3!7