Edit File: flask.cpython-37.pyc
B ��Yw) � @ s d Z ddlmZ yddlmZ W n ek r8 dZY nX dZddlZddlZddl m Z m Z m Z ddl mZmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlm Z ddl m Z ddl m Z m!Z! ddl"m#Z# e�$� Z%e%�&d�Z'ddd�Z(G dd� de)�ZdS )z� raven.contrib.flask ~~~~~~~~~~~~~~~~~~~ :copyright: (c) 2010-2012 by the Sentry Team, see AUTHORS for more details. :license: BSD, see LICENSE for more details. � )�absolute_import)� current_userFTN)�request� current_app�g)�got_request_exception�request_finished)�ClientDisconnected)� setup_logging)�Client)�Sentry)� SentryHandler)�urlparse)� to_unicode)� get_headers� get_environ)�convert_options�logging_configuredc C sF | f t |j|t|j�dg ��t|jg�B |j�dg �d|id�d��S )NZSENTRY_INCLUDE_PATHSZRAVEN_IGNORE_EXCEPTIONS�app)�dsnZ include_pathsZignore_exceptions�extra)�defaults)r �config�set�getZ import_name)� client_clsr r � r �D/opt/alt/python37/lib/python3.7/site-packages/raven/contrib/flask.py� make_client' s r c @ s� e Zd ZdZddedddejddf dd�Zedd� �Z e j d d� �Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� Zd(dd�Zdd� Zdd� Zd)d d �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� ZdS )*r a� Flask application for Sentry. Look up configuration from ``os.environ['SENTRY_DSN']``:: >>> sentry = Sentry(app) Pass an arbitrary DSN:: >>> sentry = Sentry(app, dsn='http://public:secret@example.com/1') Pass an explicit client:: >>> sentry = Sentry(app, client=client) Automatically configure logging:: >>> sentry = Sentry(app, logging=True, level=logging.ERROR) Capture an exception:: >>> try: >>> 1 / 0 >>> except ZeroDivisionError: >>> sentry.captureException() Capture a message:: >>> sentry.captureMessage('hello, world!') By default, the Flask integration will do the following: - Hook into the `got_request_exception` signal. This can be disabled by passing `register_signal=False`. - Wrap the WSGI application. This can be disabled by passing `wrap_wsgi=False`. - Capture information from Flask-Login (if available). NFTc C sX |rt |t�std��|| _|| _|| _|| _|| _|| _|| _ | | _ |rT| � |� d S )Nz&client should be an instance of Client) � isinstancer � TypeErrorr �logging�logging_exclusionsr �client�level� wrap_wsgi�register_signal�init_app) �selfr r# r r r! r" r$ r% r& r r r �__init__e s zSentry.__init__c C s( yt jS tk r Y nX t| dd �S )N�_last_event_id)r �sentry_event_id� Exception�getattr)r( r r r � last_event_idw s zSentry.last_event_idc C s* || _ y |t_W n tk r$ Y nX d S )N)r* r r+ r, )r( �valuer r r r. s c O s | j s d S | j|�d�d� d S )N�exc_info)r0 )r# �captureExceptionr )r( �args�kwargsr r r �handle_exception� s zSentry.handle_exceptionc C s� i }y|j d }W n tk r, |j}Y nX |r:||d< tsB|S ttd�sP|S y tj}W n tk rn |S X t |�r~|� }|s�|S t� � |d< dtj kr�x*tj d D ] }tt|�r�t t|�||< q�W |S )zu Requires Flask-Login (https://pypi.python.org/pypi/Flask-Login/) to be installed and setup. r � ip_addressZ login_manager�idZSENTRY_USER_ATTRS) Z access_route� IndexErrorZ remote_addr�has_flask_login�hasattrr r �is_authenticated�AttributeError�callableZget_idr r- )r( r Z user_infor5 r: �attrr r r � get_user_info� s2 zSentry.get_user_infoc C s& | � |j�r| j}n| j}| �||�S )zR Determine how to retrieve actual data by using request.mimetype. )� is_json_typeZmimetype� get_json_data� get_form_data� get_http_info_with_retriever)r( r � retrieverr r r � get_http_info� s zSentry.get_http_infoc C s |dkS )Nzapplication/jsonr )r( � content_typer r r r? � s zSentry.is_json_typec C s |j S )N)Zform)r( r r r r rA � s zSentry.get_form_datac C s |j S )N)�data)r( r r r r r@ � s zSentry.get_json_datac C sz |dkr| j }t�|j�}y ||�}W n tk r> i }Y nX d|j|j|jf |j|j |t t |j ��t t |j ��d�S )zT Exact method for getting http_info but with form data work around. Nz %s://%s%s)�urlZ query_string�methodrF �headers�env)rA r �urlsplitrG r �scheme�netloc�path�queryrH �dictr �environr )r( r rC ZurlpartsrF r r r rB � s z#Sentry.get_http_info_with_retrieverc O s� d | _ tjr | jj�tjj� y| j�| �t�� W n4 t k rh } z| jj � t |�� W d d }~X Y nX y| j� | �t�� W n4 t k r� } z| jj � t |�� W d d }~X Y nX d S )N)r. r �url_ruler# � transaction�push�ruleZ http_contextrD r, �logger� exceptionr � user_contextr> )r( r2 r3 �er r r �before_request� s $zSentry.before_requestc O s: | j r| j |jd< | jj�� tjr6| jj�tjj � |S )Nz X-Sentry-ID) r. rI r# �context�clearr rR rS �poprU )r( �sender�responser2 r3 r r r � after_request� s zSentry.after_requestc C sd |d k r|| _ |d k r || _|d k r,|| _n"| jd krN|rH|jrHd| _nd| _|d k r\|| _|d k rj|| _|d k rx|| _| js�t| j || j �| _| jr�i }| jd k r�| j|d< t | j| jd�} t | f|� |j j dkr�|j �| � tj| fdt i|�� | j�rt|j| j�|_|�| j� tj| j|d� | j�rDtj| j|d� t|d��sVi |_| |jd< d S ) NFT�exclude)r$ Zsentry_handler)r^ � extensionsZsentry)r r$ r% �debugr& r! r" r# r r r r rV � propagate� addHandlerr �send�SentryMiddlewareZwsgi_apprZ r �connectr` r r4 r9 rb ) r( r r r! r$ r" r% r& r3 �handlerr r r r' � sJ zSentry.init_appc O s: | j std��| j j||�}|r0| j �|�| _nd | _|S )Nz5captureException called before application configured)r# �AssertionErrorr1 � get_identr. )r( r2 r3 �resultr r r r1 3 s zSentry.captureExceptionc O s: | j std��| j j||�}|r0| j �|�| _nd | _|S )Nz3captureMessage called before application configured)r# rj �captureMessagerk r. )r( r2 r3 rl r r r rm < s zSentry.captureMessagec O s | j std��| j j||�S )Nz1user_context called before application configured)r# rj rX )r( r2 r3 r r r rX E s zSentry.user_contextc O s | j std��| j j||�S )Nz1tags_context called before application configured)r# rj � tags_context)r( r2 r3 r r r rn I s zSentry.tags_contextc O s | j std��| j j||�S )Nz2extra_context called before application configured)r# rj � extra_context)r( r2 r3 r r r ro M s zSentry.extra_context)N)NNNNNN)�__name__� __module__� __qualname__�__doc__r r! �NOTSETr) �propertyr. �setterr4 r> rD r? rA r@ rB rZ r` r' r1 rm rX rn ro r r r r r ; s. & , 7 r )N)*rs � __future__r Z flask_loginr � ImportErrorr8 r! Zblinker�flaskr r r Z flask.signalsr r Zwerkzeug.exceptionsr Z raven.confr Z raven.baser Zraven.middlewarer rg Zraven.handlers.loggingr Zraven.utils.compatr Zraven.utils.encodingr Zraven.utils.wsgir r Zraven.utils.confr � NamespaceZ raven_signals�signalr r �objectr r r r �<module> s.