관리-도구
편집 파일: starlite.cpython-38.pyc
U �Mg�) � @ s� d dl Z d dlmZ d dlmZmZ d dlmZ d dlm Z d dl m Z m Z d dl mZmZmZ zhd dlmZmZmZ d d lmZ d d lmZ d d lmZ d d lm Z d d l m Z m Z m Z d dl!m"Z" W n e#k r� ed��Y nX d dl$m%Z% e%�rTd dl$m&Z&m'Z'm(Z( d dl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z2m3Z3m4Z4 d dlm5Z5 d dl6m7Z7 dZ8G dd� de�Z9G dd� de�Z:dd� Z;d d � Z<d d � Z=d d!� Z>d"d#� Z?ee9�d$d%� �Z@dS )&� N)�OP)� DidNotEnable� Integration)�SentryAsgiMiddleware)�should_send_default_pii)�SOURCE_FOR_STYLE�TRANSACTION_SOURCE_ROUTE)�ensure_integration_enabled�event_from_exception�transaction_from_function)�Request�Starlite�State)�BaseRouteHandler)�DefineMiddleware)�get_plugin_for_value)� HTTPRoute)�ConnectionDataExtractor�is_async_callable�Ref)� BaseModelzStarlite is not installed)� TYPE_CHECKING)�Any�Optional�Union) �ASGIApp�Hint�HTTPReceiveMessage� HTTPScope�Message� Middleware�Receive�Scope�Send�WebSocketReceiveMessage)�MiddlewareProtocol)�Eventzgeneric Starlite requestc @ s& e Zd ZdZde� �Zedd� �ZdS )�StarliteIntegration�starlitez auto.http.c C s t � t� t� d S )N)�patch_app_init�patch_middlewares�patch_http_route_handle� r, r, �N/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/starlite.py� setup_once3 s z StarliteIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier�origin� staticmethodr. r, r, r, r- r' / s r' c s" e Zd Zejf� fdd� Z� ZS )� SentryStarliteASGIMiddlewarec s t � j|ddd|d� d S )NF�endpointZasgi)�appZunsafe_context_dataZtransaction_styleZmechanism_type� span_origin)�super�__init__)�selfr7 r8 �� __class__r, r- r: < s �z%SentryStarliteASGIMiddleware.__init__)r/ r0 r1 r' r3 r: � __classcell__r, r, r< r- r5 ; s r5 c s& t j� tt� �� fdd��} | t _dS )a� Replaces the Starlite class's `__init__` function in order to inject `after_exception` handlers and set the `SentryStarliteASGIMiddleware` as the outmost middleware in the stack. See: - https://starlite-api.github.io/starlite/usage/0-the-starlite-app/5-application-hooks/#after-exception - https://starlite-api.github.io/starlite/usage/7-middleware/0-middleware-intro/ c sf |� dg �}|jtft|t�r"|n|g�d� tjt_|�d�pBg }tf|�|d< � | f|�|� d S )N�after_exception)r? � middleware) �pop�update�exception_handler� isinstance�listr5 Z _run_asgi3�__call__�get)r; �args�kwargsr? r@ �Z old__init__r, r- �injection_wrapperR s ���� z)patch_app_init.<locals>.injection_wrapperN)r r: r r' )rK r, rJ r- r) G s r) c s& t j� tt� �� fdd��} | t _d S )Nc s dd� � | �D �S )Nc S s g | ] }t |��qS r, )�enable_span_for_middleware)�.0r@ r, r, r- � <listcomp>p s �zIpatch_middlewares.<locals>.resolve_middleware_wrapper.<locals>.<listcomp>r, )r; �Z old_resolve_middleware_stackr, r- �resolve_middleware_wrapperm s �z5patch_middlewares.<locals>.resolve_middleware_wrapper)r Zresolve_middlewarer r' )rP r, rO r- r* i s r* c sh t | d�r| tkr| S t| t�r*| jj� n| j� � fdd�}� jdk}|rdt| t�r^|| j_n|| _| S )NrF c � s� t �� �t�d kr&�| |���I d H S | jj� t jtj� tj d���}|� d� � � �fdd�}t �dt ���}|dk}|s~|n�}� �fdd�} t �dt ���} | d k} | s�| n�} �| ||| �I d H W 5 Q R � S Q R X d S ) N��op�namer3 �starlite.middleware_namec � sv t �� �t�d kr"�| |�I d H S t jtjt�dt���tj d��*}|� d� � �| |�I d H W 5 Q R � S Q R X d S �Nr1 rQ rT ) � sentry_sdk� get_client�get_integrationr' � start_spanr ZMIDDLEWARE_STARLITE_RECEIVE�getattr�strr3 �set_tag)rH rI �span)�middleware_name�receiver, r- �_sentry_receive� s � zNenable_span_for_middleware.<locals>._create_span_call.<locals>._sentry_receiver/ r` c � sr t �� �t�d kr �| �I d H S t jtjt�dt���tj d��(}|� d� � �| �I d H W 5 Q R � S Q R X d S rU ) rV rW rX r' rY r ZMIDDLEWARE_STARLITE_SENDrZ r[ r3 r\ )�messager] )r^ �sendr, r- � _sentry_send� s � zKenable_span_for_middleware.<locals>._create_span_call.<locals>._sentry_sendrc ) rV rW rX r' r= r/ rY r ZMIDDLEWARE_STARLITEr3 r\ rZ r[ ) r; �scoper_ rb Zmiddleware_spanr` Z receive_nameZreceive_patchedZ new_receiverc Z send_nameZ send_patchedZnew_send�Zold_call)r^ r_ rb r- �_create_span_call� s& � z5enable_span_for_middleware.<locals>._create_span_call)rf )�hasattrr5 rD r r@ rF r/ )r@ rf Znot_yet_patchedr, re r- rL x s �� 1 rL c s t j� � fdd�} | t _d S )Nc � s� t �� �t�d kr&�| �||�I d H S t �� }�d j�||d�}tddd�|�� � �d�}|I d H �� ��fdd�}tj|_ |� |� �| �||�I d H S )Nr7 )rd r_ rb T)Z parse_bodyZ parse_query�bodyc s� �� d�}| � di �}t�� dd��|d< t� r<� d |d< �d k rL�|d< d }|jd k rb|j}n t|jt�rx|jj}n|j}|d k r�t|�}dt d i}|s�t }dt i}| � |||d �� | S ) N� route_handler�request�_body� �content_length�cookies�data�sourcer6 )rj Z transactionZtransaction_info) rG �lenr rS rD �fnr �valuer r �_DEFAULT_TRANSACTION_NAMEr rB )�event�_ri Z request_info�funcZtx_nameZtx_info�Zextracted_request_dataZ request_datard r, r- �event_processor� s4 ��zHpatch_http_route_handle.<locals>.handle_wrapper.<locals>.event_processor) rV rW rX r' �get_isolation_scopeZ request_classr rA r2 �_nameZadd_event_processor)r; rd r_ rb � sentry_scoperj rh ry �Z old_handlerx r- �handle_wrapper� s( � �� $ z/patch_http_route_handle.<locals>.handle_wrapper)r �handle)r~ r, r} r- r+ � s 9r+ c C sh | � d�}|sd S t|t�r |S t|t�r2|�� S t|d�rD|�� S t|�}|rdt|j�sd|�|�S d S )N�user�asdict) rG rD �dictr rg r� r r �to_dict)rd Z scope_user�pluginr, r, r- �retrieve_user_from_scope s r� c C sd d }t � rt|�}|r2t|t�r2t�� }|�|� t| t�� j t j dd�d�\}}tj ||d� d S )NF)�typeZhandled)Zclient_optionsZ mechanism)�hint) r r� rD r� rV rz Zset_userr rW �optionsr' r2 Z capture_event)�excrd rv Z user_infor| ru r� r, r, r- rC s � rC )ArV Zsentry_sdk.constsr Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.asgir Zsentry_sdk.scoper Zsentry_sdk.tracingr r Zsentry_sdk.utilsr r r r( r r r Zstarlite.handlers.baser Zstarlite.middlewarer Zstarlite.plugins.baser Zstarlite.routes.httpr Zstarlite.utilsr r r Zpydanticr � ImportError�typingr r r r Zstarlite.typesr r r r r r r! r" Z StarliteScoper# r$ r% Zsentry_sdk._typesr& rt r' r5 r) r* rL r+ r� rC r, r, r, r- �<module> s@ 0 "I@