OwlCyberSecurity - MANAGER
Edit File: sqlalchemy.cpython-312.pyc
� G��g � �4 � d dl mZmZ d dlmZmZmZ d dlmZm Z d dl m Z m Z m Z d dlmZ d dlmZ d dlmZ d d lmZ erd d lmZ d d lmZ d d lmZ d d lm Z G d� de� Z e e � d� � Z e e � d� � Z d� Z d� Z!d� Z"y# e$ r ed� �w xY w)� )� SPANSTATUS�SPANDATA)�_check_minimum_version� 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 �* � e Zd ZdZde� �Zed� � Zy)�SqlalchemyIntegration� sqlalchemyzauto.db.c � � t t � } t t | � t t dt � t t dt � t t dt � y )N�before_cursor_execute�after_cursor_execute� handle_error) r �SQLALCHEMY_VERSIONr r r r �_before_cursor_execute�_after_cursor_execute� _handle_error)�versions �Q/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/sqlalchemy.py� setup_oncez SqlalchemyIntegration.setup_once sB � � � 2�3�� � 4�g�>��v�.�0F�G��v�-�/D�E��v�~�}�5� N)�__name__� __module__� __qualname__� identifier�origin� staticmethodr" � r# r! r r s% � � �J�� �|� $�F��6� �6r# r c �� � t ||||xr$ |j xr |j j xs d |t j �� }||_ |j � }|�t || � ||_ y y )N)� paramstyle� executemany� 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�spans r! r r * sy � � !�����M�w���M�7�?�?�3M�3M�U�QU� �)�0�0� �G� (/�G�$� � � � �D� ���T�4� �#'�� � r# c �� � t |dd � }|�d |_ |j d d d � t |dd � }|� t � 5 t |� d d d � y y # 1 sw Y y xY w)Nr0 r3 )�getattrr0 �__exit__r r )r4 r5 r6 r7 r8 r9 r: r; s r! r r @ so � � ��+�T��G� ��+/��(�����t�T�*� �7� .�� 5�D� �� (� *� �T� "� +� *� � *� *�s � A�A"c �� � | j }|�y t |dd � }|� |j t j � t |dd � }|�d |_ |j d d d � y y )Nr3 r0 )�execution_contextr= � set_statusr �INTERNAL_ERRORr0 r> )r8 r9 r@ r; r: s r! r r Q s{ � � �1�1��� �� �$�&8�$� ?�D� �� ��� �1�1�2� ��5�t��G� ��59��2�����t�T�*� r# c �L � t | � } d| v ryd| v ryd| v ryd| v ryd| v ryy )N�sqlite�postgres� postgresql�mariadb�mysql�oracle)�str)�names r! �_get_db_systemrL i sD � � �t�9�D��4����T����D����$����4��� r# c �B � t |j j � }|� | j t j |� |j j �y |j j j }|� | j t j |� |j j j }|� | j t j |� |j j j }|�!| j t j |� y y )N) rL �enginerK �set_datar � DB_SYSTEM�url�database�DB_NAME�host�SERVER_ADDRESS�port� SERVER_PORT)r; r4 � db_system�db_name�server_address� server_ports r! r2 r2 s� � � �t�{�{� /� /�0�I�� � � � �h�(�(�)�4� �{�{��� ���k�k�o�o�&�&�G��� � � �h�&�&��0��[�[�_�_�)�)�N��!� � � �h�-�-�~�>��+�+�/�/�&�&�K�� � � � �h�*�*�K�8� r# N)#�sentry_sdk.constsr r �sentry_sdk.integrationsr r r �sentry_sdk.tracing_utilsr r �sentry_sdk.utilsr r r �sqlalchemy.enginer �sqlalchemy.eventr r r r � ImportError�typingr r r r �sentry_sdk.tracingr r r r r rL r2 r* r# r! �<module>re s� �� 2� U� U� I�� � 4�(�'�<� !���%� �'� 6�K� 6� � 1�2�(� 3�(�* � 1�2� #� 3� #� +�0�,9��c � 4� �2� 3�3�4�s �B � B