OwlCyberSecurity - MANAGER
Edit File: clickhouse_driver.cpython-312.pyc
� G��g � � � d dl Z d dlmZmZ d dlmZmZmZ d dlm Z d dl m Z d dl m Z mZ d dlmZmZ er d dlmZmZ n G d � d � Z G d � d � Z e� Z d dlZ G d� de� Z ed� Z ed� Zdeeef deeef fd�Zdeeef deeef fd�Zdeeef deeef fd�Z de dej: j< ddfd�Z y# e$ r ed � �w xY w)� N)�OP�SPANDATA)�_check_minimum_version� Integration� DidNotEnable)�Span)�should_send_default_pii)�capture_internal_exceptions�ensure_integration_enabled)� TYPE_CHECKING�TypeVar)� ParamSpec�Callablec � � e Zd Zd� Zy)r c � � d | _ d | _ y �N)�args�kwargs��self�_s �X/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/clickhouse_driver.py�__init__zParamSpec.__init__ s � � �D�I� �D�K� N)�__name__� __module__� __qualname__r � r r r r s � � r r c � � e Zd Zd� Zy)� _Callablec � � y r r r s r � __getitem__z_Callable.__getitem__ s � �r N)r r r r" r r r r r s � � r r z clickhouse-driver not installed.c �, � e Zd ZdZde� �Zedd�� Zy)�ClickhouseDriverIntegration�clickhouse_driverzauto.db.Nc �l � t t t j � t t j j j � t j j _ t t j j j � t j j _ t t j j j � t j j _ t t j j d� rJt t j j j � t j j _ t t j j j � t j j _ y )N�receive_end_of_insert_query)r r$ r% �VERSION� _wrap_start� connection� Connection� send_query�_wrap_send_data�client�Client� send_data� _wrap_end�receive_end_of_query�hasattrr' �receive_resultr r r � setup_oncez&ClickhouseDriverIntegration.setup_once* s( � � � :�<M�<U�<U�V� >I� � (� (� 3� 3� >� >�> ��$�$�/�/�:� 5D� � $� $� +� +� 5� 5�5 �� � �'�'�1� @I� � $� $� +� +� @� @�@ �� � �'�'�<� �$�+�+�2�2�4Q� R�JS�!�(�(�/�/�K�K�K� � $� $� +� +� G� :C� � $� $� +� +� :� :�: �� � �'�'�6r )�returnN)r r r � identifier�origin� staticmethodr5 r r r r$ r$ & s% � �$�J�� �|� $�F�� � � r r$ �P�T�fr6 c � � � t t � � dt j dt j dt f� fd� � }|S )Nr r r6 c �� �� | d }| d }t | � dkD r| d n|j d� }t | � dkD r| d n|j d� }t j t j |t j �� }||_ t ||� |j d|� |r|j d |� |r t � r|j d |� �| i |��}|S ) Nr � � �query_id� �params)�op�namer8 �queryz db.query_id� db.params) �len�get� sentry_sdk� start_spanr �DBr$ r8 � _sentry_span� _set_db_data�set_datar ) r r r* rF rA rC �span�retr< s �r �_innerz_wrap_start.<locals>._innerL s� �� ��!�W� ��Q��� !�$�i�!�m�4��7����J�1G�� ��I��M��a��v�z�z�(�/C���$�$��u�u��.�5�5� �� #'� � ��T�:�&� � � �g�u�%� � �M�M�-�� 2� �-�/� �M�M�+�v� .� �� �� ��� r )r r$ r: r r r; )r< rR s ` r r) r) K sB �� � � ;�Q�?��a�f�f� ���� �Q� � @��: �Mr c �` � � dt j dt j dt f� fd� }|S )Nr r r6 c �p �� �| i |��}| d }t |j dd � }|��|� t � r|j d|� t � 5 |j j |j j d� d|j �� d d d � |j � |S # 1 sw Y �xY w)Nr rM z db.resultrF )�message�category�data) �getattrr* r rO r �scope�add_breadcrumb�_data�pop�finish)r r �res�instancerP r< s �r � _inner_endz _wrap_end.<locals>._inner_endn s� �� ��� �� ����7���x�*�*�N�D�A�� � ���#:�#<�� � �k�3�/�,�.�� � �)�)� �J�J�N�N�7� 3�g�D�J�J� *� � /� �K�K�M�� � /�.�s �AB,�,B5�r: r r r; )r< r` s ` r r1 r1 m s- �� ��!�&�&� �A�H�H� �� �$ �r c �` � � dt j dt j dt f� fd� }|S )Nr r r6 c � �� | d }| d }t |j dd � }|�_t ||j � t � r?|j j dg � }|j |� |j d|� �| i |��S )Nr r@ rM rG )rX r* rN r r[ rI �extendrO )r r r_ rW rP � db_paramsr< s �r �_inner_send_dataz)_wrap_send_data.<locals>._inner_send_data� s� �� ���7���A�w���x�*�*�N�D�A�� � � ��x� 2� 2� 3�&�(� �J�J�N�N�;�� ;� �� � ��&�� � �k�9�5��$�!�&�!�!r ra )r< rf s ` r r- r- � s- �� � "���� "�!�(�(� "�q� "� �r rP r* c � � | j t j d� | j t j |j � | j t j |j � | j t j |j � | j t j |j � y )N� clickhouse) rO r � DB_SYSTEM�SERVER_ADDRESS�host� SERVER_PORT�port�DB_NAME�database�DB_USER�user)rP r* s r rN rN � sz � � �M�M�(�$�$�l�3��M�M�(�)�)�:�?�?�;��M�M�(�&�&� ���8��M�M�(�"�"�J�$7�$7�8��M�M�(�"�"�J�O�O�4r ) rJ �sentry_sdk.constsr r �sentry_sdk.integrationsr r r �sentry_sdk.tracingr �sentry_sdk.scoper �sentry_sdk.utilsr r �typingr r r r r r% � ImportErrorr$ r: r; r) r1 r- r* r+ rN r r r �<module>ry s �� � *� U� U� #� 4� T� )� �*�*� � � � � �{�H�;� � �+� �B �c�N�� �C�L�� �8�A�q�D�>� �h�q�!�t�n� �D��!�Q�$� � �H�Q��T�N� �, �x��1��~� �(�1�a�4�.� �&5� �5� -� 8� 8� C� C�5� �5��i � ;� �9� :�:�;�s �C �C