관리-도구
편집 파일: pyramid.cpython-38.pyc
U �Mg� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dlm Z d dl m Z d dl m Z d dlmZ d dlmZmZmZmZ z d dlmZ d d lmZ W n ek r� ed ��Y nX d d lmZ e�rHd d l m Z d d lm Z d dl m Z d dlm Z d dlm!Z! d dlm"Z" d dl#m$Z$ d dl%m&Z& d dlm'Z' d dl(m)Z)m*Z* e+edd��r`dd� Z,n d dl-m,Z, dZ.G dd� de�Z/ee/�d d � �Z0d d � Z1G d d!� d!e �Z2d"d#� Z3dS )$� N)� Integration� DidNotEnable)�RequestExtractor)�SentryWsgiMiddleware)�should_send_default_pii)�SOURCE_FOR_STYLE)�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�reraise)� HTTPException)�RequestzPyramid not installed)� TYPE_CHECKING)�Response)�Any)�_ScopedResponse)�Callable)�Dict)�Optional)�RequestCookies)�_FieldStorageWithFile)�ExcInfo)�Event�EventProcessor�authenticated_useridc C s | j S �N�r )�request� r �M/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/pyramid.pyr + s r �� route_nameZ route_patternc @ s4 e Zd ZdZde� �ZdZd dd�Zedd� �Zd S ) �PyramidIntegration�pyramidz auto.http.� r! 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 r r �__init__= s ��zPyramidIntegration.__init__c sz ddl m} | j� t�� �� fdd��}|| _ttd�rPtj��fdd�}|t_| jj �t t ���fdd ��}|| j_ d S ) Nr )�routerc sj t �� �t�}|d kr(� | |f|�|�S tt �� |j|� t �� }|�t t � |�|�� � | |f|�|�S r ) � sentry_sdk� get_client�get_integrationr" � _set_transaction_name_and_sourceZget_current_scoper' Zget_isolation_scopeZadd_event_processor�_make_event_processor�weakref�ref)�registryr �args�kwargs� integration�scope)� old_call_viewr r �sentry_patched_call_viewM s ��z?PyramidIntegration.setup_once.<locals>.sentry_patched_call_view�invoke_exception_viewc sJ � | f|�|�}| j rFt| j �rF|jdkrFt�� �t�d k rFt| j � |S )Ni� )�exc_info�allZ status_intr+ r, r- r" �_capture_exception)r( r3 r4 �rv)�old_invoke_exception_viewr r �$sentry_patched_invoke_exception_viewc s ��� �� zKPyramidIntegration.setup_once.<locals>.sentry_patched_invoke_exception_viewc s& �� fdd�}t |tjd�}|||�S )Nc s@ z� �| |�W S t k r: t�� }t|� t|� Y nX d S r )� Exception�sysr: r< r )�environ�start_responseZeinfo)� old_wsgi_callr( r r � sentry_patched_inner_wsgi_cally s zgPyramidIntegration.setup_once.<locals>.sentry_patched_wsgi_call.<locals>.sentry_patched_inner_wsgi_call)Z span_origin)r r" �origin)r( rB rC rE Z middleware)rD �r( r �sentry_patched_wsgi_callv s �z?PyramidIntegration.setup_once.<locals>.sentry_patched_wsgi_call) r# r* Z _call_view� functools�wraps�hasattrr r9 ZRouter�__call__r r" )r* r8 r? rH r )r7 r> rD r � setup_onceF s z PyramidIntegration.setup_onceN)r! ) �__name__� __module__� __qualname__� identifierrF r' r) � staticmethodrM r r r r r" 7 s r" c C sN | d d kst | d t�r d S t| t�� jddd�d�\}}tj||d� d S )Nr r# F)�typeZhandled)Zclient_optionsZ mechanism)�hint)� issubclassr r r+ r, �optionsZ capture_event)r: �eventrT r r r r< � s � r< c C sF z,|j j|j jd�}| j|| t| d� W n tk r@ Y nX d S )Nr )�source)Z matched_route�name�patternZset_transaction_namer r@ )r6 r' r Zname_for_styler r r r. � s �� r. c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�PyramidRequestExtractorc C s | j jS r )r �path_urlrG r r r �url� s zPyramidRequestExtractor.urlc C s | j jS r )r rB rG r r r �env� s zPyramidRequestExtractor.envc C s | j jS r )r �cookiesrG r r r r_ � s z PyramidRequestExtractor.cookiesc C s | j jS r )r �textrG r r r �raw_data� s z PyramidRequestExtractor.raw_datac C s dd� | j j�� D �S )Nc S s" i | ]\}}t |d d�s||�qS ��filenameN��getattr��.0�key�valuer r r � <dictcomp>� s � z0PyramidRequestExtractor.form.<locals>.<dictcomp>�r �POST�itemsrG r r r �form� s �z PyramidRequestExtractor.formc C s dd� | j j�� D �S )Nc S s" i | ]\}}t |d d�r||�qS rb rd rf r r r rj � s � z1PyramidRequestExtractor.files.<locals>.<dictcomp>rk rG r r r �files� s �z PyramidRequestExtractor.filesc C s4 |j }zt�|�� �jW S tk r. Y dS X d S )Nr )�file�os�fstat�fileno�st_sizer@ )r( Zpostdatarp r r r � size_of_file� s z$PyramidRequestExtractor.size_of_fileN) rN rO rP r] r^ r_ ra rn ro ru r r r r r[ � s r[ c s � fdd�}|S )Nc sj � � }|d kr| S t � � t|��| � W 5 Q R X t� rft � �" | �di �}|�dt|�� W 5 Q R X | S )N�user�id)r r[ Zextract_into_eventr � setdefaultr )rW rT r Z user_info�� weak_requestr r �pyramid_event_processor� s z6_make_event_processor.<locals>.pyramid_event_processorr )rz r5 r{ r ry r r/ � s r/ )4rI rq rA r0 r+ Zsentry_sdk.integrationsr r Z$sentry_sdk.integrations._wsgi_commonr Z sentry_sdk.integrations.wsgir Zsentry_sdk.scoper Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r Zpyramid.httpexceptionsr Zpyramid.requestr � ImportError�typingr Zpyramid.responser r r r r r Z webob.cookiesr Z webob.requestr r Zsentry_sdk._typesr r re r Zpyramid.securityr% r" r< r. r[ r/ r r r r �<module> sJ T *