관리-도구
편집 파일: litestar.cpython-38.pyc
U �Mg�* � @ s� d dl Z d dlmZ d dlmZmZ d dlmZ d dlm Z d dl m Z d dl m Z mZ d dlmZmZmZ zDd d lmZmZ d d lmZ d d lmZ d d lmZ d d l m Z W n e k r� ed��Y nX d dl m Z e �rDd dl m!Z!m"Z"m#Z# d dl$m%Z% d dl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z-m.Z.m/Z/ d dlm0Z0 d dl1m2Z2m3Z3 dZ4G dd� de�Z5G dd� de�Z6dd� Z7d d � Z8d d � Z9d d!� Z:d"d#� Z;ee5�d$d%� �Z<dS )&� N)�OP)� DidNotEnable� Integration)�SentryAsgiMiddleware)� ignore_logger)�should_send_default_pii)�SOURCE_FOR_STYLE�TRANSACTION_SOURCE_ROUTE)�ensure_integration_enabled�event_from_exception�transaction_from_function)�Request�Litestar)�BaseRouteHandler)�DefineMiddleware)� HTTPRoute)�ConnectionDataExtractorzLitestar is not installed)� TYPE_CHECKING)�Any�Optional�Union)�ASGIApp)�HTTPReceiveMessage� HTTPScope�Message� Middleware�Receive�Scope�Send�WebSocketReceiveMessage)�MiddlewareProtocol)�Event�Hintzgeneric Litestar requestc @ s& e Zd ZdZde� �Zedd� �ZdS )�LitestarIntegration�litestarz auto.http.c C s t � t� t� td� d S )Nr$ )�patch_app_init�patch_middlewares�patch_http_route_handler � r( r( �N/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/litestar.py� setup_once0 s z LitestarIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier�origin� staticmethodr* r( r( r( r) r# , s r# c s" e Zd Zejf� fdd� Z� ZS )� SentryLitestarASGIMiddlewarec s t � j|ddd|d� d S )NF�endpointZasgi)�appZunsafe_context_dataZtransaction_styleZmechanism_type� span_origin)�super�__init__)�selfr3 r4 �� __class__r( r) r6 C s �z%SentryLitestarASGIMiddleware.__init__)r+ r, r- r# r/ r6 � __classcell__r( r( r8 r) r1 B s r1 c s& t j� tt� �� fdd��} | t _dS )a\ Replaces the Litestar class's `__init__` function in order to inject `after_exception` handlers and set the `SentryLitestarASGIMiddleware` as the outmost middleware in the stack. See: - https://docs.litestar.dev/2/usage/applications.html#after-exception - https://docs.litestar.dev/2/usage/middleware/using-middleware.html c sP t f|�d�pg �|d< tjt_|�d�p,g }tf|�|d< � | f|�|� d S )NZafter_exception� middleware)�exception_handler�getr1 Z _run_asgi3�__call__)r7 �args�kwargsr; �Z old__init__r( r) �injection_wrapperZ s � �z)patch_app_init.<locals>.injection_wrapperN)r r6 r r# )rB r( rA r) r% O 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>q s �zIpatch_middlewares.<locals>.resolve_middleware_wrapper.<locals>.<listcomp>r( )r7 �Z old_resolve_middleware_stackr( r) �resolve_middleware_wrappern s �z5patch_middlewares.<locals>.resolve_middleware_wrapper)r Zresolve_middlewarer r# )rG r( rF r) r& j s r& c sh t | d�r| tkr| S t| t�r*| jj� n| j� � fdd�}� jdk}|rdt| t�r^|| j_n|| _| S )Nr> 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�namer/ �litestar.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 �Nr- rH rK ) � sentry_sdk� get_client�get_integrationr# � start_spanr ZMIDDLEWARE_LITESTAR_RECEIVE�getattr�strr/ �set_tag)r? r@ �span)�middleware_name�receiver( r) �_sentry_receive� s � zNenable_span_for_middleware.<locals>._create_span_call.<locals>._sentry_receiver+ rW 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 rL ) rM rN rO r# rP r ZMIDDLEWARE_LITESTAR_SENDrQ rR r/ rS )�messagerT )rU �sendr( r) � _sentry_send� s � zKenable_span_for_middleware.<locals>._create_span_call.<locals>._sentry_sendrZ ) rM rN rO r# r9 r+ rP r ZMIDDLEWARE_LITESTARr/ rS rQ rR ) r7 �scoperV rY Zmiddleware_spanrW Z receive_nameZreceive_patchedZ new_receiverZ Z send_nameZ send_patchedZnew_send�Zold_call)rU rV rY r) �_create_span_call� s& � z5enable_span_for_middleware.<locals>._create_span_call)r] )�hasattrr1 � isinstancer r; r> r+ )r; r] Znot_yet_patchedr( r\ r) rC y s �� 1 rC 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 )Nr3 )r[ rV rY 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|jd�rx|jj}n|j}|d k r�t|�}d td i}|s�t }d t i}| � |||d �� | S ) N� route_handler�request�_body� �content_length�cookies�data�value�sourcer2 )rb Z transactionZtransaction_info) r= �lenr rJ r^ �fnrh r r �_DEFAULT_TRANSACTION_NAMEr �update)�event�_ra Z request_info�funcZtx_nameZtx_info�Zextracted_request_dataZ request_datar[ r( r) �event_processor� s4 ��zHpatch_http_route_handle.<locals>.handle_wrapper.<locals>.event_processor) rM rN rO r# �get_isolation_scopeZ request_classr �popr. �_nameZadd_event_processor)r7 r[ rV rY � sentry_scoperb r` rr �Z old_handlerq r) �handle_wrapper� s( � �� % z/patch_http_route_handle.<locals>.handle_wrapper)r �handle)rx r( rw r) r' � s :r' c C s. | � d�}t|t�r|S t|d�r*|�� S d S )N�user�asdict)r= r_ �dictr^ r{ )r[ Z scope_userr( 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} r_ r| rM rs Zset_userr rN �optionsr# r. Z capture_event)�excr[ Z user_inforv rn r r( r( r) r< s � r< )=rM Zsentry_sdk.constsr Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.asgir Z sentry_sdk.integrations.loggingr Zsentry_sdk.scoper Zsentry_sdk.tracingr r Zsentry_sdk.utilsr r r r$ r r Zlitestar.handlers.baser Zlitestar.middlewarer Zlitestar.routes.httpr Zlitestar.data_extractorsr � ImportError�typingr r r r Zlitestar.types.asgi_typesr Zlitestar.typesr r r r r r Z LitestarScoper r r Zsentry_sdk._typesr! r" rl r# r1 r% r& rC r' r} r< r( r( r( r) �<module> s@ ( IA