관리-도구
편집 파일: dramatiq.cpython-38.pyc
U �Mg� � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZmZm Z d dl m Z d dl m Z d dlmZmZ d dlmZ d d lmZ er�d d lmZmZmZmZmZ d d lmZm Z G d d � d e�Z dd� Z G dd� de�Z dd� Z G dd� d�Z!dS )� N)� Integration)�request_body_within_bounds)�AnnotatedValue�capture_internal_exceptions�event_from_exception)�Broker)�Message)� Middleware�default_middleware)�Retry)� TYPE_CHECKING)�Any�Callable�Dict�Optional�Union)�Event�Hintc @ s e Zd ZdZdZedd� �ZdS )�DramatiqIntegrationa? Dramatiq integration for Sentry Please make sure that you call `sentry_sdk.init` *before* initializing your broker, as it monkey patches `Broker.__init__`. This integration was originally developed and maintained by https://github.com/jacobsvante and later donated to the Sentry project. �dramatiqc C s t � d S �N)�_patch_dramatiq_broker� r r �N/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/dramatiq.py� setup_once& s z DramatiqIntegration.setup_onceN)�__name__� __module__� __qualname__�__doc__� identifier� staticmethodr r r r r r s r c s t j� � fdd�} | t _d S )Nc s� t �� �t�}z|�d�}W n2 tk rN t|�dkrF|d }g }nd }Y nX |d krltdd� tD ��}nt|�}|d k r�dd� |D �}|� dt � � ||d< � | f|�|� d S )N� middleware� r c s s | ] }|� V qd S r r ��.0�mr r r � <genexpr>A s zP_patch_dramatiq_broker.<locals>.sentry_patched_broker__init__.<locals>.<genexpr>c S s g | ]}t |t�s|�qS r )� isinstance�SentryMiddlewarer# r r r � <listcomp>F s zQ_patch_dramatiq_broker.<locals>.sentry_patched_broker__init__.<locals>.<listcomp>) � sentry_sdk� get_client�get_integrationr �pop�KeyError�len�listr �insertr( )�self�args�kw� integrationr! �Zoriginal_broker__init__r r � sentry_patched_broker__init__0 s z=_patch_dramatiq_broker.<locals>.sentry_patched_broker__init__)r �__init__)r7 r r6 r r , s r c @ s( e Zd ZdZdd� Zddd�dd�ZdS )r( z� A Dramatiq middleware that automatically captures and sends exceptions to Sentry. This is automatically added to every instantiated broker via the DramatiqIntegration. c C s` t �� �t�}|d krd S t �� |_|j�� t �� }|j|_ |� d|j � |� t ||�� d S )NZdramatiq_message_id)r* r+ r, r Z new_scope�_scope_manager� __enter__Zget_current_scope� actor_nameZ transactionZ set_extraZ message_idZadd_event_processor� _make_message_event_processor)r2 �broker�messager5 �scoper r r �before_process_messageX s z'SentryMiddleware.before_process_messageN)�result� exceptionc C s� t �� �t�}|d krd S |�|j�}|j�d�p<|j�d�}zR|d k r�|rVt ||�s�t |t �s�t |t �� jtj dd�d�\}} t j|| d� W 5 |j� d d d � X d S )N�throwsF)�typeZhandled)Zclient_optionsZ mechanism)�hint)r* r+ r, r Z get_actorr; �options�getr9 �__exit__r' r r r Z capture_event) r2 r= r> rA rB r5 ZactorrC �eventrE r r r �after_process_messagef s. ������ z&SentryMiddleware.after_process_message)r r r r r@ rJ r r r r r( O s r( c s � fdd�}|S )Nc s$ t � � t� ��| � W 5 Q R X | S r )r �DramatiqMessageExtractor�extract_into_event)rI rE �r> r r �inner� s z,_make_message_event_processor.<locals>.innerr )r> r5 rN r rM r r< � s r< c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )rK c C s t |�� �| _d S r )�dict�asdict� message_data)r2 r> r r r r8 � s z!DramatiqMessageExtractor.__init__c C s t t�| j��S r )r/ �json�dumpsrQ )r2 r r r �content_length� s z'DramatiqMessageExtractor.content_lengthc C sb t �� }|�� sd S |�di �}|�di �}d|d< d }t|| �� �sPt�� }n| j}||d< d S )N�contextsr rD �data) r* r+ Z is_active� setdefaultr rT r Z removed_because_over_size_limitrQ )r2 rI �clientrU Z request_inforV r r r rL � s z+DramatiqMessageExtractor.extract_into_eventN)r r r r8 rT rL r r r r rK � s rK )"rR r* Zsentry_sdk.integrationsr Z$sentry_sdk.integrations._wsgi_commonr Zsentry_sdk.utilsr r r Zdramatiq.brokerr Zdramatiq.messager Zdramatiq.middlewarer r Zdramatiq.errorsr �typingr r r r r r Zsentry_sdk._typesr r r r r( r< rK r r r r �<module> s" #3