Edit File: boto3.cpython-37.pyc
B Z~�_1 � @ s� d dl mZ d dlmZ d dlmZmZ d dlmZ d dl m Z d dl m Z e r�d dl mZ d dl mZ d d l mZ d d l mZ y4d d lmZ d d lmZ d d lmZ d dlmZ W n ek r� ed��Y nX G dd� de�Z dd� Z dd� Z dd� Z dS )� )�absolute_import)�Hub)� Integration� DidNotEnable)�Span)�partial)�MYPY)�Any)�Dict)�Optional)�Type)� __version__)� BaseClient)� StreamingBody)� AWSRequestzbotocore is not installedc @ s e Zd ZdZedd� �ZdS )�Boto3IntegrationZboto3c st y t ttt�d�d d� ��} W n& ttfk rF td�t���Y nX | dk rXtd��t j � � fdd�}|t _ d S )N�.� z Unparsable botocore version: {})� � z#Botocore 1.12 or newer is required.c sX � | f|�|� | j }|jj�� }|j�dtt|d�� |j�dt� |j�dt � d S )Nzrequest-created)� service_idz after-callzafter-call-error) �metaZ service_modelr Z hyphenize�events�registerr �_sentry_request_created�_sentry_after_call�_sentry_after_call_error)�self�args�kwargsr r )� orig_init� �N/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/boto3.py�sentry_patched_init) s z8Boto3Integration.setup_once.<locals>.sentry_patched_init) �tuple�map�int�BOTOCORE_VERSION�split� ValueError� TypeErrorr �formatr �__init__)�versionr# r! )r r" � setup_once s zBoto3Integration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr. r! r! r! r" r s r c K sp t j}|�t�d krd S d| |f }|j|d|d�}|�d| � |�d|� |�d|j� |�� ||j d< d S )Nz aws.%s.%sz aws.request)�hub�op� descriptionzaws.service_idzaws.operation_namezaws.request.url�_sentrysdk_span) r �currentZget_integrationr Z start_spanZset_tag�set_data�url� __enter__�context)r �requestZoperation_namer r4 r6 �spanr! r! r" r 8 s r c s� | � dd �}|d krd S |�d d d � |�d�}t|t�s>d S |jd|jd��|j�|j� ��fdd�}||_� �fdd�}||_d S ) Nr7 ZBodyzaws.request.stream)r5 r6 c s> y� | |�}|s�� � |S tk r8 �� � � Y nX d S )N)�finish� Exception)r r �ret)� orig_read�streaming_spanr! r" �sentry_streaming_body_readf s z6_sentry_after_call.<locals>.sentry_streaming_body_readc s �� � � | |� d S )N)r? )r r )� orig_closerC r! r" �sentry_streaming_body_closes s z7_sentry_after_call.<locals>.sentry_streaming_body_close) �pop�__exit__�get� isinstancer Z start_childr6 �read�close)r< �parsedr r> �bodyrD rF r! )rE rB rC r" r Q s r c K s. | � dd �}|d krd S |�t|�|d � d S )Nr7 )rG rH �type)r< � exceptionr r> r! r! r" r { s r N) � __future__r � sentry_sdkr Zsentry_sdk.integrationsr r Zsentry_sdk.tracingr Zsentry_sdk._functoolsr Zsentry_sdk._typesr �typingr r r r Zbotocorer r' Zbotocore.clientr Zbotocore.responser Zbotocore.awsrequestr � ImportErrorr r r r r! r! r! r" �<module> s* *