OwlCyberSecurity - MANAGER
Edit File: pymongo.cpython-312.pyc
� G��g� � � � d dl Z d dlZd dlZd dlmZmZmZ d dlmZm Z d dl m Z d dl m Z d dlmZ d dlmZ d d lmZ erd d lmZmZmZ d d lmZmZmZ g d �Z d � Z d� Z G d� dej> � Z G d� de � Z!y# e$ r ed� �w xY w)� N)� SPANSTATUS�SPANDATA�OP)� DidNotEnable� Integration)�should_send_default_pii)�Span)�capture_internal_exceptions)� monitoringzPymongo not installed)� TYPE_CHECKING)�Any�Dict�Union)�CommandFailedEvent�CommandStartedEvent�CommandSucceededEvent) �insert�ordered�find�limit� singleBatch� aggregate� createIndexes�indexes�delete� findAndModify�renameCollection�to�dropc � � | D ]� }|t v }|r�|dk( xr d| v}|r� |dk( }|r| | D ] }|D ] }d||< � � �:|dv }|r| | D ] }d| | |< � �S|dk( } | r#| | D ] } d| v r| d ng D ] } d| d | < � � �}d| |< �� | S )N�updater � documentsz%s)�filter�queryr! �pipelinez$match)�SAFE_COMMAND_ATTRIBUTES) �command�key� is_safe_field�update_db_command� is_document�doc�doc_key� is_dict_field�item_key�is_pipeline_fieldr% � match_keys �N/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/pymongo.py� _strip_piir3 - s � ���� 6�6� � � � �8�O� N��w�0N�� � � �[�(� � � �s�|�� "�G�#'�C��L� #� $� � � <�<� � � #�C�L��)-��� �X�&� )� � �:� -�� � #�C�L��7?�8�7K��(�!3�QS�!S�I�48�H�X�&�y�1� "T� )� � ��� �I �L �N� c � � i }d|t j < | j }|�||t j < | j d }|�||t j < | j d }|�||t j < |S )N�mongodbr � )r � DB_SYSTEM� database_name�DB_NAME� connection_id�SERVER_ADDRESS� SERVER_PORT)�event�data�db_name�server_address� server_ports r2 � _get_db_datarC X s� � � �D� (�D�� � � ��!�!�G���!(��X� � � ��(�(��+�N��!�(6��X� $� $�%��%�%�a�(�K�� �%0��X� !� !�"� �Kr4 c �* � e Zd Zd� Zd� Zd� Zd� Zd� Zy)� CommandTracerc � � i | _ y �N)�_ongoing_operations)�selfs r2 �__init__zCommandTracer.__init__n s � �#%�� r4 c � � |j S rG )� request_id)rI r> s r2 �_operation_keyz CommandTracer._operation_keyr s � �� � � r4 c �� � t j � j t � �y t � 5 t t j |j � � }|j dd � |j dd � |j dd � d|j t j dt j |j t j |j! |j � i} |j" d |d< t% |j" d � |d < d i i}|j( |d d < |j* |d d < |j- t/ |� � |j d � d }t% |� |d d< t3 � s t5 |� }t7 j8 |t$ �� }t j: t< j> |t j@ �� }|jC � D ]) \ }} |jE || � |jG || � �+ |jC � D ] \ } } |jG | | � � t � 5 t jH |dt< j> |�� d d d � |jK � | jL | jO |� < d d d � y # t& $ r Y ���w xY w# t0 $ r Y ��Uw xY w# 1 sw Y �^xY w# 1 sw Y y xY w)Nz$dbz $clusterTimez $signaturezdb.namer6 r z net.peer.namer7 z net.peer.port� operation_ids� operation�request�lsid�id�session)�default)�op�name�originr$ )�message�category�typer? )(� sentry_sdk� get_client�get_integration�PyMongoIntegrationr �dict�copy�deepcopyr' �popr9 r r8 � DB_OPERATION� command_name�DB_MONGODB_COLLECTION�getr; �str� TypeError� operation_idrL r! rC �KeyErrorr r3 �json�dumps� start_spanr �DBrX �items�set_tag�set_data�add_breadcrumb� __enter__rH rM ) rI r> r' �tagsr? rR r$ �span�tag�valuer( s r2 �startedzCommandTracer.startedv s� � � � � � "� 2� 2�3E� F� N� � (� *��4�=�=�� � �7�8�G� �K�K��t� $� �K�K��� -� �K�K� �d� +� �5�.�.��"�"�I��%�%�u�'9�'9��.�.�� � �E�<N�<N�0O� �D� �(-�(;�(;�A�(>��_�%�(+�E�,?�,?��,B�(C��_�%� $�R�(�D�16�1C�1C�D� � !�+� .�/4�/?�/?�D� � !�)� ,� �K�K� �U�+� ,� � �{�{�6�*�4�0��36�t�9��_�%�i�0� +�,�$�W�-���J�J�w��4�E� �(�(��5�5��)�0�0��D� #�j�j�l� ��U� � � �S�%�(�� � �c�5�)� +� #�j�j�l� ��U�� � �c�5�)� +� -�.��)�)� !�G�"�%�%�d�� /� DH�>�>�CS�D� $� $�T�%8�%8��%?� @�s +� *��" � �� �� � �� ��. /�.��i +� *�sn �C K.�4-K�!AK.�$%K� CK.� (K"�4K.� K� K.�K�K.� K �K.� K � K.�"K+ �'K.�.K7c �, � t j � j t � �y | j j | j |� � }|j t j � |j d d d � y # t $ r Y y w xY wrG ) r\ r] r^ r_ rH rc rM � set_statusr �INTERNAL_ERROR�__exit__rk �rI r> rv s r2 �failedzCommandTracer.failed� sz � � � � � "� 2� 2�3E� F� N� � ��+�+�/�/��0C�0C�E�0J�K�D� �O�O�J� 5� 5� 6� �M�M�$��d� +��� � � �� �A B � B�Bc �, � t j � j t � �y | j j | j |� � }|j t j � |j d d d � y # t $ r Y y w xY wrG ) r\ r] r^ r_ rH rc rM r{ r �OKr} rk r~ s r2 � succeededzCommandTracer.succeeded� sx � � � � � "� 2� 2�3E� F� N� � ��+�+�/�/��0C�0C�E�0J�K�D� �O�O�J�M�M� *� �M�M�$��d� +��� � � �r� N)�__name__� __module__� __qualname__rJ rM ry r r� � r4 r2 rE rE m s � �&� �>T�@ � r4 rE c �* � e Zd ZdZde� �Zed� � Zy)r_ �pymongozauto.db.c �> � t j t � � y rG )r �registerrE r� r4 r2 � setup_oncez PyMongoIntegration.setup_once� s � � ���M�O�,r4 N)r� r� r� � identifierrX � staticmethodr� r� r4 r2 r_ r_ � s% � ��J�� �|� $�F��-� �-r4 r_ )"ra rl r\ �sentry_sdk.constsr r r �sentry_sdk.integrationsr r �sentry_sdk.scoper �sentry_sdk.tracingr �sentry_sdk.utilsr r� r � ImportError�typingr r r r �pymongo.monitoringr r r r& r3 rC �CommandListenerrE r_ r� r4 r2 �<module>r� s� �� � � � 6� 6� =� 4� #� 8�0�"� !��'�'�� �� �"(�V�*_�J�.�.� _�D-�� -��E � 0� �.� /�/�0�s �A? �?B