OwlCyberSecurity - MANAGER
Edit File: pyramid.cpython-38.pyc
U ��\ � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dl m Z m Z d dl m Z mZ d dlmZ d dlmZ d d lmZ d d lmZ e ed d�r�d d � Z!n d d l"m!Z! G dd� de�Z#dd� Z$G dd� de�Z%dd� Z&dS )� )�absolute_importN)� HTTPException)�Request)�Hub�_should_send_default_pii)�capture_internal_exceptions�event_from_exception)�reraise)� Integration)�RequestExtractor)�SentryWsgiMiddleware�authenticated_useridc C s | j S �N�r )�request� r �P/opt/alt/python38/lib/python3.8/site-packages/sentry_sdk/integrations/pyramid.pyr ! s r c @ s* e Zd ZdZdZddd�Zedd� �ZdS ) �PyramidIntegration�pyramidN� route_namec C s& d}||kr t d||f ��|| _d S )N)r � route_patternz7Invalid value for transaction_style: %s (must be in %s))� ValueError�transaction_style)�selfr ZTRANSACTION_STYLE_VALUESr r r �__init__0 s ��zPyramidIntegration.__init__c s@ ddl m} | j� � fdd�}|| _| j��fdd�}|| _d S )Nr )�Routerc s� t j}|�t�}|d kr*� | |f|�|�S |�� � }|�tt�|�|�� W 5 Q R X z� | |f|�|�W S t k r� t � � }t |� t |� Y nX d S r )r �current�get_integrationr Zconfigure_scopeZadd_event_processor�_make_event_processor�weakref�ref� Exception�sys�exc_info�_capture_exceptionr )r r �args�kwargs�hub� integrationZscoper# )�old_handle_requestr r � sentry_patched_handle_requestA s �zDPyramidIntegration.setup_once.<locals>.sentry_patched_handle_requestc s< t j}|�t�}|d kr$�� ||�S t�� fdd��||�S )Nc s � �f| �|�S r r )�a�kw)� old_wsgi_callr r r �<lambda>_ � zQPyramidIntegration.setup_once.<locals>.sentry_patched_wsgi_call.<locals>.<lambda>)r r r r r )r �environZstart_responser' r( )r- �r r �sentry_patched_wsgi_callX s �z?PyramidIntegration.setup_once.<locals>.sentry_patched_wsgi_call)Zpyramid.routerr Zhandle_request�__call__)r r* r2 r )r) r- r � setup_once: s z PyramidIntegration.setup_once)r )�__name__� __module__� __qualname__� identifierr r � staticmethodr4 r r r r r + s r c K sd | d d kst | d t�r d S tj}|�t�d kr6d S t| |jjddd�d�\}}|j ||d� d S )Nr r F)�typeZhandled)Zclient_optionsZ mechanism)�hint) � issubclassr r r r r r Zclient�optionsZ capture_event)r# r& r' �eventr; r r r r$ f 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 Zpath_urlr1 r r r �urlw s zPyramidRequestExtractor.urlc C s | j jS r )r r0 r1 r r r �envz s zPyramidRequestExtractor.envc C s | j jS r )r �cookiesr1 r r r rB ~ s z PyramidRequestExtractor.cookiesc C s | j jS r )r �textr1 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 ZPOST�itemsr1 r r r �form� s �z PyramidRequestExtractor.formc C s dd� | j j�� D �S )Nc S s" i | ]\}}t |d d�r||�qS rE rG rI r r r rM � s � z1PyramidRequestExtractor.files.<locals>.<dictcomp>rN r1 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 ZpostdatarR r r r � size_of_file� s z$PyramidRequestExtractor.size_of_fileN) r5 r6 r7 r@ rA rB rD rP rQ rW r r r r r? v s r? 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� � | � di �}t |�|d< W 5 Q R X | S )Nr Z transactionr �user�id) r Z matched_route�name�patternr! r r? Zextract_into_eventr � setdefaultr )r> r; r Z user_info�r( � weak_requestr r �event_processor� s" z._make_event_processor.<locals>.event_processorr )r^ r( r_ r r] r r � s r )'� __future__r rS r" r Zpyramid.httpexceptionsr Zpyramid.requestr Zsentry_sdk.hubr r Zsentry_sdk.utilsr r Zsentry_sdk._compatr Zsentry_sdk.integrationsr Z$sentry_sdk.integrations._wsgi_commonr Z sentry_sdk.integrations.wsgir ZResponseZAnyZ_ScopedResponse�Callable�Dict�OptionalZRequestCookiesZcgi_FieldStorageZExcInforH r Zpyramid.securityr r$ r? r r r r r �<module> s$ ;)