File: //opt/alt/python37/lib/python3.7/site-packages/raven/contrib/__pycache__/flask.cpython-37.opt-1.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_exceptionsZextra)�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)r/ )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_infor4 r9 �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' Z content_typer r r r> � s zSentry.is_json_typec C s |j S )N)Zform)r' r r r r r@ � 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�methodrD �headers�env)r@ r ZurlsplitrE r ZschemeZnetloc�pathZqueryrF �dictr �environr )r' r rB ZurlpartsrD r r r rA � 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_contextrC r+ �loggerZ exceptionr � user_contextr= )r' r1 r2 �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- rG r" �context�clearr rL rM �poprO )r' �senderZresponser1 r2 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 ) NFTZexclude)r# Zsentry_handler)rW �
extensionsZsentry)r r# r$ �debugr% r r! r" r r r
r
rP Z propagateZ
addHandlerr �send�SentryMiddlewareZwsgi_apprS r ZconnectrX r r3 r8 rY )
r' r r r r# r! r$ r% r2 Zhandlerr r r r& � sJ
zSentry.init_appc O s, | j j||�}|r"| j �|�| _nd | _|S )N)r" r0 � get_identr- )r' r1 r2 �resultr r r r0 3 s
zSentry.captureExceptionc O s, | j j||�}|r"| j �|�| _nd | _|S )N)r" �captureMessager] r- )r' r1 r2 r^ r r r r_ < s
zSentry.captureMessagec O s | j j||�S )N)r" rQ )r' r1 r2 r r r rQ E s zSentry.user_contextc O s | j j||�S )N)r" � tags_context)r' r1 r2 r r r r` I s zSentry.tags_contextc O s | j j||�S )N)r" �
extra_context)r' r1 r2 r r r ra M s zSentry.extra_context)N)NNNNNN)�__name__�
__module__� __qualname__�__doc__r r ZNOTSETr( �propertyr- �setterr3 r= rC r> r@ r? rA rS rX r& r0 r_ rQ r` ra r r r r r ; s. & ,
7 r )N)*re Z
__future__r Z flask_loginr � ImportErrorr7 r ZblinkerZflaskr r r Z
flask.signalsr r Zwerkzeug.exceptionsr Z
raven.confr
Z
raven.baser Zraven.middlewarer r\ Zraven.handlers.loggingr
Zraven.utils.compatr Zraven.utils.encodingr Zraven.utils.wsgir r Zraven.utils.confr Z NamespaceZ
raven_signals�signalr r �objectr r r r �<module> s.