관리-도구
편집 파일: fastapi.cpython-38.pyc
U �Mg� � @ s d dl Z d dlmZ d dlmZ d dlZd dlmZ d dlm Z d dl m Z m Z d dl mZmZ d dlmZ er�d d lmZmZmZ d d lmZ zd d lmZmZ W n ek r� ed ��Y nX z d dlZW n ek r� ed ��Y nX dZ G dd� de�Z dd� Z dd� Z dS )� N)�deepcopy)�wraps)� DidNotEnable)�should_send_default_pii)�SOURCE_FOR_STYLE�TRANSACTION_SOURCE_ROUTE)�transaction_from_function�logger)� TYPE_CHECKING)�Any�Callable�Dict)�Event)�StarletteIntegration�StarletteRequestExtractorzStarlette is not installedzFastAPI is not installedzgeneric FastAPI requestc @ s e Zd ZdZedd� �ZdS )�FastApiIntegration�fastapic C s t � d S �N)�patch_get_request_handler� r r �M/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/fastapi.py� setup_once( s z FastApiIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr r r r r r % s r c C s� d}|dkr*|j �d�}|rZt|�p&d}n0|dkrZ|j �d�}|rZt|dd �}|d k rZ|}|sht}t}nt| }| j||d� t� d||� d S )N� �endpoint�url�route�path)�sourcez;[FastAPI] Set transaction name and source on scope: %s / %s) �scope�getr �getattr�_DEFAULT_TRANSACTION_NAMEr r Zset_transaction_namer �debug)r# �transaction_style�request�namer r r! r" r r r � _set_transaction_name_and_source. s* �r+ c s t jj� � fdd�} | t j_d S )Nc s^ |� d�}|rD|jd k rDt�|j�sD|j�t���fdd��}||_�| |�� � fdd�}|S )N� dependantc s& t �� }|jd k r |j�� � | |�S r )� sentry_sdk�get_isolation_scopeZprofileZupdate_active_thread_id)�args�kwargs� sentry_scope)�old_callr r � _sentry_callX s zTpatch_get_request_handler.<locals>._sentry_get_request_handler.<locals>._sentry_callc � 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 sN | � di �}� r>d� kr*t� r*� d |d< d� kr>� d |d< t|�| d< | S )Nr) �cookies�data)r$ r r )�event�hintZ request_info��infor r �event_processoru s z�patch_get_request_handler.<locals>._sentry_get_request_handler.<locals>._sentry_app.<locals>._make_request_event_processor.<locals>.event_processorr )�req� integrationr: r8 r r � _make_request_event_processors s zzpatch_get_request_handler.<locals>._sentry_get_request_handler.<locals>._sentry_app.<locals>._make_request_event_processor) r- Z get_clientZget_integrationr r+ Zget_current_scoper( r. r Zextract_request_infor �_nameZadd_event_processor)r/ r0 r< r) r1 Z extractorr= )�old_appr8 r � _sentry_appd s$ � �zSpatch_get_request_handler.<locals>._sentry_get_request_handler.<locals>._sentry_app)r$ �call�asyncio�iscoroutinefunctionr )r/ r0 r, r3 r@ �Zold_get_request_handler)r? r2 r �_sentry_get_request_handlerN s �� � (z>patch_get_request_handler.<locals>._sentry_get_request_handler)r ZroutingZget_request_handler)rE r rD r r J s @r ) rB �copyr � functoolsr r- Zsentry_sdk.integrationsr Zsentry_sdk.scoper Zsentry_sdk.tracingr r Zsentry_sdk.utilsr r �typingr r r r Zsentry_sdk._typesr Z!sentry_sdk.integrations.starletter r r � ImportErrorr& r r+ r r r r r �<module> s.