관리-도구
편집 파일: wsgi.cpython-38.pyc
U �Mg* � @ s� d dl Z d dlmZ d dlZd dlmZmZ d dlmZ d dl m Z d dl m Z d dl mZmZmZ d dlmZ d d l mZ d d lmZmZ d d lmZmZmZmZ d d l m Z e �rZd d l m Z d dl m Z d dl m Z d dl m!Z! d dl m"Z" d dl m#Z# d dl m$Z$ d dl m%Z% d dlm&Z& d dl'm(Z(m)Z) e$d�Z*e$d�Z+e$d�Z,G dd� de%�Z-ed �Z.d2d d �Z/d3d"d#�Z0G d$d%� d%�Z1d4d&d'�Z2d(d)� Z3d*d+� Z4d,d-� Z5G d.d/� d/�Z6d0d1� Z7dS )5� N)�partial)�get_host� _get_headers)�continue_trace)�OP��should_send_default_pii)� DEFAULT_HTTP_METHODS_TO_CAPTURE�_filter_headers� nullcontext)� track_session)�use_isolation_scope)� Transaction�TRANSACTION_SOURCE_ROUTE)� ContextVar�capture_internal_exceptions�event_from_exception�reraise)� TYPE_CHECKING)�Callable)�Dict)�Iterator)�Any)�Tuple)�Optional)�TypeVar)�Protocol)�ExcInfo)�Event�EventProcessor�WsgiResponseIter�WsgiResponseHeaders� WsgiExcInfoc @ s e Zd Zddd�ZdS )� StartResponseNc C s d S �N� )�self�status�response_headers�exc_infor% r% �J/opt/hc_python/lib/python3.8/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 | � d��||�S )N�latin1)�encode�decode)�s�charset�errorsr% r% r* �wsgi_decoding_dance4 s r7 Fc C sV | � dd��d�}| � dd��d�}|� d|� �}d| � d�t| |�t|��d�f S )zPReturn the absolute URL without query string for the given WSGI environment.Z SCRIPT_NAME� �/Z PATH_INFOz %s://%s/%szwsgi.url_scheme)�get�rstrip�lstripr r7 )�environ�use_x_forwarded_for� script_nameZ path_info�pathr% r% r* �get_request_url9 s �rA c @ s( e Zd ZdZddefdd�Zdd� ZdS ) �SentryWsgiMiddleware��appr> � span_origin�http_methods_to_captureFZmanualc C s || _ || _|| _|| _d S r$ rC )r& rD r> rE rF r% r% r* �__init__P s z SentryWsgiMiddleware.__init__c C s2 t �d�r| �||�S t �d� z�t�� ��}t|dd��� t� �& |�� d|_ |� t || j �� W 5 Q R X |�dd�� � }d }|| jkr�t|tjdt| jd �}|d k r�tj|d |id �nt� �> z| �|tt||��}W n tk r� tt� � Y nX W 5 Q R X W 5 Q R X W 5 Q R X W 5 t �d� X t||�S ) NFT�request)Z session_mode�wsgi�REQUEST_METHODr8 zgeneric WSGI request)�op�name�source�originZ wsgi_environ)Zcustom_sampling_context) �_wsgi_middleware_appliedr: rD �set� sentry_sdkZisolation_scoper r Zclear_breadcrumbs�_nameZadd_event_processor�_make_wsgi_event_processorr> �upperrF r r Z HTTP_SERVERr rE Zstart_transactionr r �_sentry_start_response� BaseExceptionr �_capture_exception�_ScopedResponse)r& r= Zstart_response�scope�method� transaction�responser% r% r* r+ ] sZ �� � ��� ��2 z SentryWsgiMiddleware.__call__N)r, r- r. � __slots__r rG r+ r% r% r% r* rB H s � rB c C sZ t � �, t|�dd�d �}|d k r.|�|� W 5 Q R X |d krJ| ||�S | |||�S d S )N� � r )r �int�splitZset_http_status)Zold_start_responser[ r' r( r) Z status_intr% r% r* rU � s rU c c s< ddg}t � r|dg7 }|D ]}|| kr || | fV q dS )z� Returns our explicitly included environment variables we want to capture (server name, port and remote addr if pii is enabled). Z SERVER_NAMEZ SERVER_PORT� REMOTE_ADDRNr )r= �keys�keyr% r% r* � _get_environ� s re c C s\ z| d � d�d �� W S ttfk r0 Y nX z | d W S tk rP 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_IPrb )ra �strip�KeyError� IndexErrorr: )r= r% r% r* � get_client_ip� s rj c C sX t �� } | d }t|t�o"|jdk}|sTt| t�� jddd�d�\}}tj ||d� | S )z� Captures the current exception and sends it to Sentry. Returns the ExcInfo tuple to it can be reraised afterwards. r_ )r NrI F)�typeZhandled)Zclient_optionsZ mechanism)�hint) �sysr) � isinstance� SystemExit�coder rQ Z get_client�optionsZ capture_event)r) �eZshould_skip_capture�eventrl r% r% r* rW � s � rW c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S ) rX aY Users a separate scope for each response chunk. This will make WSGI apps more tolerant against: - WSGI servers streaming responses from a different thread/from different threads than the one that called start_response - close() not being called - WSGI servers streaming responses interleaved from the same thread )� _response�_scopec C s || _ || _d S r$ )ru rt )r& rY r\ r% r% r* rG � s z_ScopedResponse.__init__c c sx t | j�}t| j��R z t|�}W n> tk rD Y W 5 Q R � qtY n tk r` tt� � Y nX W 5 Q R X |V q d S r$ ) �iterrt r ru �next� StopIterationrV r rW )r& �iterator�chunkr% r% r* �__iter__� s z_ScopedResponse.__iter__c C sZ t | j��F z| j�� W n0 tk r. Y n tk rJ tt� � Y nX W 5 Q R X d S r$ )r ru rt �close�AttributeErrorrV r rW )r& r% r% r* r| s z_ScopedResponse.closeN)r, r- r. �__doc__r] rG r{ r| r% r% r% r* rX � s rX c s\ t | �� t| |��| �d��| �d��tt| ���ttt| ����� �����fdd�}|S )NZ QUERY_STRINGrJ c sl t � �\ | �di �}t� r6| �di �}� r6|�d� � �|d< �|d< �|d< �|d< �|d< W 5 Q R X | S ) NrH �user� ip_address�url� query_stringrZ �env�headers)r � setdefaultr )rs rl Z request_infoZ user_info�Z client_ipr� r� rZ r� � request_urlr% r* �event_processor# s z3_make_wsgi_event_processor.<locals>.event_processor)rj rA r: �dictre r r )r= r> r� r% r� r* rS s rS )r/ r0 )F)N)8rm � functoolsr rQ Zsentry_sdk._werkzeugr r Zsentry_sdk.apir Zsentry_sdk.constsr Zsentry_sdk.scoper Z$sentry_sdk.integrations._wsgi_commonr r r Zsentry_sdk.sessionsr r Zsentry_sdk.tracingr r Zsentry_sdk.utilsr r r r �typingr r r r r r r r r r Zsentry_sdk._typesr r r r! r" r# rO r7 rA rB rU re rj rW rX rS r% r% r% r* �<module> sL N � ,