관리-도구
편집 파일: __init__.cpython-37.pyc
B ��Y � @ s� d Z ddlmZ ddlZddlmZ ddlmZmZm Z m Z ddl m Z G dd� dej �Zdd d �Zdejfd d �ZG dd� de�ZdS )z� raven.contrib.celery ~~~~~~~~~~~~~~~~~~~~ :copyright: (c) 2010-2012 by the Sentry Team, see AUTHORS for more details. :license: BSD, see LICENSE for more details. � )�absolute_importN)�SoftTimeLimitExceeded)�after_setup_logger� task_failure� task_prerun� task_postrun)� SentryHandlerc @ s e Zd Zdd� ZdS )� CeleryFilterc C s2 t |di �}t|t�s |jdkS |�d|jdk�S )N�dataZ _log_errorZinternal)�getattr� isinstance�dict�funcName�get)�self�record� extra_data� r �N/opt/alt/python37/lib/python3.7/site-packages/raven/contrib/celery/__init__.py�filter s zCeleryFilter.filterN)�__name__� __module__� __qualname__r r r r r r s r Fc C s t | |d��� d S )N)�ignore_expected)�SentryCeleryHandler�install)�clientr r r r �register_signal s r c sB t � � t| ����|� ��� � � �fdd�}tj|dd� d S )Nc s8 x(|j D ] }t|�tkr|�� � dS qW |��� d S )NF)�handlers�typer � addFilter� addHandler)�sender�logger�loglevelZlogfile�formatZcolorize�kw�h)�filter_�handlerr r �process_logger_event( s z4register_logger_signal.<locals>.process_logger_eventF)�weak)r r �setLevelr r �connect)r r# r$ r* r )r( r) r �register_logger_signal! s r. c @ s> e Zd Zddd�Zdd� Zdd� Zdd � Zd d � Zd d � ZdS )r Fc C s || _ || _d S )N)r r )r r r r r r �__init__8 s z SentryCeleryHandler.__init__c C s4 t j| jdd� tj| jdd� tj| jdd� d S )NF)r+ )r r- �handle_task_prerunr �handle_task_postrunr �process_failure_signal)r r r r r < s zSentryCeleryHandler.installc C s( t �| j� t�| j� t�| j� d S )N)r Z disconnectr0 r r1 r r2 )r r r r � uninstallA s z SentryCeleryHandler.uninstallc K sd | j r"t|d�r"t|j|j�r"d S t|jt�rBddt|d|�g}nd }| jj||||d�|d� d S )N�throwsZceleryr �name)�task_id�task�args�kwargs)�extra� fingerprint) r �hasattrr � exceptionr4 r r r ZcaptureException)r r" r6 r8 r9 Zeinfor& r; r r r r2 F s z*SentryCeleryHandler.process_failure_signalc K s | j j�� | j j�|j� d S )N)r �context�activate� transaction�pushr5 )r r" r6 r7 r& r r r r0 Z s z&SentryCeleryHandler.handle_task_prerunc K s | j j�|j� | j j�� d S )N)r r@ �popr5 r> �clear)r r" r6 r7 r& r r r r1 ^ s z'SentryCeleryHandler.handle_task_postrunN)F) r r r r/ r r3 r2 r0 r1 r r r r r 7 s r )F)�__doc__� __future__r �loggingZcelery.exceptionsr Zcelery.signalsr r r r Zraven.handlers.loggingr �Filterr r �ERRORr. �objectr r r r r �<module> s