OwlCyberSecurity - MANAGER
Edit File: celery.cpython-38.pyc
U ��\G � @ s� d dl mZ d dlZd dlmZmZ d dlmZ d dlm Z m Z d dl m Z d dl mZ d dlmZ G d d � d e�Zd d � Zd d� Zddd�Zdd� Zdd� ZdS )� )�absolute_importN)�SoftTimeLimitExceeded�Retry)�Hub)�capture_internal_exceptions�event_from_exception)�reraise)� Integration)� ignore_loggerc @ s e Zd ZdZedd� �ZdS )�CeleryIntegration�celeryc s< dd l m m} | j� � fdd�}|| _t� td� d S )Nr c s4 t ||j�|_t ||j�|_t|� | |f|�|��S �N)�_wrap_task_call�__call__�run� _wrap_tracer)�name�task�args�kwargs�Zold_build_tracer� �O/opt/alt/python38/lib/python3.8/site-packages/sentry_sdk/integrations/celery.py�sentry_build_tracer s z9CeleryIntegration.setup_once.<locals>.sentry_build_tracerzcelery.worker.job)Zcelery.app.traceZapp�traceZ build_tracer�_patch_worker_exitr )r r r r r � setup_once s z CeleryIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr r r r r r s r c s � �fdd�}|S )Nc sl t j}|�t�d kr � | |�S |�� �<}d|_|�� |�t�f| �|�� � | |�W 5 Q R � S Q R X d S )Nr ) r �current�get_integrationr Z push_scope�_nameZclear_breadcrumbsZadd_event_processor�_make_event_processor)r r �hubZscope��fr r r �_inner/ s z _wrap_tracer.<locals>._innerr �r r( r) r r' r r ( s r c s � �fdd�}|S )Nc sR z � | |�W S t k rL t�� }t� � t�|� W 5 Q R X t|� Y nX d S r )� Exception�sys�exc_infor �_capture_exceptionr )r r r- r' r r r) A s z _wrap_task_call.<locals>._innerr r* r r' r r > s r c s � ��fdd�}|S )Nc s� t � � �j| d< W 5 Q R X t � �$ | �di �}�j� �d�|d< W 5 Q R X d|kr�t � �. t|d d t�r�ddt�d ��g| d < W 5 Q R X | S ) NZ transaction�extra)Z task_namer r z celery-jobr- r r r r Z fingerprint)r r � setdefault� issubclassr �getattr)�event�hintr/ �r r r r r �event_processorN s � �z._make_event_processor.<locals>.event_processorr )r �uuidr r �requestr6 r r5 r r% M s r% c C sv t j}|�t�d krd S t|d t�r*d S t| d�rHt|d | j�rHd S t||j j ddd�d�\}}|j ||d� d S )N� �throwsr F)�typeZhandled)Zclient_optionsZ mechanism)r4 ) r r" r# r � isinstancer �hasattrr: r Zclient�optionsZ capture_event)r r- r& r3 r4 r r r r. h s � r. c s( ddl m} | j� � fdd�}|| _d S )Nr )�Workerc sD z� | |�W �S t � �" tj}|�t�d k r4|�� W 5 Q R X X d S r )r r r" r# r �flush)r r r& �Z old_workloopr r �sentry_workloop� s z+_patch_worker_exit.<locals>.sentry_workloop)Z billiard.poolr? Zworkloop)r? rB r rA r r { s r )N)� __future__r r, Zcelery.exceptionsr r Zsentry_sdk.hubr Zsentry_sdk.utilsr r Zsentry_sdk._compatr Zsentry_sdk.integrationsr Z sentry_sdk.integrations.loggingr r r r r% r. r r r r r �<module> s