Edit File: aiohttp.cpython-37.pyc
B Z~�_ � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZmZ d dl m Z d dl m Z m Z d dlmZ d dlmZmZmZmZmZmZ y,d dlZd d lmZ d d lm Z m Z m Z W n e k r� ed ��Y nX d d l m!Z! e!�rTd d l"m#Z# d dl$m%Z% d dl&m'Z' d dl&m(Z( d dl&m)Z) d dl&m*Z* d dl&m+Z+ d dl&m,Z, d dlm-Z- d dl m.Z. dZ/G dd� de�Z0dd� Z1d d � Z2d Z3d d � Z4dS )!� N)�reraise)�Hub)� Integration� DidNotEnable)� ignore_logger)�_filter_headers�request_body_within_bounds)� Transaction)�capture_internal_exceptions�event_from_exception�transaction_from_function�HAS_REAL_CONTEXTVARS�CONTEXTVARS_ERROR_MESSAGE�AnnotatedValue)� __version__)� Application� HTTPException� UrlDispatcherzAIOHTTP not installed)�MYPY)�Request)�AbstractMatchInfo)�Any)�Dict)�Optional)�Tuple)�Callable)�Union)�ExcInfo)�EventProcessor)� handler_name�method_and_path_patternc @ s& e Zd ZdZddd�Zedd� �ZdS ) �AioHttpIntegration�aiohttpr c C s" |t krtd|t f ��|| _d S )Nz7Invalid value for transaction_style: %s (must be in %s))�TRANSACTION_STYLE_VALUES� ValueError�transaction_style)�selfr% � r' �P/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/aiohttp.py�__init__4 s zAioHttpIntegration.__init__c s� y t ttt�d�d d� ��} W n& ttfk rF td�t���Y nX | dk rXtd��t shtdt ��t d� t j � � fdd �}|t _ tj��fd d �}|t_d S ) N�.� z AIOHTTP version unparseable: {})� � z AIOHTTP 3.4 or newer required.zSThe aiohttp integration for Sentry requires Python 3.7+ or aiocontextvars package.zaiohttp.serverc � s< t j}|�t�d kr,� | |f|�|�I d H S t�|�}t t j���}|�� � }|�� |�t |�� W d Q R X t j |j ddd�}|j |d|id��� y� | |�I d H }W np tk r� } z|�| j� � W d d } ~ X Y nB tjk r� |�d� � Y n" tk �r tt|�� Y nX |�|j� |S Q R X W d Q R X d S )Nz http.serverzgeneric AIOHTTP request)�op�nameZaiohttp_request)Zcustom_sampling_contextZ cancelled)r �current�get_integrationr! �weakref�ref�configure_scopeZclear_breadcrumbsZadd_event_processor�_make_request_processorr Zcontinue_from_headers�headersZstart_transactionr Zset_http_statusZ status_code�asyncioZCancelledErrorZ set_status� Exceptionr �_capture_exception�status) r& �request�args�kwargs�hub� weak_request�scope� transaction�response�e)� old_handler' r( �sentry_app_handleW s4 z8AioHttpIntegration.setup_once.<locals>.sentry_app_handlec � s� � | |�I d H }t j}|�t�}d }yN|jdkr<t|j�}n4|jdkrp|�� }|�d�p`|�d�}d� |j |�}W n t k r� Y nX |d k r�t j� � � }||_ W d Q R X |S )Nr r �path� formatterz{} {})r r0 r1 r! r% r �handlerZget_info�get�format�methodr8 r4 rA ) r& r; �rvr> Z integrationr/ Z route_info�patternr@ )�old_urldispatcher_resolver' r( � sentry_urldispatcher_resolve� s" zCAioHttpIntegration.setup_once.<locals>.sentry_urldispatcher_resolve)�tuple�map�int�AIOHTTP_VERSION�split� TypeErrorr$ r rJ r r r r �_handler �resolve)�versionrE rO r' )rD rN r( � setup_once= s$ ) z AioHttpIntegration.setup_onceN)r )�__name__� __module__� __qualname__� identifierr) � staticmethodrY r' r' r' r( r! 1 s r! c s � fdd�}|S )Nc s� � � }|d kr| S t � �r | �di �}d|j|j|jf |d< |j|d< |j|d< d|ji|d< tj }t t |j ��|d< t ||�|d < W d Q R X | S ) Nr; z %s://%s%s�url� query_stringrK Z REMOTE_ADDR�envr6 �data)r � setdefault�scheme�hostrF r` rK Zremoter r0 r �dictr6 �get_aiohttp_request_data)�event�hintr; Z request_infor> )r? r' r( �aiohttp_processor� s z2_make_request_processor.<locals>.aiohttp_processorr' )r? rj r' )r? r( r5 � s r5 c C s6 t �� }t|| jjddd�d�\}}| j||d� |S )Nr" F)�typeZhandled)Zclient_optionsZ mechanism)ri )�sys�exc_infor �client�optionsZ capture_event)r> rm rh ri r' r' r( r9 � s r9 z8[Can't show request body due to implementation details.]c C sd |j }|d k rVt| jt|��s@tddddt|�ggt|�d��S |jpHd}|�|d�S |jr`tS d S )N� z!config�xr )�rem�lenzutf-8�replace) Z _read_bytesr rn rs r �charset�decodeZ can_read_body�BODY_NOT_READ_MESSAGE)r> r; Z bytes_body�encodingr' r' r( rg � s rg )5rl r2 Zsentry_sdk._compatr Zsentry_sdk.hubr Zsentry_sdk.integrationsr r Z sentry_sdk.integrations.loggingr Z$sentry_sdk.integrations._wsgi_commonr r Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r r r r7 r" r rS Z aiohttp.webr r r � ImportErrorZsentry_sdk._typesr Zaiohttp.web_requestr Z aiohttp.abcr �typingr r r r r r r r r# r! r5 r9 rw rg r' r' r' r( �<module> s@ o%