관리-도구
편집 파일: flask.cpython-38.pyc
U �Mg� � @ s� d dl Z d dlmZmZ d dlmZmZ d dlmZ d dl m Z d dl m Z d dl mZmZmZmZ d dlmZ er�d d lmZmZmZmZ d d lmZmZ d d lmZ d d l m Z m Z z d dl Z W n e k r� dZ Y nX z@d d l!m"Z"m#Z# d dl!m$Z% d dl&m'Z'm(Z(m)Z) d dl*m+Z+ W n e k �r< ed��Y nX z d dl,Z,W n e k �rh ed��Y nX dZ-G dd� de�Z.dd� Z/dd� Z0dd� Z1G d d � d e�Z2d d � Z3ee.�d d!� �Z4d"d#� Z5dS )$� N)� DidNotEnable� Integration)� DEFAULT_HTTP_METHODS_TO_CAPTURE�RequestExtractor)�SentryWsgiMiddleware)�should_send_default_pii)�SOURCE_FOR_STYLE)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�package_version)� TYPE_CHECKING)�Any�Callable�Dict�Union)�Event�EventProcessor)�_ScopedResponse)� FileStorage�ImmutableMultiDict)�Flask�Request��request)�before_render_template�got_request_exception�request_started)�MarkupzFlask is not installedzblinker is not installed)�endpoint�urlc @ s8 e Zd ZdZde� �ZdZdefdd�Zedd� �Z d S ) �FlaskIntegration�flaskz auto.http.� r c C s4 |t krtd|t f ��|| _tttj|��| _d S )Nz7Invalid value for transaction_style: %s (must be in %s))�TRANSACTION_STYLE_VALUES� ValueError�transaction_style�tuple�map�str�upper�http_methods_to_capture)�selfr& r+ � r- �K/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/flask.py�__init__: s ��zFlaskIntegration.__init__c sb t d�} | d krtd��| dk r(td��t�t� t�t� t�t� t j � � fdd�}|t _ d S )Nr" zUnparsable Flask version.)r � z Flask 0.10 or newer is required.c sZ t �� �t�d kr �� ||�S t �� �t�}t�� fdd�tj|rH|jntd�}|||�S )Nc s � �f| �|�S �Nr- )�a�kw)�old_appr, r- r. �<lambda>a � zNFlaskIntegration.setup_once.<locals>.sentry_patched_wsgi_app.<locals>.<lambda>)Z span_originr+ )� sentry_sdk� get_client�get_integrationr! r �originr+ r )r, �environZstart_response� integrationZ middleware�r4 �r, r. �sentry_patched_wsgi_appY s �� z<FlaskIntegration.setup_once.<locals>.sentry_patched_wsgi_app) r r r �connect�_add_sentry_tracer �_request_startedr �_capture_exceptionr �__call__)�versionr? r- r= r. � setup_onceH s zFlaskIntegration.setup_onceN) �__name__� __module__� __qualname__� identifierr: r& r r/ � staticmethodrF r- r- r- r. r! 4 s � r! c K s4 d|kr d S t �� }t|�� �}||d< ||d< d S )NZ sentry_traceZsentry_trace_meta)r7 �get_current_scoper Ztrace_propagation_meta)�sender�template�context�extra�scopeZ trace_metar- r- r. rA n s rA c C sF z,|j j|j jd�}| j|| t| d� W n tk r@ Y nX d S )N)r r )�source)Zurl_rule�ruler Zset_transaction_namer � Exception)rQ r& r Zname_for_styler- r- r. � _set_transaction_name_and_sourcey s �� rU c K sV t �� �t�}|d krd S t�� }tt �� |j|� t � � }t | ||�}|� |� d S r1 ) r7 r8 r9 r! � flask_requestZ_get_current_objectrU rL r& Zget_isolation_scope� _make_request_event_processorZadd_event_processor)�app�kwargsr< r rQ Z evt_processorr- r- r. rB � s � rB c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�FlaskRequestExtractorc C s | j jS r1 )r r; r> r- r- r. �env� s zFlaskRequestExtractor.envc C s dd� | j j�� D �S )Nc S s4 i | ],\}}|t |t�r,t|�d kr,|d n|�qS )� r )� isinstance�list�len)�.0�k�vr- r- r. � <dictcomp>� s � z1FlaskRequestExtractor.cookies.<locals>.<dictcomp>)r �cookies�itemsr> r- r- r. rd � s �z FlaskRequestExtractor.cookiesc C s | j �� S r1 )r �get_datar> r- r- r. �raw_data� s z FlaskRequestExtractor.raw_datac C s | j jS r1 )r �formr> r- r- r. rh � s zFlaskRequestExtractor.formc C s | j jS r1 )r �filesr> r- r- r. ri � s zFlaskRequestExtractor.filesc C s | j jS r1 )r �is_jsonr> r- r- r. rj � s z FlaskRequestExtractor.is_jsonc C s | j jdd�S )NT)Zsilent)r Zget_jsonr> r- r- r. �json� s zFlaskRequestExtractor.jsonc C s |j S r1 )�content_length)r, �filer- r- r. � size_of_file� s z"FlaskRequestExtractor.size_of_fileN) rG rH rI r[ rd rg rh ri rj rk rn r- r- r- r. rZ � s rZ c s � fdd�}|S )Nc sP � d kr | S t � � t� ��| � W 5 Q R X t� rLt � � t| � W 5 Q R X | S r1 )r rZ Zextract_into_eventr �_add_user_to_event)�event�hintr r- r. �inner� s z,_make_request_event_processor.<locals>.innerr- )rX r r< rr r- r r. rW � s rW c K s0 t |t�� jddd�d�\}}tj||d� d S )Nr" F)�typeZhandled)Zclient_optionsZ mechanism)rq )r r7 r8 �optionsZ capture_event)rM � exceptionrY rp rq r- r- r. rC � s � rC c C s� t d kr d S t j}|d kr d S t� �� | �di �}z|�d|�� � W n tk rZ Y nX z|�d|j� W n tk r� Y nX z|�d|j� W n tk r� Y nX W 5 Q R X d S )N�user�id�email�username) � flask_loginZ current_userr � setdefaultZget_id�AttributeErrorrx rT ry )rp rv Z user_infor- r- r. ro � s& ro )6r7 Zsentry_sdk.integrationsr r Z$sentry_sdk.integrations._wsgi_commonr r Z sentry_sdk.integrations.wsgir Zsentry_sdk.scoper Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r �typingr r r r r Zsentry_sdk._typesr r r Zwerkzeug.datastructuresr r rz � ImportErrorr" r r r rV Z flask.signalsr r r Z markupsafer Zblinkerr$ r! rA rU rB rZ rW rC ro r- r- r- r. �<module> sJ : %