관리-도구
편집 파일: asyncpg.cpython-38.pyc
U �Mg� � @ s� d dl mZ d dlZd dlmZmZmZmZmZ d dl Z d dl m Z m Z d dl mZmZ d dlmZ d dlmZmZ d dlmZmZmZ zd dlZd d lmZ W n e k r� ed ��Y nX eej �Z e dk r�e d k r�ed ��G d d� de�Z ed�Z ddd�dd�Z!eded�Z"ej#dd�ddddd d �d d ��Z$dd�dddd �d!d"�Z%d#d#d�d$d%�Z&ddd�d&d'�Z'd(d)d*d+�d,d-�Z(dS ).� )� annotationsN)�Any�TypeVar�Callable� Awaitable�Iterator)�OP�SPANDATA)� Integration� DidNotEnable)�Span)�add_query_source�record_sql_queries)�ensure_integration_enabled� parse_version�capture_internal_exceptions)� BaseCursorzasyncpg not installed.)r � r zasyncpg >= 0.23.0 requiredc @ sD e Zd ZdZde� �ZdZdd�dd�dd�Zedd �d d ��Zd S ) �AsyncPGIntegration�asyncpgzauto.db.F)� record_params�boolc C s |t _d S )N)r �_record_params)�selfr � r �M/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/asyncpg.py�__init__# s zAsyncPGIntegration.__init__�None)�returnc C sh t tjj�tj_ttjj�tj_ttjjdd�tj_ttjj�tj_ttjj �tj_ t tj j �tj _ d S )NT�� executemany) � _wrap_executer � Connection�execute�_wrap_connection_methodZ_executeZ _executemany�_wrap_cursor_creation�cursor�prepare�_wrap_connect_addrZ connect_utilsZ _connect_addrr r r r � setup_once& s �� � �z AsyncPGIntegration.setup_onceN) �__name__� __module__� __qualname__� identifier�originr r � staticmethodr) r r r r r s r �TzCallable[..., Awaitable[T]])�fr c s dddd�� fdd� }|S )Nr r0 ��args�kwargsr c � s� t �� �t�d kr"� | |�I d H S t| �dkr>� | |�I d H S | d }td |d d dtjd��}� | |�I d H }W 5 Q R X t� � t|� W 5 Q R X |S )N� � F)r& �query� params_list� paramstyler � span_origin) � sentry_sdk� get_client�get_integrationr �lenr r. r r )r3 r4 r7 �span�res�r1 r r �_inner= s$ �z _wrap_execute.<locals>._innerr �r1 rB r rA r r! < s r! � SubCursor)�boundFr zSubCursor | None�strztuple[Any, ...] | Noner zIterator[Span])r& r7 r8 r r c c s^ t �� �t�}|d k r |js d }|r(dnd }t| ||||| d k tjd�� }|V W 5 Q R X d S )NZpyformat)r& r7 r8 r9 r Zrecord_cursor_reprr: )r; r< r= r r r r. )r& r7 r8 r Z integrationZ param_styler? r r r �_record^ s �rG )r1 r r c s dddd�� �fdd� }|S )Nr r0 r2 c � s� t �� �t�d kr"�| |�I d H S | d }t| �dkr>| d nd }td ||� d��$}t|| d � �| |�I d H }W 5 Q R X |S )Nr6 r5 r r )r; r< r= r r> rG � _set_db_data�r3 r4 r7 r8 r? r@ �r r1 r r rB { s z'_wrap_connection_method.<locals>._innerr )r1 r rB r rJ r r$ x s r$ zCallable[..., T]c s$ t t� �dddd�� fdd� �}|S )Nr r0 r2 c sd | d }t | �dkr | d nd }td ||dd��*}t|| d � � | |�}|�d|� W 5 Q R X |S )Nr6 r5 Fr r z db.cursor)r> rG rH �set_datarI rA r r rB � s � z%_wrap_cursor_creation.<locals>._inner)r r rC r rA r r% � s r% c s dddd�� fdd� }|S )Nr r0 r2 c � s t �� �t�d kr"� | |�I d H S |d j}|d j}t jtjdtj d���}|� t j d� |� d�}|r�z(|� t j|d � |� t j|d � W n tk r� Y nX |� t j|� |� t j|� t� � t jdd|jd � W 5 Q R X � | |�I d H }W 5 Q R X |S ) N�params�connect)�op�namer. � postgresql�addrr r6 r7 )�message�category�data)r; r< r= r �user�databaseZ start_spanr ZDBr. rK r � DB_SYSTEM�get�SERVER_ADDRESS� SERVER_PORT� IndexError�DB_NAME�DB_USERr Zadd_breadcrumb�_data)r3 r4 rU rV r? rQ r@ rA r r rB � s8 � �z"_wrap_connect_addr.<locals>._innerr rC r rA r r( � s r( r r r )r? �connr c C s� | � tjd� |j}|rVz(| � tj|d � | � tj|d � W n tk rT Y nX |jj}|rp| � tj |� |jj }|r�| � tj |� d S )NrP r r6 ) rK r rW Z_addrrY rZ r[ �_paramsrV r\ rU r] )r? r_ rQ rV rU r r r rH � s rH ))� __future__r � contextlib�typingr r r r r r; Zsentry_sdk.constsr r Zsentry_sdk.integrationsr r Zsentry_sdk.tracingr Zsentry_sdk.tracing_utilsr r Zsentry_sdk.utilsr r r r Zasyncpg.cursorr � ImportError� __version__Zasyncpg_versionr r0 r! rD �contextmanagerrG r$ r% r( rH r r r r �<module> s8 � �#