B ��Y2 �@stdZddlmZddlmZddlmZddlmZm Z ddl m Zddlm Z Gdd �d e �ZGd d�de�ZdS) z� raven.contrib.tornado ~~~~~~~~~~~~~~~~~~~~~ :copyright: (c) 2012 by the Sentry Team, see AUTHORS for more details :license: BSD, see LICENSE for more details �)�absolute_import)�partial)�ioloop)�AsyncHTTPClient� HTTPError)r)�ClientcsNeZdZdZ�fdd�Zdd�Zddd�Zdd d �Zdd�Zdd d�Z �Z S)�AsyncSentryClientz� A mixin class that could be used along with request handlers to asynchronously send errors to sentry. The client also captures the information from the request handlers cs$|�dd�|_tt|�j||�dS)N� validate_certT)�popr �superr�__init__)�self�args�kwargs)� __class__��O/opt/alt/python37/lib/python3.7/site-packages/raven/contrib/tornado/__init__.pyrszAsyncSentryClient.__init__cOs@|��sdS|j||�}|jfd|�dd�i|��}|d|fS)a Takes the same arguments as the super function in :py:class:`Client` and extracts the keyword argument callback which will be called on asynchronous sending of the request :return: a 32-length string identifying this event N�callbackZevent_id)� is_enabledZ build_msg�send�get)r rr�data�futurerrr�captures zAsyncSentryClient.captureNcKs|�|�}|j|||d�S)zV Serializes the message and passes the payload onto ``send_encoded``. )�auth_headerr)�encodeZsend_encoded)r rrr�messagerrrr/s zAsyncSentryClient.sendcCs`|dkri}|j��s.|�|�}|�|�dS|j||||d�}tj��|t |j ||��|S)N)�urlr�headersr)�stateZ should_try�decodeZ_log_failed_submission�_send_remoterZIOLoop�currentZ add_futurer�_handle_result)r rrrrrrrr�send_remote7s zAsyncSentryClient.send_remotec Cs�y|��Wnrtk rF}z|�|�}|�|||�Wdd}~XYnDtk r~}z|�|�}|�|||�Wdd}~XYnX|j��dS)N)�resultrr Z_failed_send� ExceptionrZset_success)r rrr�errrr#Fs z AsyncSentryClient._handle_resultcCs&|dkri}t�j||d|||jd�S)z� Initialise a Tornado AsyncClient and send the reuqest to the sentry server. If the callback is a callable, it will be called with the response. NZPOST)�method�bodyrr )rZfetchr )r rrrrrrrr!Rs zAsyncSentryClient._send_remote)NN)NN)NN)�__name__� __module__�__qualname__�__doc__rrrr$r#r!� __classcell__rr)rrrs rcsreZdZdZdd�Zdd�Zdd�Zdd �Zd d�Zdd d�Z ddd�Z dd�Z�fdd�Zd�fdd� Z �ZS)�SentryMixina� A mixin class that extracts information from the Request in a Request Handler to capture and send to sentry. This mixin class is designed to be used along with `tornado.web.RequestHandler` .. code-block:: python :emphasize-lines: 6 class MyRequestHandler(SentryMixin, tornado.web.RequestHandler): def get(self): try: fail() except Exception as e: self.captureException() While the above example would result in sequential execution, an example for asynchronous use would be .. code-block:: python :emphasize-lines: 6 class MyRequestHandler(SentryMixin, tornado.web.RequestHandler): @tornado.web.asynchronous @tornado.gen.engine def get(self): # Do something and record a message in sentry response = yield tornado.gen.Task( self.captureMessage, "Did something really important" ) self.write("Your request to do something important is done") self.finish() The mixin assumes that the application will have an attribute called `sentry_client`, which should be an instance of :py:class:`AsyncSentryClient`. This can be changed by implementing your own get_sentry_client method on your request handler. cCs|jjS)z� Returns the sentry client configured in the application. If you need to change the behaviour to do something else to get the client, then subclass this method )ZapplicationZ sentry_client)r rrr�get_sentry_client�szSentryMixin.get_sentry_clientc Cs<d|j��|jj|jj|jj|jj�dd�t|jj�d�iS)z� Extracts the data required for 'sentry.interfaces.Http' from the current request being handled by the request handler :param return: A dictionary. �request�CookieN)rr(rZquery_stringZcookiesr)r1Zfull_urlr(r)�queryrr�dict)r rrr�get_sentry_data_from_request�sz(SentryMixin.get_sentry_data_from_requestcCs6y|��}Wntk r iSXdd|r.dndiiS)z� Data for sentry.interfaces.User Default implementation only sends `is_authenticated` by checking if `tornado.web.RequestHandler.get_current_user` tests postitively for on Truth calue testing �userZis_authenticatedTF)Zget_current_userr&)r r6rrr�get_sentry_user_info�sz SentryMixin.get_sentry_user_infocCsdiiS)zf Subclass and implement this method if you need to send any extra information �extrar)r rrr�get_sentry_extra_info�sz!SentryMixin.get_sentry_extra_infocCs2i}|�|��|�|��|�|��|S)N)�updater5r7r9)r rrrr�get_default_context�s zSentryMixin.get_default_contextNcKs`|dkr|��}n.|��}t|t�r0|�|�n||dd<|}|��}t||�fd|i|��S)Nr8� extra_datar)r;� isinstancer4r:r0�getattr)r Z call_namerrZdefault_contextZclientrrr�_capture�s zSentryMixin._capturecKs|jdd|i|��S)N�captureException�exc_info)r@)r?)r rArrrrr@�szSentryMixin.captureExceptioncKs|jdd|i|��S)N�captureMessager)rB)r?)r rrrrrrB�szSentryMixin.captureMessagecsLtt|��|||�}t|t�r6|jdks2|jdkr6|S|j|||fd�|S)zvOverride implementation to report all exceptions to sentry. log_exception() is added in Tornado v3.1. i�iW)rA)rr/� log_exceptionr=�WebHTTPError�status_coder@)r �typ�value�tb�rv)rrrrC�s zSentryMixin.log_exception��csjttt|�d�r$tt|�j|f|�Stt|�j|f|�}d|krLdkrbnn|j|�d�d�|SdS)z�Override implementation to report all exceptions to sentry, even after self.flush() or self.finish() is called, for pre-v3.1 Tornado. rCi�iWrA)rAN)�hasattrrr/� send_errorr@r)r rErrI)rrrrL�szSentryMixin.send_error)N)N)rJ)r*r+r,r-r0r5r7r9r;r?r@rBrCrLr.rr)rrr/as( r/N)r-� __future__r� functoolsrZtornadorZtornado.httpclientrrZtornado.webrDZ raven.baserr�objectr/rrrr�sN