관리-도구
편집 파일: utils.cpython-38.pyc
U �Mgt � @ s� d dl mZ d dlmZmZmZmZmZ d dlm Z d dl m Z d dl m Z e rpd dl mZmZmZ d dlmZ dd � Zd d � Zd d � Zdd� Zdd� Zdd� Zdd� ZdS )� )�SPANDATA)�"_COMMANDS_INCLUDING_SENSITIVE_DATA� _MAX_NUM_ARGS�_MAX_NUM_COMMANDS�_MULTI_KEY_COMMANDS�_SINGLE_KEY_COMMANDS)�should_send_default_pii)�SENSITIVE_DATA_SUBSTITUTE)� TYPE_CHECKING)�Any�Optional�Sequence)�Spanc C s� | g}t |�D ]l\}}|tkr" q|| �� }|tkr>|�t� q|dk}|rZ|�t|�� qt� rp|�t|�� q|�t� qd�|�}|S )Nr � ) � enumerater �lowerr �appendr �reprr �join)�name�argsZ command_parts�i�arg�name_lowZarg_is_the_key�command� r �Q/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/redis/utils.py�_get_safe_command s r c C s4 t | t�r,z | �� W S tk r* Y dS X t| �S )N� )� isinstance�bytes�decode�UnicodeDecodeError�str��keyr r r � _safe_decode/ s r& c C sT t | tttf�r&d�dd� | D ��} n*t | t�r:t| �} n| d krHd} nt| �} | S )Nz, c s s | ]}t |�V qd S �N)r& )�.0�xr r r � <genexpr>= s z!_key_as_string.<locals>.<genexpr>r )r �dict�list�tupler r r& r# r$ r r r �_key_as_string: s r. c C s� d}|dk r"| � � tkr"t|�}n�|dk rdt|�dkrdt|d tttf�rXt|d �}q�|d f}nV|dk r�d|kr�t|d ttf�r�t|d �dkr�t|d �}n|d dk r�|d f}|S )z� Gets the key (or keys) from the given method_name. The method_name could be a redis command or a django caching command N� r r% )r r r- �lenr r+ r, )� method_namer �kwargsr% r r r � _get_safe_keyH s r3 c C s | j S r' )r )r r r r �_parse_rediscluster_commandg s r4 c C sx | � d|� | � d|� g }t|�D ]8\}}|tkr8 q^||�}|�t|d |dd � �� q$| �dt|�|d�� d S )N�redis.is_clusterzredis.transactionr r/ zredis.commands)�countZ first_ten)�set_tagr r r r �set_datar0 ) �span� is_clusterZget_command_args_fnZis_transactionZ command_stack�commandsr r r r r r �_set_pipeline_datal s ��r< c G sj | � d|� |r*| � d|� | � tj|� |rf|rf|�� }|tksV|tkrft|�dkrf| � d|d � d S )Nr5 z redis.commandr/ z redis.keyr )r7 r Z DB_OPERATIONr r r r0 )r9 r: r r r r r r �_set_client_data� s � �r= N)Zsentry_sdk.constsr Z$sentry_sdk.integrations.redis.constsr r r r r Zsentry_sdk.scoper Zsentry_sdk.utilsr �typingr r r r Zsentry_sdk.tracingr r r& r. r3 r4 r< r= r r r r �<module> s