Your IP : 216.73.216.213
��Yw) � @ so d Z d d l m Z y d d l m Z Wn e k
rD d Z Yn Xd Z d d l Z d d l Z d d l m
Z
m Z m Z d d l
m Z m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d l m Z m! Z! d d l" m# Z# e j$ � Z% e% j& d � Z' d d d � Z( Gd d � d e) � Z d S)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 sk | t | j d d | d t | j j d g � � t | j g � Bd | j j d g � d d | i i �� S) N�defaults�dsnZ
include_pathsZSENTRY_INCLUDE_PATHSZignore_exceptionsZRAVEN_IGNORE_EXCEPTIONS�extra�app)r �config�set�getZ import_name)�
client_clsr r � r �D/opt/alt/python35/lib/python3.5/site-packages/raven/contrib/flask.py� make_client' s r c @ s0 e Z d Z d Z d d e d d d e j d d d d � Z e d d � � Z e j
d d � � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z d d d � Z d d � Z d d � Z d d d d d d d d � Z d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z d S)(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 s� | r" t | t � r" t d � � | | _ | | _ | | _ | | _ | | _ | | _ | | _ | | _
| r} | j | � 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 s0 y t j SWn t k
r Yn Xt | d d � 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 _ Wn t k
r* Yn Xd S)N)r* r r+ r, )r( �valuer r r r. s
c O s* | j s
d S| j d | j d � � d S)N�exc_info)r# �captureExceptionr )r( �args�kwargsr r r �handle_exception� s zSentry.handle_exceptionc C s i } y | j d } Wn t k
r4 | j } Yn X| rE | | d <t sO | St t d � sb | Sy
t j } Wn t k
r� | SYn Xt | � r� | � } | s� | St j
� | d <d t j k r x7 t j d D]( } t t | � 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 s7 | j | j � r | j } n | j } | 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
| d k S)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 s� | d k r | j } t j | j � } y | | � } Wn t k
rQ i } Yn Xd d | j | j | j f d | j d | j d | d t
t | j � � d t
t
| j � � i S) zT
Exact method for getting http_info but with form data work around.
N�urlz %s://%s%sZ 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 | _ t j r+ | j j j t j j � y | j j | j t � � Wn> t k
r� } z | j j
j t | � � WYd d } ~ Xn Xy | j j
| j t � � Wn> t k
r� } z | j j
j t | � � WYd d } ~ Xn Xd 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 sO | j r | j | j d <| j j j � t j rK | j j j t j j � | 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 s� | d k r | | _ | d k r* | | _ | d k rB | | _ n3 | j d k ru | rl | j rl d | _ n d | _ | d k r� | | _ | d k r� | | _ | d k r� | | _ | j s� t | j | | j � | _ | j r`i } | j d k r| j | d <t
| j d | j �} t | | � | j j
d k rJ| j j | � t j | d t
| �| j r�t | j | j � | _ | j | j � t j | j d | �| j r�t j | j d | �t | d � s�i | _ | | j d <d S) NFT�excluder$ Zsentry_handlerr^ �
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 sU | j s t d � � | j j | | � } | rH | j j | � | _ n d | _ | S)Nz5captureException called before application configured)r# �AssertionErrorr1 � get_identr. )r( r2 r3 �resultr r r r1 3 s zSentry.captureExceptionc O sU | j s t d � � | j j | | � } | rH | j j | � | _ n d | _ | S)Nz3captureMessage called before application configured)r# rj �captureMessagerk r. )r( r2 r3 rl r r r rm < s zSentry.captureMessagec O s( | j s t d � � | 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 s t d � � | 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 s t d � � | 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)�__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 )*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.