Edit File: wsgi.cpython-37.opt-1.pyc
B Z~�_�( � @ s� d dl Z d dlmZ d dlmZmZ d dlmZmZm Z d dl m Z m Z m Z d dlmZ d dlmZ d dlmZ d d lmZ e�r"d d lmZ d d lmZ d d lmZ d d lmZ d dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d dlm Z e d�Z!e d�Z"e d�Z#G dd� de �Z$ed�Z%e �r<d3d d �Z&n d4d d �Z&d d � Z'd!d"� Z(G d#d$� d$e)�Z*d5d%d&�Z+d'd(� Z,d)d*� Z-d+d,� Z.d-d.� Z/G d/d0� d0e)�Z0d1d2� Z1dS )6� N)�partial)�Hub�_should_send_default_pii)� ContextVar�capture_internal_exceptions�event_from_exception)�PY2�reraise� iteritems)� Transaction)�auto_session_tracking)�_filter_headers)�MYPY)�Callable)�Dict)�Iterator)�Any)�Tuple)�Optional)�TypeVar)�Protocol)�ExcInfo)�EventProcessor�WsgiResponseIter�WsgiResponseHeaders� WsgiExcInfoc @ s e Zd Zddd�ZdS )� StartResponseNc C s d S )N� )�self�status�response_headers�exc_infor r �M/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/wsgi.py�__call__# s zStartResponse.__call__)N)�__name__� __module__� __qualname__r# r r r r" r " s r Z sentry_wsgi_middleware_applied�utf-8�replacec C s | � ||�S )N)�decode)�s�charset�errorsr r r" �wsgi_decoding_dance- s r- c C s | � d��||�S )N�latin1)�encoder) )r* r+ r, r r r" r- 4 s c C s� | � d�rZ| d }| d dkr6|�d�r6|dd� }q�| d dkr�|�d�r�|dd � }n<| � d �r�| d }| d | d fd kr�|d | d 7 }nd}|S )zEReturn the host for the given WSGI environment. Yanked from Werkzeug.Z HTTP_HOSTzwsgi.url_scheme�httpz:80N����httpsz:443���� SERVER_NAME� SERVER_PORT))r2 Z443)r0 Z80�:�unknown)�get�endswith)�environ�rvr r r" �get_host9 s r< c C s, d| � d�t| �t| � d�p d��d�f S )zPReturn the absolute URL without query string for the given WSGI environment.z %s://%s/%szwsgi.url_schemeZ PATH_INFO� �/)r8 r< r- �lstrip)r: r r r" �get_request_urlP s r@ c @ s e Zd ZdZdd� Zdd� ZdS )�SentryWsgiMiddleware)�appc C s || _ d S )N)rB )r rB r r r" �__init__^ s z SentryWsgiMiddleware.__init__c C s t �d�r| �||�S t �d� z�ttj�}t|��� |�� t� �6 |�� �"}|� � d|_ |� t |�� W d Q R X W d Q R X t j|ddd�}|j|d|id��@ y| �|tt||��}W n tk r� tt|�� Y nX W d Q R X W d Q R X W d Q R X W d t �d� X t||�S ) NFT�wsgiz http.serverzgeneric WSGI request)�op�nameZ wsgi_environ)Zcustom_sampling_context)�_wsgi_middleware_appliedr8 rB �setr �currentr r Zconfigure_scopeZclear_breadcrumbs�_nameZadd_event_processor�_make_wsgi_event_processorr Zcontinue_from_environZstart_transactionr �_sentry_start_response� BaseExceptionr �_capture_exception�_ScopedResponse)r r: Zstart_response�hubZscope� transactionr; r r r" r# b s4 4 z SentryWsgiMiddleware.__call__N)r$ r% r& � __slots__rC r# r r r r" rA [ s rA c C sR t � �$ t|�dd�d �}|�|� W d Q R X |d krB| ||�S | |||�S d S )N� � r )r �int�splitZset_http_status)Zold_start_responserQ r r r! Z status_intr r r" rL � s rL c c s@ ddg}t � r|dg7 }x"|D ]}|| kr || | fV q W dS )z� Returns our explicitly included environment variables we want to capture (server name, port and remote addr if pii is enabled). r4 r5 � REMOTE_ADDRN)r )r: �keys�keyr r r" � _get_environ� s rZ c c sr xlt | �D ]`\}}t|�}|�d�rL|dkrL|dd� �dd��� |fV q |dkr |�dd��� |fV q W dS )z, Returns only proper HTTP headers. ZHTTP_)ZHTTP_CONTENT_TYPEZHTTP_CONTENT_LENGTH� N�_�-)Z CONTENT_TYPEZCONTENT_LENGTH)r �str� startswithr( �title)r: rY �valuer r r" � _get_headers� s rb c C sX y| d � d�d �� S ttfk r. Y nX y| d S tk rL Y nX | �d�S )z� Infer the user IP address from various headers. This cannot be used in security sensitive situations since the value may be forged from a client, but it's good enough for the event payload. ZHTTP_X_FORWARDED_FOR�,r ZHTTP_X_REAL_IPrW )rV �strip�KeyError� IndexErrorr8 )r: r r r" � get_client_ip� s rg c C s` t �� }| jd k r\|d }t|t�o,|jdk}|s\t|| jjddd�d�\}}| j||d� |S )NrT )r NrD F)�typeZhandled)Zclient_optionsZ mechanism)�hint) �sysr! Zclient� isinstance� SystemExit�coder �optionsZ capture_event)rP r! �eZshould_skip_capture�eventri r r r" rN � s rN c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) rO )� _response�_hubc C s || _ || _d S )N)rr rq )r rP Zresponser r r" rC � s z_ScopedResponse.__init__c c sp t | j�}x`| j�J y t|�}W n6 tk r6 P Y n" tk rV tt| j�� Y nX W d Q R X |V q W d S )N)�iterrq rr �next� StopIterationrM r rN )r �iterator�chunkr r r" �__iter__� s z_ScopedResponse.__iter__c C sZ | j �J y| j�� W n4 tk r* Y n" tk rJ tt| j �� Y nX W d Q R X d S )N)rr rq �close�AttributeErrorrM r rN )r r r r" ry s z_ScopedResponse.closeN)r$ r% r& rR rC rx ry r r r r" rO � s rO c sZ t | �� t| ��| �d��| �d��tt| ���ttt| ����� �����fdd�}|S )NZ QUERY_STRINGZREQUEST_METHODc sl t � �\ | �di �}t� r6| �di �}� r6|�d� � �|d< �|d< �|d< �|d< �|d< W d Q R X | S ) NZrequest�userZ ip_addressZurl� query_string�method�env�headers)r � setdefaultr )rp ri Z request_infoZ user_info)� client_ipr~ r r} r| � request_urlr r" �event_processor% s z3_make_wsgi_event_processor.<locals>.event_processor)rg r@ r8 �dictrZ r rb )r: r� r )r� r~ r r} r| r� r" rK s rK )r' r( )r' r( )N)2rj Zsentry_sdk._functoolsr Zsentry_sdk.hubr r Zsentry_sdk.utilsr r r Zsentry_sdk._compatr r r Zsentry_sdk.tracingr Zsentry_sdk.sessionsr Z$sentry_sdk.integrations._wsgi_commonr Zsentry_sdk._typesr �typingr r r r r r r r r r r r r r rG r- r< r@ �objectrA rL rZ rb rg rN rO rK r r r r" �<module> sJ 4 "