Edit File: middleware.cpython-37.pyc
B ��Y � @ sp d Z ddlmZ ddlmZ ddlmZmZ ddlm Z m Z m Z edd� �Z G dd � d e�Z G d d � d e�Zd S ) z� raven.middleware ~~~~~~~~~~~~~~~~~~~~~~~~ :copyright: (c) 2010-2012 by the Sentry Team, see AUTHORS for more details. :license: BSD, see LICENSE for more details. � )�absolute_import)�contextmanager)�Iterator�next)�get_current_url� get_headers� get_environc c s� y d V W n� t tfk r$ � Y nt tk rB |�| � � Y nV tk r` |�| � � Y n8 tk r� } z|jdkr�|�| � � W d d }~X Y nX d S )Nr )� StopIteration� GeneratorExit� Exception�handle_exception�KeyboardInterrupt� SystemExit�code)�environ�client�e� r �A/opt/alt/python37/lib/python3.7/site-packages/raven/middleware.py�common_exception_handling s r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �ClosingIteratorz[ An iterator that is implements a ``close`` method as-per WSGI recommendation. c C s. || _ || _t|dd �| _t|�| _d| _d S )N�closeF)�sentryr �getattr�_close�iter�iterable�closed)�selfr r r r r r �__init__, s zClosingIterator.__init__c C s | S )Nr )r r r r �__iter__3 s zClosingIterator.__iter__c C sH y$t | j| j�� t| j�S Q R X W n tk rB | �� � Y nX d S )N)r r r r r r r )r r r r �__next__6 s zClosingIterator.__next__c C sd | j r d S z0| jd k r8t| j| j�� | �� W d Q R X W d | jjj�� | jjj�� d| _ X d S )NT) r r r r r r �context�clearZ transaction)r r r r r B s zClosingIterator.closeN)�__name__� __module__� __qualname__�__doc__r r r! r r r r r r & s r c @ s4 e Zd ZdZd dd�Zdd� Zdd� Zd d d �ZdS ) �Sentryz� A WSGI middleware which will attempt to capture any uncaught exceptions and send them to Sentry. >>> from raven.base import Client >>> application = Sentry(application, Client()) Nc C s* || _ |d kr ddlm} |� }|| _d S )Nr )�Client)� applicationZ raven.baser) r )r r* r r) r r r r X s zSentry.__init__c C s@ | j �| �|�� t|| �� | �||�}W d Q R X t| ||�S )N)r Z http_context�get_http_contextr r* r )r r Zstart_responser r r r �__call___ s zSentry.__call__c C s4 |� d�t|dd�|� d�tt|��tt|��d�S )NZREQUEST_METHODT)Zstrip_querystringZ QUERY_STRING)�method�urlZ query_string�headers�env)�getr �dictr r )r r r r r r+ g s zSentry.get_http_contextc C s | j �� S )N)r ZcaptureException)r r r r r r r s zSentry.handle_exception)N)N)r$ r% r& r' r r, r+ r r r r r r( O s r( N)r' � __future__r � contextlibr Zraven.utils.compatr r Zraven.utils.wsgir r r r r �objectr( r r r r �<module> s )