Edit File: flask.cpython-37.opt-1.pyc
B Z~�_ � @ s� d dl mZ d dlZd dlmZmZ d dlmZmZ d dl m Z m Z d dl m Z d dlmZ d dlmZ er�d d l mZ d d lmZ d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlm Z y d dl Z W n e k r� dZ Y nX y0d dl m Z m!Z!m"Z"m#Z#m$Z% d dl&m'Z'm(Z( W n e k �rL e d��Y nX y d dl)Z)W n e k �rx e d��Y nX dZ*G dd� de �Z+dd� Z,G dd � d e�Z-d d � Z.d d � Z/d!d"� Z0dS )#� )�absolute_importN)�Hub�_should_send_default_pii)�capture_internal_exceptions�event_from_exception)� Integration� DidNotEnable)�SentryWsgiMiddleware)�RequestExtractor)�MYPY)�_ScopedResponse)�Any)�Dict)�ImmutableTypeConversionDict)�ImmutableMultiDict)� FileStorage)�Union)�Callable)�EventProcessor)�Request�Flask�_request_ctx_stack�_app_ctx_stack� __version__)�got_request_exception�request_startedzFlask is not installedzblinker is not installed)�endpoint�urlc @ s* e Zd ZdZdZddd�Zedd� �ZdS ) �FlaskIntegration�flaskNr 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$ �N/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/flask.py�__init__; s zFlaskIntegration.__init__c s� y t ttt�d�d d� ��} W n& ttfk rF td�t���Y nX | dk rXtd��t � t � t � t � tj� � fdd�}|t_d S )N�.� z Unparsable Flask version: {})r � z Flask 0.10 or newer is required.c s4 t j�t�d kr �� ||�S t�� fdd��||�S )Nc s � �f| �|�S )Nr$ )�a�kw)�old_appr# r$ r% �<lambda>Y � zNFlaskIntegration.setup_once.<locals>.sentry_patched_wsgi_app.<locals>.<lambda>)r �current�get_integrationr r )r# �environZstart_response)r, )r# r% �sentry_patched_wsgi_appT s z<FlaskIntegration.setup_once.<locals>.sentry_patched_wsgi_app)�tuple�map�int� FLASK_VERSION�splitr! � TypeErrorr �formatr Zconnect�_request_startedr �_capture_exceptionr �__call__)�versionr2 r$ )r, r% � setup_onceD s zFlaskIntegration.setup_once)r )�__name__� __module__� __qualname__Z identifierr" r& � staticmethodr> r$ r$ r$ r% r 6 s r c K s� t j}|�t�}|d kr d S tjj}|�� �r}tjj }y.|j dkrN|j j |_ n|j dkrb|j j|_ W n tk rx Y nX t�|�}t|||�}|�|� W d Q R X d S )Nr r )r r/ r0 r r �top�appZconfigure_scoper �requestr" Zurl_ruler Z transactionZrule� Exception�weakref�ref� _make_request_event_processorZadd_event_processor) �sender�kwargs�hub� integrationrD ZscoperE � weak_requestZ evt_processorr$ r$ r% r: ` s$ r: c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�FlaskRequestExtractorc C s | j jS )N)rE r1 )r# r$ r$ r% �env} s zFlaskRequestExtractor.envc C s | j jS )N)rE �cookies)r# r$ r$ r% rQ � s z FlaskRequestExtractor.cookiesc C s | j �� S )N)rE �get_data)r# r$ r$ r% �raw_data� s z FlaskRequestExtractor.raw_datac C s | j jS )N)rE �form)r# r$ r$ r% rT � s zFlaskRequestExtractor.formc C s | j jS )N)rE �files)r# r$ r$ r% rU � s zFlaskRequestExtractor.filesc C s | j jS )N)rE �is_json)r# r$ r$ r% rV � s z FlaskRequestExtractor.is_jsonc C s | j �� S )N)rE Zget_json)r# r$ r$ r% �json� s zFlaskRequestExtractor.jsonc C s |j S )N)Zcontent_length)r# �filer$ r$ r% � size_of_file� s z"FlaskRequestExtractor.size_of_fileN) r? r@ rA rP rQ rS rT rU rV rW rY r$ r$ r$ r% rO | s rO c s � fdd�}|S )Nc sV � � }|d kr| S t � � t|��| � W d Q R X t� rRt � � t| � W d Q R X | S )N)r rO Zextract_into_eventr �_add_user_to_event)�event�hintrE )rN r$ r% �inner� s z,_make_request_event_processor.<locals>.innerr$ )rD rN rM r] r$ )rN r% rI � s rI c K sJ t j}|�t�d krd S |j}t||jddd�d�\}}|j||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)r\ )r r/ r0 r �clientr �optionsZ capture_event)rJ Z exceptionrK rL r_ r[ r\ r$ r$ r% r; � s r; c C s� t d kr d S t j}|d kr d S t� �� | �di �}y|�d|�� � W n tk rZ Y nX y|�d|j� W n tk r� Y nX y |�d|j� |�d|j� W n tk r� Y nX W d Q R X d S )N�user�id�email�username) � flask_loginZ current_userr � setdefaultZget_id�AttributeErrorrc rF rd )r[ ra Z user_infor$ r$ r% rZ � s( rZ )1Z __future__r rG Zsentry_sdk.hubr r Zsentry_sdk.utilsr r Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.wsgir Z$sentry_sdk.integrations._wsgi_commonr Zsentry_sdk._typesr r �typingr r Zwerkzeug.datastructuresr r r r r r re � ImportErrorr r r r r r r6 Z flask.signalsr r Zblinkerr r r: rO rI r; rZ r$ r$ r$ r% �<module> sJ * "