Edit File: falcon.cpython-37.pyc
B Z~�_� � @ s. d dl mZ d dlmZ d dlmZmZ d dlmZ d dl m Z d dl m Z m Z d dlmZ er�d dlmZ d d lmZ d d lmZ d d lmZ y d d lZd d lZd d lmZ W n ek r� ed��Y nX G dd� de�ZG dd� de�Z dZ G dd� de�Z dd� Z dd� Z dd� Z!d d � Z"d d � Z#d S ) � )�absolute_import)�Hub)� Integration� DidNotEnable)�RequestExtractor)�SentryWsgiMiddleware)�capture_internal_exceptions�event_from_exception)�MYPY)�Any)�Dict)�Optional)�EventProcessorN)� __version__zFalcon not installedc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�FalconRequestExtractorc C s | j jS )N)�request�env)�self� r �O/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/falcon.pyr s zFalconRequestExtractor.envc C s | j jS )N)r �cookies)r r r r r s z FalconRequestExtractor.cookiesc C s d S )Nr )r r r r �form$ s zFalconRequestExtractor.formc C s d S )Nr )r r r r �files( s z FalconRequestExtractor.filesc C s | � � }|dkrdS d S d S )Nr z [REQUEST_CONTAINING_RAW_DATA])�content_length)r r r r r �raw_data, s z FalconRequestExtractor.raw_datac C s* y| j jS tjjk r$ | j jS X d S )N)r Zmedia�falcon�errorsZHTTPBadRequestZ_media)r r r r �json9 s zFalconRequestExtractor.jsonN) �__name__� __module__� __qualname__r r r r r r r r r r r s r c @ s e Zd ZdZdd� ZdS )�SentryFalconMiddlewarez9Captures exceptions in Falcon requests and send to Sentryc O sJ t j}|�t�}|d kr d S |�� � }d|_|�t||�� W d Q R X d S )Nr )r �current�get_integration�FalconIntegrationZconfigure_scope�_nameZadd_event_processor� _make_request_event_processor)r �reqZresp�args�kwargs�hub� integrationZscoper r r �process_requestH s z&SentryFalconMiddleware.process_requestN)r r r �__doc__r, r r r r r! E s r! )� uri_template�pathc @ s* e Zd ZdZdZddd�Zedd� �ZdS ) r$ r Nr. c C s" |t krtd|t f ��|| _d S )Nz7Invalid value for transaction_style: %s (must be in %s))�TRANSACTION_STYLE_VALUES� ValueError�transaction_style)r r2 r r r �__init__\ s zFalconIntegration.__init__c C sf yt ttt�d���} W n& ttfk r> td�t���Y nX | dk rPtd��t � t � t � d S )N�.z Unparsable Falcon version: {})� � z Falcon 1.4 or newer required.) �tuple�map�int�FALCON_VERSION�splitr1 � TypeErrorr �format�_patch_wsgi_app�_patch_handle_exception�_patch_prepare_middleware)�versionr r r � setup_oncee s z FalconIntegration.setup_once)r. )r r r � identifierr2 r3 � staticmethodrB r r r r r$ W s r$ c s t jj� � fdd�} | t j_d S )Nc s@ t j}|�t�}|d kr$�� ||�S t�� fdd��}|||�S )Nc s � �| |�S )Nr )ZenviZ start_resp)�original_wsgi_appr r r �<lambda>� � zB_patch_wsgi_app.<locals>.sentry_patched_wsgi_app.<locals>.<lambda>)r r" r# r$ r )r r Zstart_responser* r+ Zsentry_wrapped)rE )r r �sentry_patched_wsgi_appy s z0_patch_wsgi_app.<locals>.sentry_patched_wsgi_app)r �API�__call__)rH r )rE r r> u s r> c s t jj� � fdd�} | t j_d S )Nc s� t |d t�r|d }n|d }� | f|�� }tj}|�t�}|d k r|t|�s||j}t||j ddd�d�\}}|j ||d� |S )Nr � r F)�typeZhandled)Zclient_optionsZ mechanism)�hint) � isinstance� Exceptionr r" r# r$ �_is_falcon_http_error�clientr �optionsZ capture_event) r r( �exZ was_handledr* r+ rQ �eventrM )�original_handle_exceptionr r � sentry_patched_handle_exception� s z@_patch_handle_exception.<locals>.sentry_patched_handle_exception)r rI Z_handle_exception)rV r )rU r r? � s r? c s" t jj� d� fdd� } | t j_d S )NFc s2 t j}|�t�}|d k r(t� g| p$g } � | |�S )N)r r" r# r$ r! )Z middlewareZindependent_middlewarer* r+ )�original_prepare_middlewarer r �!sentry_patched_prepare_middleware� s zD_patch_prepare_middleware.<locals>.sentry_patched_prepare_middleware)NF)r Z api_helpersZprepare_middleware)rX r )rW r r@ � s r@ c C s t | tjtjjf�S )N)rN r � HTTPErrorZ http_statusZ HTTPStatus)rS r r r rP � s rP c s � �fdd�}|S )Nc sN � j dkr�j| d< n� j dkr*�j| d< t� � t���| � W d Q R X | S )Nr. Z transactionr/ )r2 r. r/ r r Zextract_into_event)rT rM )r+ r' r r �inner� s z,_make_request_event_processor.<locals>.innerr )r' r+ rZ r )r+ r' r r& � s r& )$� __future__r Zsentry_sdk.hubr Zsentry_sdk.integrationsr r Z$sentry_sdk.integrations._wsgi_commonr Z sentry_sdk.integrations.wsgir Zsentry_sdk.utilsr r Zsentry_sdk._typesr �typingr r r r r Zfalcon.api_helpersr r: � ImportErrorr �objectr! r0 r$ r> r? r@ rP r& r r r r �<module> s4 * #