B ��Y �@s�dZddlmZddlZddlmZddlmZmZm Z m Z ddlmZGdd�dej �Zdd d�Zdejfdd �ZGdd�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@seZdZdd�ZdS)�CeleryFiltercCs2t|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�filters zCeleryFilter.filterN)�__name__� __module__�__qualname__rrrrrr sr FcCst||d��dS)N)�ignore_expected)�SentryCeleryHandler�install)�clientrrrr�register_signalsrcsBt��t|��|��fdd�}tj|dd�dS)Ncs8x(|jD]}t|�tkr|��dSqW|��dS)NF)�handlers�typer� addFilter� addHandler)�sender�logger�loglevelZlogfile�formatZcolorize�kw�h)�filter_�handlerrr�process_logger_event(s z4register_logger_signal..process_logger_eventF)�weak)r r�setLevelr r�connect)rr#r$r*r)r(r)r�register_logger_signal!s r.c@s>eZdZddd�Zdd�Zdd�Zdd �Zd d�Zdd �ZdS)rFcCs||_||_dS)N)rr)rrrrrr�__init__8szSentryCeleryHandler.__init__cCs4tj|jdd�tj|jdd�tj|jdd�dS)NF)r+)rr-�handle_task_prerunr�handle_task_postrunr�process_failure_signal)rrrrr<szSentryCeleryHandler.installcCs(t�|j�t�|j�t�|j�dS)N)rZ disconnectr0rr1rr2)rrrr� uninstallAszSentryCeleryHandler.uninstallcKsd|jr"t|d�r"t|j|j�r"dSt|jt�rBddt|d|�g}nd}|jj||||d�|d�dS)N�throwsZceleryr�name)�task_id�task�args�kwargs)�extra�fingerprint) r�hasattrr� exceptionr4rrrZcaptureException)rr"r6r8r9Zeinfor&r;rrrr2Fsz*SentryCeleryHandler.process_failure_signalcKs |jj��|jj�|j�dS)N)r�context�activate�transaction�pushr5)rr"r6r7r&rrrr0Zsz&SentryCeleryHandler.handle_task_preruncKs |jj�|j�|jj��dS)N)rr@�popr5r>�clear)rr"r6r7r&rrrr1^sz'SentryCeleryHandler.handle_task_postrunN)F) rrrr/rr3r2r0r1rrrrr7s r)F)�__doc__� __future__r�loggingZcelery.exceptionsrZcelery.signalsrrrrZraven.handlers.loggingr�Filterr r�ERRORr.�objectrrrrr�s