관리-도구
편집 파일: falcon.cpython-38.pyc
U �Mg�$ � @ s� d dl Z d dlmZmZ d dlmZ d dlmZ d dlm Z d dl m Z m Z m Z mZ d dlmZ er�d dlmZ d d lmZ d d lmZ d d lmZmZ zd dlZd d lmZ W n ek r� ed ��Y nX z d dlZej Z ej Z dZ W n. ek �r d dl!Zej"Z ej#Z dZ Y nX G dd� de�Z$G dd� d�Z%dZ&G dd� de�Z'dd� Z(dd� Z)dd � Z*d d � Z+d d � Z,d!d"� Z-d#d$� Z.dS )%� N)� Integration� DidNotEnable)�RequestExtractor)�SentryWsgiMiddleware)�SOURCE_FOR_STYLE)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception� parse_version)� TYPE_CHECKING)�Any)�Dict)�Optional)�Event�EventProcessor)� __version__zFalcon not installedTFc @ sJ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zer>d d � Z nd d � Z dS )�FalconRequestExtractorc C s | j jS �N)�request�env��self� r �L/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/falcon.pyr / s zFalconRequestExtractor.envc C s | j jS r )r �cookiesr r r r r 3 s z FalconRequestExtractor.cookiesc C s d S r r r r r r �form7 s zFalconRequestExtractor.formc C s d S r r 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* z | j jW S tjjk r$ Y d S X d S r )r �media�falcon�errors�HTTPBadRequestr r r r �jsonN s zFalconRequestExtractor.jsonc C s0 z | j jW S tjjk r* | j j Y S X d S r )r r r r! r" Z_mediar r r r r# W s N) �__name__� __module__� __qualname__r r r r r �FALCON3r# r r r r r . s r c @ s e Zd ZdZdd� ZdS )�SentryFalconMiddlewarez9Captures exceptions in Falcon requests and send to Sentryc O s< t �� �t�}|d krd S t �� }d|_|�t||�� d S )Nr )� sentry_sdk� get_client�get_integration�FalconIntegrationZget_isolation_scope�_nameZadd_event_processor� _make_request_event_processor)r �req�resp�args�kwargs� integration�scoper r r �process_requestf s z&SentryFalconMiddleware.process_requestN)r$ r% r& �__doc__r5 r r r r r( c s r( �� uri_template�pathc @ s4 e Zd ZdZde� �ZdZd dd�Zedd� �Zd S ) r, r z auto.http.� r8 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 r= r r r �__init__z s ��zFalconIntegration.__init__c C sD t t�} | d kr td�t���| dk r.td��t� t� t� d S )Nz Unparsable Falcon version: {})� � z Falcon 1.4 or newer required.)r �FALCON_VERSIONr �format�_patch_wsgi_app�_patch_handle_exception�_patch_prepare_middleware)�versionr r r � setup_once� s z FalconIntegration.setup_onceN)r8 ) r$ r% r& � identifier�originr= r> � staticmethodrG r r r r r, t s r, c s t j� � fdd�} | t _d S )Nc sD t �� �t�}|d kr"�� ||�S t�� fdd�tjd�}|||�S )Nc s � �| |�S r r )ZenviZ start_resp)�original_wsgi_appr r r �<lambda>� � zB_patch_wsgi_app.<locals>.sentry_patched_wsgi_app.<locals>.<lambda>)Z span_origin)r) r* r+ r, r rI )r r Zstart_responser3 Zsentry_wrapped�rK r r �sentry_patched_wsgi_app� s �z0_patch_wsgi_app.<locals>.sentry_patched_wsgi_app)�falcon_app_class�__call__)rO r rN r rC � s rC c s& t j� tt� �� fdd��} | t _d S )Nc s� d }}t � �* tdd� |D ��}tdd� |D ��}W 5 Q R X � | f|�� }|d ks\|d kr`|S t||�r�t|t�� jddd�d�\}}tj||d� |S ) Nc s s | ]}t |t�r|V qd S r )� isinstance� Exception��.0�argumentr r r � <genexpr>� s zS_patch_handle_exception.<locals>.sentry_patched_handle_exception.<locals>.<genexpr>c s s | ]}t |tj�r|V qd S r )rR r �ResponserT r r r rW � s r F)�typeZhandled)Zclient_optionsZ mechanism)�hint)r �next� _exception_leads_to_http_5xxr r) r* �optionsZ capture_event)r r1 �ex�responseZ was_handled�eventrZ �Zoriginal_handle_exceptionr r � sentry_patched_handle_exception� s" � � z@_patch_handle_exception.<locals>.sentry_patched_handle_exception)rP Z_handle_exceptionr r, )rb r ra r rD � s rD c s t j� d� fdd� } | t _d S )NFc s@ |r� | ||�S t �� �t�}|d k r6t� g| p2g } � | |�S r )r) r* r+ r, r( )Z middlewareZindependent_middlewareZasgir3 �Zoriginal_prepare_middlewarer r �!sentry_patched_prepare_middleware� s zD_patch_prepare_middleware.<locals>.sentry_patched_prepare_middleware)NFF)�falcon_helpersZprepare_middleware)rd r rc r rE � s �rE c C sH t | tj�o| jpd�d�}t | tjtjjf� }|s:|oFt pFt|�S )Nr: �5) rR r � HTTPError�status� startswithZ http_status� HTTPStatusr' �_has_http_5xx_status)r^ r_ Zis_server_errorZis_unhandled_errorr r r r\ � s � � �r\ c C s | j �d�S )Nrf )rh ri )r_ r r r rk � s rk c C s. |j |jd�}|| | d< dt| i| d< d S )Nr7 Z transaction�sourceZtransaction_info)r8 r9 r )r` r= r Zname_for_styler r r � _set_transaction_name_and_source� s � rm c s � �fdd�}|S )Nc s2 t | � j�� t� � t���| � W 5 Q R X | S r )rm r= r r Zextract_into_event)r` rZ �r3 r/ r r �event_processor s z6_make_request_event_processor.<locals>.event_processorr )r/ r3 ro r rn r r. s r. )/r) Zsentry_sdk.integrationsr r Z$sentry_sdk.integrations._wsgi_commonr Z sentry_sdk.integrations.wsgir Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r �typingr r r r Zsentry_sdk._typesr r r r rA � ImportErrorZfalcon.app_helpersZ app_helpersre ZApprP r' Zfalcon.api_helpersZ api_helpersZAPIr r( r; r, rC rD rE r\ rk rm r. r r r r �<module> sJ 5 &