관리-도구
편집 파일: sqlalchemy.cpython-38.pyc
U �Mg� � @ s d dl mZmZ d dlmZmZ d dlmZmZ d dl m Z m Z m Z z(d dl mZ d dlmZ d dlmZ W n ek r� ed��Y nX d d lmZ er�d d lmZ d d lmZ d d lmZ d d lmZ G dd� de�Z e e �dd� �Z e e �dd� �Z dd� Z dd� Z dd� Z!dS )� )� SPANSTATUS�SPANDATA)� Integration� DidNotEnable)�add_query_source�record_sql_queries)�capture_internal_exceptions�ensure_integration_enabled� parse_version)�Engine)�listen)� __version__zSQLAlchemy not installed.)� TYPE_CHECKING)�Any)�ContextManager)�Optional)�Spanc @ s& e Zd ZdZde� �Zedd� �ZdS )�SqlalchemyIntegration� sqlalchemyzauto.db.c C sV t t�} | d kr td�t���| dk r.td��ttdt� ttdt� ttdt� d S )Nz!Unparsable SQLAlchemy version: {})� � z!SQLAlchemy 1.2 or newer required.Zbefore_cursor_executeZafter_cursor_execute� handle_error) r �SQLALCHEMY_VERSIONr �formatr r �_before_cursor_execute�_after_cursor_execute� _handle_error)�version� r �P/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/sqlalchemy.py� setup_once s � z SqlalchemyIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier�origin� staticmethodr r r r r r s r c G sR t ||||r|jr|jjp d |tjd�}||_|�� }|d k rNt|| � ||_d S )N)� paramstyle� executemanyZ span_origin) r �dialectr' r r% �_sentry_sql_span_manager� __enter__� _set_db_data�_sentry_sql_span) �conn�cursor� statement� parameters�contextr( �args�ctx_mgr�spanr r r r 2 s � r c G sZ t |dd �}|d k r(d |_|�d d d � t |dd �}|d k rVt� � t|� W 5 Q R X d S )Nr* r- )�getattrr* �__exit__r r )r. r/ r0 r1 r2 r3 r4 r5 r r r r H s � r c G s^ | j }|d krd S t|dd �}|d k r2|�tj� t|dd �}|d k rZd |_|�d d d � d S )Nr- r* )�execution_contextr6 Z set_statusr �INTERNAL_ERRORr* r7 )r2 r3 r8 r5 r4 r r r r Y s �r c C sH t | �} d| krdS d| kr dS d| kr,dS d| kr8dS d| krDdS d S )NZsqlite�postgres� postgresqlZmariadb�mysql�oracle)�str)�namer r r �_get_db_systemq s r@ c C s� t |jj�}|d k r"| �tj|� |jjd kr2d S |jjj}|d k rR| �tj|� |jjj }|d k rr| �tj |� |jjj }|d k r�| �tj |� d S )N) r@ Zenginer? �set_datar Z DB_SYSTEM�urlZdatabaseZDB_NAME�hostZSERVER_ADDRESS�portZ SERVER_PORT)r5 r. Z db_systemZdb_name�server_addressZ server_portr r r r, � s r, N)"Zsentry_sdk.constsr r Zsentry_sdk.integrationsr r Zsentry_sdk.tracing_utilsr r Zsentry_sdk.utilsr r r Zsqlalchemy.enginer Zsqlalchemy.eventr r r r � ImportError�typingr r r r Zsentry_sdk.tracingr r r r r r@ r, r r r r �<module> s.