Edit File: bottle.cpython-37.pyc
B Z~�_+ � @ s. d dl mZ d dlmZ d dlmZmZmZ d dlm Z m Z d dl m Z d dl mZ d dlmZ er�d dl mZ d d lmZ d d lmZ d d lmZ d d lmZ d d lmZmZmZ d dlmZ y d dlm Z m Z m Z m Z m!Z" W n e#k r� e d��Y nX dZ$G dd� de �Z%G dd� de�Z&dd� Z'dS )� )�absolute_import)�Hub)�capture_internal_exceptions�event_from_exception�transaction_from_function)� Integration� DidNotEnable)�SentryWsgiMiddleware)�RequestExtractor)�MYPY)�_ScopedResponse)�Any)�Dict)�Callable)�Optional)� FileUpload� FormsDict� LocalRequest)�EventProcessor)�Bottle�Route�request� HTTPResponse� __version__zBottle not installed)�endpoint�urlc @ s* e Zd ZdZdZddd�Zedd� �ZdS ) �BottleIntegration�bottleNr 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)�selfr � r" �O/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/bottle.py�__init__- s zBottleIntegration.__init__c s� yt ttt�d���} W n& ttfk r> td�| ���Y nX | dk rPtd��t j � � fdd�}|t _ t j ��fdd�}|t _ t j ��fd d �}|t _ d S ) N�.z Unparsable Bottle version: {})r � z Bottle 0.12 or newer required.c s< t j}|�t�}|d kr$�� ||�S t�� fdd��||�S )Nc s � �f| �|�S )Nr" )�a�kw)�old_appr! r" r# �<lambda>N � zOBottleIntegration.setup_once.<locals>.sentry_patched_wsgi_app.<locals>.<lambda>)r �current�get_integrationr r )r! �environZstart_response�hub� integration)r) )r! r# �sentry_patched_wsgi_appF s z=BottleIntegration.setup_once.<locals>.sentry_patched_wsgi_appc sx t j}|�t�}|d kr"� | |�S |�� }|�@ | }|�� � }d|_|�t|t |�� W d Q R X � | |�}W d Q R X |S )Nr ) r r, r- r Z push_scopeZconfigure_scope�_nameZadd_event_processor� _make_request_event_processor�bottle_request)r! r. r/ r0 Z scope_manager�appZscope�res)� old_handler" r# �_patched_handleW s z5BottleIntegration.setup_once.<locals>._patched_handlec sF t j���t�}�| f|�|��|d kr,�S �j� � ��fdd�}|S )Nc sv y�| |�}W nb t k r$ � Y nN tk rp } z0t|� jddd�d�\}}�j||d� |�W d d }~X Y nX |S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)�hint)r � Exceptionr �optionsZ capture_event)�args�kwargsr6 � exception�eventr: )�clientr/ �prepared_callbackr" r# �wrapped_callback} s zUBottleIntegration.setup_once.<locals>.patched_make_callback.<locals>.wrapped_callback)r r, r- r rA )r! r= r> r0 rC )�old_make_callback)rA r/ rB r# �patched_make_callbackr s z;BottleIntegration.setup_once.<locals>.patched_make_callback)�tuple�map�int�BOTTLE_VERSION�split� TypeErrorr r �formatr �__call__�_handler Z_make_callback)�versionr1 r8 rE r" )r) r7 rD r# � setup_once7 s z BottleIntegration.setup_once)r )�__name__� __module__� __qualname__� identifierr r$ � staticmethodrP r" r" r" r# r ( s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�BottleRequestExtractorc C s | j jS )N)r r. )r! r" r" r# �env� s zBottleRequestExtractor.envc C s | j jS )N)r �cookies)r! r" r" r# rX � s z BottleRequestExtractor.cookiesc C s | j j�� S )N)r �body�read)r! r" r" r# �raw_data� s z BottleRequestExtractor.raw_datac C s | � � r d S | jj�� S )N)�is_jsonr Zforms�decode)r! r" r" r# �form� s zBottleRequestExtractor.formc C s | � � r d S | jjS )N)r\ r �files)r! r" r" r# r_ � s z BottleRequestExtractor.filesc C s |j S )N)Zcontent_length)r! �filer" r" r# � size_of_file� s z#BottleRequestExtractor.size_of_fileN) rQ rR rS rW rX r[ r^ r_ ra r" r" r" r# rV � s rV c s � �fdd�}|S )Nc sx y>� j dkr&�jjp t�jj�| d< n� j dkr<�jj| d< W n tk rR Y nX t� � t��� | � W d Q R X | S )Nr Z transactionr ) r Zroute�namer �callbackZruler; r rV Zextract_into_event)r@ r: )r0 r r" r# �inner� s z,_make_request_event_processor.<locals>.innerr" )r5 r r0 rd r" )r0 r r# r3 � s r3 N)(� __future__r Zsentry_sdk.hubr Zsentry_sdk.utilsr r r Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.wsgir Z$sentry_sdk.integrations._wsgi_commonr Zsentry_sdk._typesr r �typingr r r r r r r r r r r r r4 r r rI � ImportErrorr r rV r3 r" r" r" r# �<module> s, l