관리-도구
편집 파일: aiohttp.cpython-38.pyc
U �Mg�2 � @ s d dl Z d dlZd dlmZ d dlZd dlmZ d dlmZm Z m Z d dl m Z m Z mZ d dlmZ d dlmZ d dlmZmZ d d lmZmZmZ d d lmZ d d l m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z( z<d dl)Z)d d l*m+Z, d d l*m-Z-m.Z. d dl/m0Z0m1Z1m2Z2 W n e3k �r2 ed��Y nX d dl4m5Z5 e5�r�d dl6m7Z7 d dl8m9Z9 d dl*m:Z:m;Z; d dl<m=Z= d dl>m?Z? d dl4m@Z@ d dl4mAZA d dl4mBZB d dl4mCZC d dl mDZD d dlEmFZFmGZG d ZHG d d � d e �ZId d � ZJd!d"� ZKd#d$� ZLd%ZMd&d'� ZNdS )(� N)�wraps)�continue_trace)�OP� SPANSTATUS�SPANDATA)�$_DEFAULT_FAILED_REQUEST_STATUS_CODES� Integration� DidNotEnable)� ignore_logger)� track_session)�_filter_headers�request_body_within_bounds)�BAGGAGE_HEADER_NAME�SOURCE_FOR_STYLE�TRANSACTION_SOURCE_ROUTE)�should_propagate_trace) �capture_internal_exceptions�ensure_integration_enabled�event_from_exception�logger� parse_url� parse_version�reraise�transaction_from_function�HAS_REAL_CONTEXTVARS�CONTEXTVARS_ERROR_MESSAGE�SENSITIVE_DATA_SUBSTITUTE�AnnotatedValue)� __version__)� ClientSession� TraceConfig)� Application� HTTPException� UrlDispatcherzAIOHTTP not installed)� TYPE_CHECKING)�Request)�UrlMappingMatchInfo)�TraceRequestStartParams�TraceRequestEndParams)�Set)�SimpleNamespace)�Any)�Optional)�Tuple)�Union)�ExcInfo)�Event�EventProcessor)� handler_name�method_and_path_patternc @ s6 e Zd ZdZde� �Zd ed�dd�Zedd� �Zd S ) �AioHttpIntegration�aiohttpz auto.http.r2 )�failed_request_status_codesc C s( |t krtd|t f ��|| _|| _d S )Nz7Invalid value for transaction_style: %s (must be in %s))�TRANSACTION_STYLE_VALUES� ValueError�transaction_style� _failed_request_status_codes)�selfr9 r6 � r< �M/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/aiohttp.py�__init__J s ��zAioHttpIntegration.__init__c s� t t�} | d kr td�t���| dk r.td��ts>tdt ��td� tj��fdd�}|t_t j �t ���fdd ��}|t _ t j � tt� �� fd d ��}|t _ d S ) Nz Unparsable AIOHTTP version: {})� � z AIOHTTP 3.4 or newer required.zSThe aiohttp integration for Sentry requires Python 3.7+ or aiocontextvars package.zaiohttp.serverc � s� t �� �t�}|d kr.� | |f|�|�I d H S t�|�}t �� ��j}t|dd���P |�� |� � |� t |�� t |j �}t|tjdttjd�}t j|d|id��� z� | |�I d H } W n� tk r� } z$|�| j� | j|jkr�t� � W 5 d } ~ X Y nH tjtfk �r$ |� t j � � Y n t k �rB t t� � Y nX z | j!} W n t"k �rd Y n X |�| � | W 5 Q R � W 5 Q R � W 5 Q R � S Q R X W 5 Q R X W 5 Q R X d S )N�request)Z session_modezgeneric AIOHTTP request)�op�name�source�originZaiohttp_request)Zcustom_sampling_context)#� sentry_sdk� get_client�get_integrationr4 �weakref�refZisolation_scoper Z generate_propagation_contextZclear_breadcrumbsZadd_event_processor�_make_request_processor�dict�headersr r Z HTTP_SERVERr rE Zstart_transactionr" �set_http_status� status_coder: �_capture_exception�asyncioZCancelledError�ConnectionResetErrorZ set_statusr Z CANCELLED� Exceptionr �status�AttributeError) r; rA �args�kwargs� integration� weak_request�scoperM Z transaction�response�eZresponse_status)� old_handler< r= �sentry_app_handleq sT � � �� z8AioHttpIntegration.setup_once.<locals>.sentry_app_handlec � s� � | |�I d H }t �� �t�}|d kr*|S d }zN|jdkrFt|j�}n4|jdkrz|�� }|�d�pj|�d�}d� |j |�}W n t k r� Y nX |d k r�t � � j |t|j d� |S )Nr2 r3 �path� formatterz{} {})rD )rF rG rH r4 r9 r �handler�get_info�get�format�methodrS �get_current_scopeZset_transaction_namer )r; rA �rvrX rC Z route_info�pattern)�old_urldispatcher_resolver< r= � sentry_urldispatcher_resolve� s( �zCAioHttpIntegration.setup_once.<locals>.sentry_urldispatcher_resolvec s4 t |�d�pd�}t� }|�|� ||d< � | |�S )NZ trace_configsr< )�listrc �create_trace_config�append)rV rW Zclient_trace_configs� trace_config)�old_client_session_initr< r= �init� s z+AioHttpIntegration.setup_once.<locals>.init)r �AIOHTTP_VERSIONr rd r r r r! Z_handler# �resolver r r> r r4 )�versionr^ rj rp r< )ro r] ri r= � setup_onceY s. �� ? z AioHttpIntegration.setup_onceN)r2 ) �__name__� __module__� __qualname__� identifierrE r r> � staticmethodrt r< r< r< r= r4 F s �� r4 c C s2 dd� } dd� }t � }|j�| � |j�|� |S )Nc � sB t �� �t�d krd S |j�� }d }t� � tt|j �dd�}W 5 Q R X t j t j d||r^|j nt f tjd�}|�tj|� |d k r�|�d|j � |�tj|j� |�tj|j� t �� }t|t|j ���r8t �� j|d�D ]Z\}}t�dj|||j d�� |t k�r,|j � t ��r,|j | d | 7 < q�||j |< q�||_ d S ) NF)�sanitizez%s %s)rB rC rE �url)�spanzE[Tracing] Adding `{key}` header {value} to outgoing request to {url}.)�key�valuer{ �,) rF rG rH r4 re �upperr r �strr{ Z start_spanr Z HTTP_CLIENTr rE �set_datar Z HTTP_METHODZ HTTP_QUERY�queryZ HTTP_FRAGMENT�fragmentr rf Z iter_trace_propagation_headersr �debugrd r rM rc r| ) �session�trace_config_ctx�paramsre � parsed_urlr| �clientr} r~ r< r< r= �on_request_start� sN ���� ��� z-create_trace_config.<locals>.on_request_startc � sB |j d krd S |j }|�t|jj�� |�d|jj� |�� d S )N�reason)r| rN �intr[ rT r� r� �finish)r� r� r� r| r< r< r= �on_request_end s z+create_trace_config.<locals>.on_request_end)r r� rm r� )r� r� rn r< r< r= rl � s / rl c s � fdd�}|S )Nc s� � � }|d kr| S t � �j | �di �}d|j|j|jf |d< |j|d< |j|d< d|ji|d< tt |j ��|d< t |�|d < W 5 Q R X | S ) NrA z %s://%s%sr{ � query_stringre Z REMOTE_ADDR�envrM �data) r � setdefault�scheme�hostr_ r� re �remoter rL rM �get_aiohttp_request_data)�event�hintrA Z request_info�rY r< r= �aiohttp_processor( s � z2_make_request_processor.<locals>.aiohttp_processorr< )rY r� r< r� r= rK & s rK c C s8 t �� } t| t�� jddd�d�\}}tj||d� | S )Nr5 F)�typeZhandled)Zclient_optionsZ mechanism)r� )�sys�exc_infor rF rG �optionsZ capture_event)r� r� r� r< r< r= rP I s � rP z8[Can't show request body due to implementation details.]c C sL | j }|d k r>tt�� t|��s(t�� S | jp0d}|�|d�S | j rHt S d S )Nzutf-8�replace) Z _read_bytesr rF rG �lenr Z removed_because_over_size_limit�charset�decodeZ can_read_body�BODY_NOT_READ_MESSAGE)rA Z bytes_body�encodingr< r< r= r� X s r� )Or� rI � functoolsr rF Zsentry_sdk.apir Zsentry_sdk.constsr r r Zsentry_sdk.integrationsr r r Z sentry_sdk.integrations.loggingr Zsentry_sdk.sessionsr Z$sentry_sdk.integrations._wsgi_commonr r Zsentry_sdk.tracingr r r Zsentry_sdk.tracing_utilsr Zsentry_sdk.utilsr r r r r r r r r r r r rQ r5 r rq r r Z aiohttp.webr! r" r# � ImportError�typingr$ Zaiohttp.web_requestr% Zaiohttp.web_urldispatcherr& r' r( �collections.abcr) �typesr* r+ r, r- r. r/ Zsentry_sdk._typesr0 r1 r7 r4 rl rK rP r� r� r< r< r< r= �<module> sP 8 D#