관리-도구
편집 파일: starlette.cpython-38.pyc
U �Mg�d � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlZd dlm Z d dl m Z m Z m Z d dlmZmZmZmZ d dlmZ d dlmZ d d lmZmZmZ d d lm Z m Z m Z m Z m Z m!Z!m"Z" d d l#m$Z$ e$�r d d l#m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z, d d l-m.Z.m/Z/ zxd dl0Z0d dl0m1Z2 d dl3m4Z4 d dl5m6Z6 d dl7m8Z8 d dl9m:Z: d dl;m<Z< d dl=m>Z> d dl?m@Z@mAZAmBZCmDZD W n eEk �r� e d��Y nX zd dlFmGZG W n" eEk �r� d dlHmGZG Y nX z d dlIZIW n eEk �r� dZIY nX dZJdZKG dd� de �ZLd d � ZMe eL�d9d d ��ZNd!d"� ZOe eL�d#d$� �ZPd%d&� ZQd'd(� ZRd)d*� ZSd+d,� ZTd-d.� ZUd/d0� ZVG d1d2� d2�ZWd3d4� ZXd5d6� ZYd7d8� ZZdS ):� N)�Set)�deepcopy)�OP)� DidNotEnable� Integration�$_DEFAULT_FAILED_REQUEST_STATUS_CODES)� DEFAULT_HTTP_METHODS_TO_CAPTURE�HttpCodeRangeContainer�_is_json_content_type�request_body_within_bounds)�SentryAsgiMiddleware)�should_send_default_pii)�SOURCE_FOR_STYLE� TRANSACTION_SOURCE_COMPONENT�TRANSACTION_SOURCE_ROUTE)�AnnotatedValue�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�logger� parse_version�transaction_from_function)� TYPE_CHECKING)�Any� Awaitable�Callable� Container�Dict�Optional�Tuple�Union)�Event�HttpStatusCodeRange)� __version__)� Starlette)� UploadFile)� Middleware)�AuthenticationMiddleware)�Request)�Match)�ASGIApp�Receive�Scope�SendzStarlette is not installed)�ExceptionMiddlewarezgeneric Starlette request)�endpoint�urlc @ s< e Zd ZdZde� �ZdZdedefdd�Ze dd � �Z d S ) �StarletteIntegration� starlettez auto.http.� r0 Tc C sv |t krtd|t f ��|| _|| _tttj|��| _t |t �rH|| _ n*t j dtdd� |d krht| _ n t|�| _ d S )Nz7Invalid value for transaction_style: %s (must be in %s)zgPassing a list or None for failed_request_status_codes is deprecated. Please pass a set of int instead.� )� stacklevel)�TRANSACTION_STYLE_VALUES� ValueError�transaction_style�middleware_spans�tuple�map�str�upper�http_methods_to_capture� isinstancer �failed_request_status_codes�warnings�warn�DeprecationWarningr r )�selfr8 r@ r9 r> � rE �O/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/starlette.py�__init__T s, �� ���z StarletteIntegration.__init__c C sB t t�} | d kr td�t���t� t� t� | dkr>t� d S )Nz Unparsable Starlette version: {})r � )r �STARLETTE_VERSIONr �format�patch_middlewares�patch_asgi_app�patch_request_response�patch_templates)�versionrE rE rF � setup_oncex s �z StarletteIntegration.setup_onceN) �__name__� __module__� __qualname__� identifier�originr8 r r rG � staticmethodrP rE rE rE rF r1 N s � $r1 c s* | j � � fdd�}� jdk}|r&|| _ | S )Nc � s t �� �t�}|d ks |js4�| |��f|�I d H S | jj� t| ||�\}}|d k rft �� j ||d� t j t j � tj d���}|�d� � � �fdd�} t�dt���} | dk} | s�| n�} � �fdd �} t�dt���}|d k}|s�| n�}�| || |f|�I d H W 5 Q R � S Q R X d S ) N��source��op�namerU �starlette.middleware_namec � sT t jtjt�dt���tjd��*}|�d� � �| |�I d H W 5 Q R � S Q R X d S �NrS rY r\ ) � sentry_sdk� start_spanr Z MIDDLEWARE_STARLETTE_RECEIVE�getattrr< r1 rU �set_tag��args�kwargs�span)�middleware_name�receiverE rF �_sentry_receive� s � zO_enable_span_for_middleware.<locals>._create_span_call.<locals>._sentry_receiverQ rh c � sT t jtjt�dt���tjd��*}|�d� � �| |�I d H W 5 Q R � S Q R X d S r] ) r^ r_ r ZMIDDLEWARE_STARLETTE_SENDr` r< r1 rU ra rb )rf �sendrE rF � _sentry_send� s � zL_enable_span_for_middleware.<locals>._create_span_call.<locals>._sentry_sendrj )r^ � get_client�get_integrationr1 r9 � __class__rQ � _get_transaction_from_middleware�get_current_scope�set_transaction_namer_ r ZMIDDLEWARE_STARLETTErU ra r` r< )�app�scoperg ri rd � integrationr[ rX Zmiddleware_spanrh Z receive_nameZreceive_patchedZ new_receiverj Z send_nameZ send_patchedZnew_send��old_call)rf rg ri rF �_create_span_call� s4 �� z6_enable_span_for_middleware.<locals>._create_span_call)rv �%_sentry_authenticationmiddleware_call� _sentry_exceptionmiddleware_call)�__call__rQ )�middleware_classrv �not_yet_patchedrE rt rF �_enable_span_for_middleware� s 7 r| Fc C s2 t | t�� jtj|d�d�\}}tj||d� d S )N)�type�handled)Zclient_optionsZ mechanism)�hint)r r^ rk �optionsr1 rT Z capture_event)� exceptionr~ �eventr rE rE rF �_capture_exception� s � r� c sD | j �dt��k}|r@�fdd�}|| _ | j� � fdd�}|| _dS )zX Capture all exceptions in Starlette app and also extract user information. �_sentry_middleware_initc sD �| f|�|� | j �� � � fdd�}| j �� D ]}|| j |< q0d S )Nc � s� t �� �t�}|d }|d k rPt|d�o>t|jt�o>|j|jk}|rPt |dd� d }t |�j D ]}|� kr^� | } qxq^|d kr�d S t |�r�|| f|�|�I d H S || f|�|�S d S )Nr � status_codeT)r~ ) r^ rk rl r1 �hasattrr? r� �intr@ r� r} �__mro__�_is_async_callable)rD rc rd rs �expZis_http_server_errorZ old_handler�cls�Z old_handlersrE rF �!_sentry_patched_exception_handler� s, � � � zfpatch_exception_middleware.<locals>._sentry_middleware_init.<locals>._sentry_patched_exception_handler)Z_exception_handlers�copy�keys)rD rc rd r� �key��old_middleware_initr� rF r� � s z;patch_exception_middleware.<locals>._sentry_middleware_initc � s t |� � | |||�I d H d S �N��_add_user_to_sentry_scope�rD rr rg ri rt rE rF rx s zDpatch_exception_middleware.<locals>._sentry_exceptionmiddleware_callN)rG r< ry )rz r{ r� rx rE )ru r� rF �patch_exception_middleware� s * r� c C s� d| kr dS t � sdS i }| d }t|dd�}|r@|�d|j� t|dd�}|r^|�d|j� t|dd�}|r||�d|j� t�� }||_dS )zZ Extracts user information from the ASGI scope and adds it to Sentry's scope. �userN�username�id�email) r r` � setdefaultr� r� r� r^ �get_isolation_scoper� )rr Z user_infoZstarlette_userr� Zuser_idr� � sentry_scoperE rE rF r� * s" r� c s, | j � dt� �k}|r(� fdd�}|| _ dS )z/ Add user information to Sentry scope. rw c � s � | |||�I d H t |� d S r� r� r� rt rE rF rw U s zNpatch_authentication_middleware.<locals>._sentry_authenticationmiddleware_callN)ry r< )rz r{ rw rE rt rF � patch_authentication_middlewareJ s r� c s, t j� dt� �k} | r(� fdd�}|t _dS )za Patches Starlettes `Middleware` class to record spans for every middleware invoked. r� c sP |t kr� | |f|�S t|�}� | |f|� |tkr<t|� |tkrLt|� d S r� )r r| r' r� r. r� )rD r� r� Zspan_enabled_clsr� rE rF r� i s z2patch_middlewares.<locals>._sentry_middleware_initN)r& rG r< )r{ r� rE r� rF rK ] s rK c s t j� � fdd�} | t _dS )zG Instrument Starlette ASGI app using the SentryAsgiMiddleware. c � sp t �� �t�}|d kr*�� |||�I d H S t�� fdd�tj|jtj|rN|jnt d�}|j |_ ||||�I d H S )Nc s � �f| �|�S r� rE )�a�kw)�old_apprD rE rF �<lambda>� � zBpatch_asgi_app.<locals>._sentry_patched_asgi_app.<locals>.<lambda>)Zmechanism_typer8 Z span_originr> ) r^ rk rl r1 r rT r8 rU r> r Z _run_asgi3ry )rD rr rg ri rs Z middleware�r� �rD rF �_sentry_patched_asgi_app� s �� z0patch_asgi_app.<locals>._sentry_patched_asgi_appN)r$ ry )r� rE r� rF rL z s rL c C s2 t | tj�r| j} q t�| �p0t| �o0t�| j�S r� )r? � functools�partial�func�asyncio�iscoroutinefunction�callablery )�objrE rE rF r� � s �r� c s t jj� � fdd�} | t j_d S )Nc sD | � t � �}|r"� fdd�}|} nt�� �� fdd��}|} �| �S )Nc � s� t �� �t�}|d kr&�| |�I d H S | d }tt �� |j|� t �� }t|�}|� � I d H � � fdd�}tj |_ |� |||�� �| |�I d H S )Nr c s � fdd�}|S )Nc sH | � di �}� r8d� kr$� d |d< d� kr8� d |d< t|�| d< | S )N�request�cookies�data��getr �r� r � request_info��inforE rF �event_processor� s z�patch_request_response.<locals>._sentry_request_response.<locals>._sentry_async_func.<locals>._make_request_event_processor.<locals>.event_processorrE ��reqrs r� r� rE rF � _make_request_event_processor� s z{patch_request_response.<locals>._sentry_request_response.<locals>._sentry_async_func.<locals>._make_request_event_processor) r^ rk rl r1 � _set_transaction_name_and_sourcero r8 r� �StarletteRequestExtractor�extract_request_inforT �_name�add_event_processor)rc rd rs r� r� � extractorr� �Zold_funcr� rF �_sentry_async_func� s( �� �zTpatch_request_response.<locals>._sentry_request_response.<locals>._sentry_async_funcc s� t �� �t�}|d kr �| |�S t �� }|jd k r<|j�� | d }t||j|� t |�}|� � � � fdd�}tj |_ |� |||�� �| |�S )Nr c s � fdd�}|S )Nc s( | � di �}� r� |d< t|�| d< | S )Nr� r� r� r� �r� rE rF r� � s z�patch_request_response.<locals>._sentry_request_response.<locals>._sentry_sync_func.<locals>._make_request_event_processor.<locals>.event_processorrE r� r� rE rF r� � s zzpatch_request_response.<locals>._sentry_request_response.<locals>._sentry_sync_func.<locals>._make_request_event_processor)r^ rk rl r1 r� ZprofileZupdate_active_thread_idr� r8 r� � extract_cookies_from_requestrT r� r� )rc rd rs r� r� r� r� r� r� rF �_sentry_sync_func� s, � � �zSpatch_request_response.<locals>._sentry_request_response.<locals>._sentry_sync_func)r� r� �wraps)r� Z is_coroutiner� r� �Zold_request_responser� rF �_sentry_request_response� s --z8patch_request_response.<locals>._sentry_request_response)r2 ZroutingZrequest_response)r� rE r� rF rM � s jrM c sb zddl m� W n tk r& Y d S X ddlm} | j�dt��k}|r^� �fdd�}|| _d S )Nr ��Markup)�Jinja2Templates� _sentry_jinja2templates_initc sB � fdd�}|� dg � ||d kr2|d �|� �| f|�|�S )Nc s � t �� �� �}d|iS )NZsentry_trace_meta)r^ ro Ztrace_propagation_meta)r� Z trace_metar� rE rF �add_sentry_trace_meta- s � �zTpatch_templates.<locals>._sentry_jinja2templates_init.<locals>.add_sentry_trace_metaZcontext_processors)r� �append)rD rc rd r� �r� Zold_jinja2templates_initrE rF r� + s z5patch_templates.<locals>._sentry_jinja2templates_init)Z markupsafer� � ImportErrorZstarlette.templatingr� rG r< )r� r{ r� rE r� rF rN s �rN c @ sT e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dS )r� z� Extracts useful information from the Starlette request (like form data or cookies) and adds it to the Sentry event. Nc C s || _ d S r� )r� )rD r� rE rE rF rG H s z"StarletteRequestExtractor.__init__c C s d }t � r| �� }|S r� )r r� )rD r� rE rE rF r� L s z6StarletteRequestExtractor.extract_cookies_from_requestc � s" t �� }i }t� �� t� r(| �� |d< | �� I d H }|sJ|W 5 Q R � S |rtt||�stt�� |d< |W 5 Q R � S | � � I d H }|r�||d< |W 5 Q R � S | � � I d H }|r�i }|� � D ]&\}}t |t �} | s�|nt�� ||< q�||d< |W 5 Q R � S t�� |d< |W 5 Q R � S Q R X d S )Nr� r� )r^ rk r r r� �content_lengthr r Z removed_because_over_size_limit�json�form�itemsr? r% Zremoved_because_raw_data) rD �clientr� r� r� r� Z form_datar� �val�is_filerE rE rF r� T s> � �� z.StarletteRequestExtractor.extract_request_infoc � s d| j jkr t| j jd �S d S )Nzcontent-length)r� �headersr� r� rE rE rF r� � s z(StarletteRequestExtractor.content_lengthc C s | j jS r� )r� r� r� rE rE rF r� � s z!StarletteRequestExtractor.cookiesc � s, t d kr d S | j�� I d H | j�� I d H S r� )� multipartr� �bodyr� r� rE rE rF r� � s z StarletteRequestExtractor.formc C s t | jj�d��S )Nz content-type)r r� r� r� r� rE rE rF �is_json� s z!StarletteRequestExtractor.is_jsonc � s | � � s d S | j�� I d H S r� )r� r� r� r� rE rE rF r� � s z StarletteRequestExtractor.json) rQ rR rS �__doc__r� rG r� r� r� r� r� r� r� rE rE rE rF r� @ s 0 r� c C sD | � d�}|sd S |jD ]&}|�| �}|d tjkr|j S qd S )N�routerr )r� Zroutes�matchesr) ZFULL�path)rr r� Zroute�matchrE rE rF � _transaction_name_from_router� s r� c C st d }t | }|dkr2|j�d�}|rDt|�p.d }n|dkrDt|j�}|d krTt}t}| j||d� t� d||� d S )Nr/ r0 rW z=[Starlette] Set transaction name and source on scope: %s / %s) r rr r� r r� �_DEFAULT_TRANSACTION_NAMEr rp r �debug)rr r8 r� r[ rX r/ rE rE rF r� � s" �r� c C s@ d }d }|j dkr"t| j�}t}n|j dkr8t|�}t}||fS )Nr/ r0 )r8 r rm r r� r )rq Z asgi_scopers r[ rX rE rE rF rn � s rn )F)[r� r� rA �collections.abcr r� r r^ Zsentry_sdk.constsr Zsentry_sdk.integrationsr r r Z$sentry_sdk.integrations._wsgi_commonr r r r Z sentry_sdk.integrations.asgir Zsentry_sdk.scoper Zsentry_sdk.tracingr r r Zsentry_sdk.utilsr r r r r r r �typingr r r r r r r r r Zsentry_sdk._typesr! r" r2 r# rI Zstarlette.applicationsr$ Zstarlette.datastructuresr% Zstarlette.middlewarer& Z#starlette.middleware.authenticationr' Zstarlette.requestsr( Zstarlette.routingr) Zstarlette.typesr* r+ r, ZStarletteScoper- r� Z starlette.middleware.exceptionsr. Zstarlette.exceptionsr� r� r6 r1 r| r� r� r� r� rK rL r� rM rN r� r� r� rn rE rE rE rF �<module> sn $ ( <G M ! q*h