OwlCyberSecurity - MANAGER
Edit File: flask.cpython-38.opt-1.pyc
U ��\h � @ s d dl mZ d dlZd dlmZmZ d dlmZmZ d dl m Z d dl m Z d dl mZ z d dlZW n ek r| dZY nX d dlmZmZm Z m Z d d l m Z m Z m!Z!m"Z" G d d � d e �Z#d d � Z$dd� Z%dd� Z&G dd� de�Z'dd� Z(dd� Z)dd� Z*dS )� )�absolute_importN)�Hub�_should_send_default_pii)�capture_internal_exceptions�event_from_exception)� Integration)�SentryWsgiMiddleware)�RequestExtractor)�Request�Flask�_request_ctx_stack�_app_ctx_stack)�appcontext_pushed�appcontext_tearing_down�got_request_exception�request_startedc @ s* e Zd ZdZdZddd�Zedd� �ZdS ) �FlaskIntegration�flaskN�endpointc C s& d}||kr t d||f ��|| _d S )N)r �urlz7Invalid value for transaction_style: %s (must be in %s))� ValueError�transaction_style)�selfr ZTRANSACTION_STYLE_VALUES� r �N/opt/alt/python38/lib/python3.8/site-packages/sentry_sdk/integrations/flask.py�__init__( s ��zFlaskIntegration.__init__c sD t �t� t�t� t�t� t�t� t j � � fdd�} | t _ d S )Nc s4 t j�t�d kr �� ||�S t�� fdd��||�S )Nc s � �f| �|�S �Nr )�a�kw)�old_appr r r �<lambda>A � zNFlaskIntegration.setup_once.<locals>.sentry_patched_wsgi_app.<locals>.<lambda>)r �current�get_integrationr r )r �environZstart_response�r �r r �sentry_patched_wsgi_app< s �z<FlaskIntegration.setup_once.<locals>.sentry_patched_wsgi_app) r Zconnect� _push_appctxr � _pop_appctxr �_request_startedr �_capture_exceptionr �__call__)r' r r% r � setup_once2 s zFlaskIntegration.setup_once)r )�__name__� __module__� __qualname__Z identifierr r � staticmethodr- r r r r r # s r c O sJ t j}|�t�d k rF|�� }|�� |tj_|� � � }d|_ W 5 Q R X d S )Nr ) r r" r# r Z push_scope� __enter__r �top�sentry_sdk_scope_manager�configure_scope�_name)�args�kwargs�hub� scope_manager�scoper r r r( H s r( c O s( t tjdd �}|d k r$|�d d d � d S )Nr4 )�getattrr r3 �__exit__)r7 r8 r: r r r r) U s r) c K s\ t j}|�t�}|d kr d S t�tjj�}t jj }|� � �}|� t |||�� W 5 Q R X d S r )r r" r# r �weakref�refr r3 �requestr �appr5 Zadd_event_processor� _make_request_event_processor)�senderr8 r9 � integration� weak_requestrA r; r 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 r )r@ r$ r& r r r �envn s zFlaskRequestExtractor.envc C s | j jS r )r@ �cookiesr& r r r rH r s z FlaskRequestExtractor.cookiesc C s | j jS r )r@ �datar& r r r �raw_datav s z FlaskRequestExtractor.raw_datac C s | j jS r )r@ �formr& r r r rK z s zFlaskRequestExtractor.formc C s | j jS r )r@ �filesr& r r r rL ~ s zFlaskRequestExtractor.filesc C s | j jS r )r@ �is_jsonr& r r r rM � s z FlaskRequestExtractor.is_jsonc C s | j �� S r )r@ Zget_jsonr& r r r �json� s zFlaskRequestExtractor.jsonc C s |j S r )Zcontent_length)r �filer r r � size_of_file� s z"FlaskRequestExtractor.size_of_fileN) r. r/ r0 rG rH rJ rK rL rM rN rP r r r r rF m s rF c s � �fdd�}|S )Nc s� �� }|d kr| S z2� j dkr,|jj| d< n� j dkrB|jj| d< W n tk rX Y nX t� � t|��| � W 5 Q R X t� r�t� � t | � W 5 Q R X | S )Nr Z transactionr ) r Zurl_ruler Zrule� Exceptionr rF Zextract_into_eventr �_add_user_to_event)�event�hintr@ �rD rE r r �inner� s z,_make_request_event_processor.<locals>.innerr )rA rE rD rV r rU r rB � s rB c K sF t j}|�t�d krd S t||jjddd�d�\}}|j||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)rT )r r" r# r r Zclient�optionsZ capture_event)rC Z exceptionr8 r9 rS rT r r r r+ � s � r+ c C s� t d kr d S t j}|d kr d S t� �� | �di �}z|�� |d< W n tk rV Y nX z|j |d< |d< W n tk r� Y nX z|j|d< W n tk r� Y nX W 5 Q R X d S )N�user�id�email�username) � flask_loginZ current_userr � setdefaultZget_id�AttributeErrorr[ rQ r\ )rS rY Z user_infor r r rR � s& rR )+Z __future__r r> Zsentry_sdk.hubr r Zsentry_sdk.utilsr r Zsentry_sdk.integrationsr Z sentry_sdk.integrations.wsgir Z$sentry_sdk.integrations._wsgi_commonr Z_ScopedResponseZAnyZDictZImmutableTypeConversionDictZImmutableMultiDictZ FileStorageZUnion�Callabler] � ImportErrorr r r r r Z flask.signalsr r r r r r( r) r* rF rB r+ rR r r r r �<module> s( %