Your IP : 18.191.84.179
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.