U b+dw@s.ddlmZmZddlmZGdddZdS)) RedisError ResponseError str_if_bytesc@s@eZdZdZddZddZddZdd 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'. cCsi|_||dSN)commands initialize)selfZredis_connectionr 9/tmp/pip-unpacked-wheel-cdsyf3nb/redis/commands/parser.py__init__szCommandsParser.__init__cCsZ|d}g}|D] }tdd|Dr||q|D]}||||<q8||_dS)NZCOMMANDcss|]}|VqdSr)isupper).0xr r r sz,CommandsParser.initialize..)execute_commandanyappendpoplowerr)r rrZuppercase_commandscmdr r r r s  zCommandsParser.initializecKsi}t|d}||d<t|d|d<dd|dD|d<|d |d <|d |d <|d |d<t|dkr|d|d<|d|d<|d|d<|S)NrnamearitycSsg|] }t|qSr r)rflagr r r !sz3CommandsParser.parse_subcommand..flags first_key_pos last_key_pos step_countZtipsZkey_specifications subcommands)rintlen)r commandoptionsZcmd_dictcmd_namer r r parse_subcommands        zCommandsParser.parse_subcommandc stdkrdSd}||jkr||}|d}||jkrV|tddn&||||jkr|t|d|j|}d|dkr|j |f}nd|dks|d dkr|j }n|d dkrR|d dkrR|d dkrRd }d|krH|dd}|dD]&} t | d|kr | | }d}q |sRdS|d } | dkrttt | } 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. rNrrz( command doesn't exist in Redis commandsZ movablekeysrZpubsubrr%r!r#Fr)|Tcsg|] }|qSr r )rposargsr r rksz+CommandsParser.get_keys..)r+rrsplitlistr rupperget_get_moveable_keys_get_pubsub_keysrr/absrange) r redis_connr3r.Zcmd_name_splitr,keysZ is_subcmdZ subcmd_nameZsubcmdr#Zkeys_posr r2r get_keys.sR                 zCommandsParser.get_keysc Gsg}|d}||}|t|dd}z|jd|}WnHtk r}z*|}d|kshd|krrWY dS|W5d}~XYnX|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. rrNCOMMAND GETKEYSzInvalid argumentsz The command has no key arguments)r?)r4r5rr__str__)r r<r3piecesr.r=emessager r r r8os   z!CommandsParser._get_moveable_keyscGst|dkrdSdd|D}|d}d}|dkrX|d}|dkr|dd}n,|d krrt|dd}n|d kr|dg}|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 rNcSsg|] }t|qSr r)rargr r r rsz3CommandsParser._get_pubsub_keys..rZPUBSUBr)ZCHANNELSZNUMSUB)Z SUBSCRIBEZ PSUBSCRIBEZ UNSUBSCRIBEZ PUNSUBSCRIBEZPUBLISH)r+r6r5)r r3r,r=Z pubsub_typer r r r9s    zCommandsParser._get_pubsub_keysN) __name__ __module__ __qualname____doc__r r r/r>r8r9r r r r rs ArN)Zredis.exceptionsrrZ redis.utilsrrr r r r s