a

��b�
�@sRdd
lm
Z
mZmZmZmZmZ
ddlmZm	Z	
e
r@ddl
mZ
Gdd�d�ZdS)�)�
TYPE_CHECKING�Any�Dict�Optional�Tuple�Union)�
RedisError�
ResponseError)
�ClusterNodec@speZ
dZd
ZdZdd�
dd�Zdeddd�d	d
�Zeee	e
dfd�d
d�Zeee	e
dfd�dd�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'.

    NOTE: Due to a bug in redis<7.0, this 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 with EVAL or EVALSHA.
    )�commands�nodeN)
�returnc

Cs
i|_dS)
N)
r)
�self�r�8/usr/lib/python3.9/site-packages/redis/asyncio/parser.py�__init__s
zCommandsParser.__init__r
)r
rc�s�|
r
|
|_|j�
d
�
IdH}|��D]`\}}d|dvrBd||<q$|ddkrd|ddkrdd||<q$|ddkr$|ddkr$d||<q$d	d
�|��D�
|_dS)NZCOMMANDZmovablekeys�flags����
first_key_posr
�last_key_pos�
c
Ssi|]\}
}|
��|�qSr)
�upper)�.0�cmd�commandrrr�
<dictcomp>*�z-CommandsParser.initialize.<locals>.<dictcomp>)r
�execute_command�itemsr)rr
rrrrrr�
initializes












zCommandsParser.initialize.)�argsrc
�s
t|
�
d
krdSz|j
|
d}Wntty�


|
d��t|
dd��
}
|
d}||j
v
r�|��IdH
||j
v
r�t|���d��
�
|j
|}Yn0|dkr�|
df
S|dkr�dS|dkr�|j|
�IdHS|d}|dkr�t|
�
|}|
|d|d|d�S)	N�r
rz( command doesn't exist in Redis commandsrrrZ
step_count)	�lenr�KeyError�split�listr rr�_get_moveable_keys)rr!rZcmd_namerrrr�get_keys/s0









�








zCommandsParser.get_keysc

�srz|jj
d
g
|
�
R�IdH}WnNtyl
}
z6|��}d|vsFd|vrTWYd}~dS|�
WYd}~n
d}~00|S)NzCOMMAND GETKEYSzInvalid argumentsz The command has no key arguments)r
rr	�__str__)rr!�keys�
e�messagerrrr'Rs



��
z!CommandsParser._get_moveable_keys)
N)
�__name__�
__module__�__qualname__�__doc__�	__slots__rrr rr�strr(r'rrrrr	s
#rN)
�typingrrrrrrZredis.exceptionsrr	Zredis.asyncio.clusterr
rrrrr�<module>
s