관리-도구
편집 파일: pymongo.cpython-38.pyc
U �Mg� � @ s d dl Z d dlZd dlZd dlmZmZmZ d dlmZm Z d dl m Z d dl m Z d dlmZ zd dlmZ W n ek r� ed��Y nX d d lmZ er�d d lmZmZmZ d d lmZmZmZ d d dddddddddddg Z dd� Z dd � Z G d d � d ej �Z G d d � d e �Z!dS )!� 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�limitZ singleBatchZ aggregateZ createIndexesZindexes�delete� findAndModifyZrenameCollection�toZdropc C s� | D ]�}|t k}|rq|dko$d| k}|r,q|dk}|rZ| | D ]}|D ] }d||< qHq@q|dk}|r�| | D ]}d| | |< qnq|dk} | r�| | D ]*} d| kr�| d ng D ]} d| d | < q�q�qd| |< q| S )N�updater Z documentsz%s)�filter�queryr �pipelinez$match)�SAFE_COMMAND_ATTRIBUTES) �command�keyZ is_safe_fieldZupdate_db_commandZ is_document�docZdoc_keyZ is_dict_fieldZitem_keyZis_pipeline_fieldr Z match_key� r! �M/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/pymongo.py� _strip_pii- s4 r# c C sb i }d|t j< | j}|d k r&||t j< | jd }|d k rB||t j< | jd }|d k r^||t j< |S )N�mongodbr � )r � DB_SYSTEM� database_nameZDB_NAME� connection_idZSERVER_ADDRESSZ SERVER_PORT)�event�dataZdb_name�server_addressZ server_portr! r! r" � _get_db_dataX s r, c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � CommandTracerc C s i | _ d S �N)�_ongoing_operations)�selfr! r! r" �__init__n s zCommandTracer.__init__c C s |j S r. )� request_id)r0 r) r! r! r" �_operation_keyr s z CommandTracer._operation_keyc C s� t �� �t�d krd S t� ��� tt�|j��}|� dd � |� dd � |� dd � d|j t j dt j |jt j|�|j�i}z$|jd |d< t|jd �|d < W n tk r� Y nX d i i}|j|d d < |j|d d < |�t|�� z"|� d �d }t|�|d d< W n tk �r Y nX t� �s0t|�}tj |td�}t j t j |tj d�}|�!� D ]"\}} |�"|| � |�#|| � �qZ|�!� D ]\} } |�#| | � �q�t� � t j$|dt j |d� W 5 Q R X |�%� | j&| �'|�< W 5 Q R X d S )Nz$dbz $clusterTimez $signaturezdb.namer$ r z net.peer.namer% z net.peer.portZ operation_idsZ operation�request�lsid�id�session)�default)�op�name�originr )�message�category�typer* )(� sentry_sdk� get_client�get_integration�PyMongoIntegrationr �dict�copy�deepcopyr �popr' r r& Z DB_OPERATION� command_nameZDB_MONGODB_COLLECTION�getr( �str� TypeErrorZ operation_idr2 r r, �KeyErrorr r# �json�dumpsZ start_spanr ZDBr; �itemsZset_tag�set_dataZadd_breadcrumb� __enter__r/ r3 ) r0 r) r �tagsr* r5 r �span�tag�valuer r! r! r" �startedv sf �� �zCommandTracer.startedc C sb t �� �t�d krd S z0| j�| �|��}|�tj � |� d d d � W n t k r\ Y d S X d S r. ) r? r@ rA rB r/ rF r3 � set_statusr �INTERNAL_ERROR�__exit__rK �r0 r) rR r! r! r" �failed� s zCommandTracer.failedc C s` t �� �t�d krd S z0| j�| �|��}|�tj � |� d d d � W n t k rZ Y nX d S r. ) r? r@ rA rB r/ rF r3 rV r �OKrX rK rY r! r! r" � succeeded� s zCommandTracer.succeededN)�__name__� __module__� __qualname__r1 r3 rU rZ r\ r! r! r! r" r- m s @ r- c @ s& e Zd ZdZde� �Zedd� �ZdS )rB �pymongozauto.db.c C s t �t� � d S r. )r �registerr- r! r! r! r" � setup_once� s z PyMongoIntegration.setup_onceN)r] r^ r_ � identifierr; � staticmethodrb r! r! r! r" rB � s rB )"rD rL r? Zsentry_sdk.constsr r r Zsentry_sdk.integrationsr r Zsentry_sdk.scoper Zsentry_sdk.tracingr Zsentry_sdk.utilsr r` r � ImportError�typingr r r r Zpymongo.monitoringr r r r r# r, ZCommandListenerr- rB r! r! r! r" �<module> sB �+b