OwlCyberSecurity - MANAGER
Edit File: flask.cpython-38.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 �connect� _push_appctxr � _pop_appctxr �_request_startedr �_capture_exceptionr �__call__)r' r r% r � setup_once2 s zFlaskIntegration.setup_once)r )�__name__� __module__� __qualname__� 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 )Nr6 )�getattrr r5 �__exit__)r9 r: 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 r5 �requestr �appr7 Zadd_event_processor� _make_request_event_processor)�senderr: r; � integration� weak_requestrC 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 )rB r$ r& r r r �envn s zFlaskRequestExtractor.envc C s | j jS r )rB �cookiesr& r r r rJ r s z FlaskRequestExtractor.cookiesc C s | j jS r )rB �datar& r r r �raw_datav s z FlaskRequestExtractor.raw_datac C s | j jS r )rB �formr& r r r rM z s zFlaskRequestExtractor.formc C s | j jS r )rB �filesr& r r r rN ~ s zFlaskRequestExtractor.filesc C s | j jS r )rB �is_jsonr& r r r rO � s z FlaskRequestExtractor.is_jsonc C s | j �� S r )rB 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/ r0 r1 rI rJ rL rM rN rO rP rR r r r r rH m s rH 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 rH Zextract_into_eventr �_add_user_to_event)�event�hintrB �rF rG r r �inner� s z,_make_request_event_processor.<locals>.innerr )rC rG rF rX r rW r rD � s rD c K sF t j}|�t�d krd S t||jjddd�d�\}}|j||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)rV )r r" r# r r Zclient�optionsZ capture_event)rE � exceptionr: r; rU rV 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^ rS r_ )rU r\ Z user_infor r r rT � s& rT )+� __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_ScopedResponseZAny�DictZImmutableTypeConversionDictZImmutableMultiDictZ FileStorage�Union�Callabler` � ImportErrorr r r r r Z flask.signalsr r r r r r) r* r+ rH rD r, rT r r r r �<module> s( %