관리-도구
편집 파일: boto3.cpython-38.pyc
U �Mg� � @ s d dl mZ d dlZd dlmZmZ d dlmZmZ d dl m Z d dl m Z m Z mZmZ d dlmZ er�d dlmZ d d lmZ d d lmZ d d lmZ z4d d lmZ d d lmZ d dlm Z d dl m Z W n e k r� ed��Y nX G dd� de�Z e e �dd� �Z!dd� Z"dd� Z#dS )� )�partialN)�OP�SPANDATA)� Integration� DidNotEnable)�Span)�capture_internal_exceptions�ensure_integration_enabled� parse_url� parse_version)� TYPE_CHECKING)�Any)�Dict)�Optional)�Type)� __version__)� BaseClient)� StreamingBody)� AWSRequestzbotocore is not installedc @ s& e Zd ZdZde� �Zedd� �ZdS )�Boto3IntegrationZboto3z auto.http.c sJ t t�} | d kr td�t���| dk r.td��tj� � fdd�}|t_d S )Nz 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 �Z orig_init� �K/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/boto3.py�sentry_patched_init3 s �z8Boto3Integration.setup_once.<locals>.sentry_patched_init)r �BOTOCORE_VERSIONr �formatr �__init__)�versionr% r# r" r$ � setup_once# s � zBoto3Integration.setup_onceN)�__name__� __module__� __qualname__� identifier�origin� staticmethodr* r# r# r# r$ r s r c K s� d| |f }t jtj|tjd�}t� �B t|jdd�}|� d|j� |� t j |j � |� t j |j� W 5 Q R X |�d| � |�d|� |� t j|j� |�� ||jd< d S ) Nz aws.%s.%s��op�namer/ F)�sanitizezaws.request.urlzaws.service_idzaws.operation_name�_sentrysdk_span)� sentry_sdkZ start_spanr Z HTTP_CLIENTr r/ r r �url�set_datar Z HTTP_QUERY�queryZ HTTP_FRAGMENT�fragmentZset_tagZ HTTP_METHOD�method� __enter__�context)r �requestZoperation_namer! � description�span� parsed_urlr# r# r$ r B s � r c s� | � dd �}|d krd S |�d d d � |�d�}t|t�s>d S |jtj|jt j d��|j �|j � ��fdd�}||_ � �fdd�}||_ d S )Nr5 ZBodyr1 c s@ z � | |�}|s�� � |W S tk r: �� � � Y nX d S �N)�finish� Exception)r r! �ret)� orig_read�streaming_spanr# r$ �sentry_streaming_body_readu s z6_sentry_after_call.<locals>.sentry_streaming_body_readc s �� � � | |� d S rB )rC )r r! )� orig_closerG r# r$ �sentry_streaming_body_close� s z7_sentry_after_call.<locals>.sentry_streaming_body_close) �pop�__exit__�get� isinstancer Z start_childr ZHTTP_CLIENT_STREAMr? r r/ �read�close)r= �parsedr! r@ �bodyrH rJ r# )rI rF rG r$ r _ s$ � r c K s. | � dd �}|d krd S |�t|�|d � d S )Nr5 )rK rL �type)r= � exceptionr! r@ r# r# r$ r � s r )$� functoolsr r6 Zsentry_sdk.constsr r Zsentry_sdk.integrationsr r Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r �typingr r r r r Zbotocorer r& Zbotocore.clientr Zbotocore.responser Zbotocore.awsrequestr � ImportErrorr r r r r# r# r# r$ �<module> s. # +