File: //proc/self/root/usr/lib/python3.9/site-packages/redis/commands/__pycache__/parser.cpython-39.pyc
a
��bw � @ s. d dl mZmZ d dlmZ G dd� d�ZdS )� )�
RedisError�
ResponseError�� str_if_bytesc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�CommandsParsera
Parses Redis commands to get command keys.
COMMAND output is used to determine key locations.
Commands that do not have a predefined key location are flagged with
'movablekeys', and these commands' keys are determined by the command
'COMMAND GETKEYS'.
c C s i | _ | �|� d S �N)�commands�
initialize)�selfZredis_connection� r �9/usr/lib/python3.9/site-packages/redis/commands/parser.py�__init__ s zCommandsParser.__init__c C sZ |� d�}g }|D ] }tdd� |D ��r|�|� q|D ]}|�|�||�� < q8|| _d S )NZCOMMANDc s s | ]}|� � V qd S r )�isupper)�.0�xr r r � <genexpr> � z,CommandsParser.initialize.<locals>.<genexpr>)�execute_command�any�append�pop�lowerr )r
�rr Zuppercase_commands�cmdr r r r s
zCommandsParser.initializec K s� i }t |d �}||d< t|d �|d< dd� |d D �|d< |d |d
< |d |d < |d
|d< t|�dkr�|d |d< |d |d< |d |d< |S )Nr �name� Zarityc S s g | ] }t |��qS r r )r �flagr r r �
<listcomp>! r z3CommandsParser.parse_subcommand.<locals>.<listcomp>� �flags� �
first_key_pos� � last_key_pos� �
step_count� Ztips� Zkey_specifications� � subcommands)r �int�len)r
�command�optionsZcmd_dict�cmd_namer r r �parse_subcommand s z CommandsParser.parse_subcommandc s� t � �dk rdS � d �� }|| jvr||�� }|d }|| jv rV|t� dd� � � n&| �|� || jvr|t|�� � d���| j�|�}d|d v r�| j |g� �R � }n�d|d v s�|d dkr�| j
� � }n�|d
dk�rT|d dk�rT|d dk�rTd
}d|v �rJ|� d� d �� � �}|d D ]&} t | d �|k�r"| � | �}d}�q"|�sTdS |d }
|
dk �rvt � �t
|
� }
tt|d |
d |d
��} � fdd�| D �}|S )ap
Get the keys from the passed command.
NOTE: Due to a bug in redis<7.0, this function does not work properly
for EVAL or EVALSHA when the `numkeys` arg is 0.
- issue: https://github.com/redis/redis/issues/9493
- fix: https://github.com/redis/redis/pull/9733
So, don't use this function with EVAL or EVALSHA.
r Nr r z( command doesn't exist in Redis commandsZ movablekeysr Zpubsubr r% r! r# Fr) �|Tc s g | ] }� | �qS r r )r �pos��argsr r r k r z+CommandsParser.get_keys.<locals>.<listcomp>)r+ r r �split�listr r �upper�get�_get_moveable_keys�_get_pubsub_keysr r/ �abs�range) r
�
redis_connr3 r. Zcmd_name_splitr, �keysZ is_subcmdZ subcmd_nameZsubcmdr# Zkeys_posr r2 r �get_keys. sR
�
�
�
�
�zCommandsParser.get_keysc
G s� g }|d }||� � }|t|dd� � }z|jdg|�R � }W nN ty� } z6|�� }d|v sjd|v rxW Y d}~dS |�W Y d}~n
d}~0 0 |S )aA
NOTE: Due to a bug in redis<7.0, this function does not work properly
for EVAL or EVALSHA when the `numkeys` arg is 0.
- issue: https://github.com/redis/redis/issues/9493
- fix: https://github.com/redis/redis/pull/9733
So, don't use this function with EVAL or EVALSHA.
r r NzCOMMAND GETKEYSzInvalid argumentsz The command has no key arguments)r4 r5 r r �__str__)r
r< r3 �piecesr. r= �e�messager r r r8 o s ��z!CommandsParser._get_moveable_keysc G s� t |�dk rdS dd� |D �}|d �� }d}|dkrX|d �� }|dv r�|dd� }n,|d v rrt|dd� �}n|d
kr�|d g}|S ) z�
Get the keys from pubsub command.
Although PubSub commands have predetermined key locations, they are not
supported in the 'COMMAND's output, so the key positions are hardcoded
in this method
r Nc S s g | ] }t |��qS r r )r �argr r r r � r z3CommandsParser._get_pubsub_keys.<locals>.<listcomp>r ZPUBSUBr )ZCHANNELSZNUMSUB)Z SUBSCRIBEZ
PSUBSCRIBEZ UNSUBSCRIBEZ PUNSUBSCRIBEZPUBLISH)r+ r6 r5 )r
r3 r, r= Z pubsub_typer r r r9 � s
z CommandsParser._get_pubsub_keysN)
�__name__�
__module__� __qualname__�__doc__r
r r/ r> r8 r9 r r r r r s
A r N)Zredis.exceptionsr r Z redis.utilsr r r r r r �<module> s