U g,+@s~ddlmZmZmZmZmZmZddlmZm Z ddl m Z erLddl m Z GdddZGdddeZGd d d eZd S) ) TYPE_CHECKINGAnyDictOptionalTupleUnion) RedisError ResponseError str_if_bytes) ClusterNodec@seZdZddZddZdS)AbstractCommandsParsercGst|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 NcSsg|] }t|qSr ).0argrr;/tmp/pip-unpacked-wheel-f3sx1i9r/redis/_parsers/commands.py sz;AbstractCommandsParser._get_pubsub_keys..rZPUBSUB)ZCHANNELSZNUMSUBZ SHARDCHANNELSZ SHARDNUMSUB)Z SUBSCRIBEZ PSUBSCRIBEZ UNSUBSCRIBEZ PUNSUBSCRIBE)ZPUBLISHZSPUBLISH)lenupperlist)selfargscommandkeysZ pubsub_typerrr_get_pubsub_keys s    z'AbstractCommandsParser._get_pubsub_keyscKsi}t|d}||d<t|d|d<dd|dD|d<|d |d <|d |d <|d |d<t|dkr|d|d<|d|d<|d|d<|S)NrnameraritycSsg|] }t|qSrr )rflagrrrr-sz;AbstractCommandsParser.parse_subcommand..rflags first_key_pos last_key_pos step_countZtipsZkey_specifications subcommands)r intr)rroptionsZcmd_dictcmd_namerrrparse_subcommand(s        z'AbstractCommandsParser.parse_subcommandN)__name__ __module__ __qualname__rr.rrrrr sr c@s0eZdZdZddZddZddZdd Zd S) 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)rZredis_connectionrrr__init__AszCommandsParser.__init__cCsX|}g}|D] }tdd|Dr||q|D]}||||<q6||_dS)Ncss|]}|VqdSr3)isupper)rxrrr Isz,CommandsParser.initialize..)ranyappendpoplowerr4)rrr4Zuppercase_commandscmdrrrr5Es zCommandsParser.initializec 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)p 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. rNrr( command doesn't exist in Redis commands movablekeysr pubsubrr&r"r$Fr*|Tcsg|] }|qSrrrposrrrrsz+CommandsParser.get_keys..rr=r4splitrr5rrget_get_moveable_keysrr r.absrange) r redis_connrr-cmd_name_splitrr is_subcmd subcmd_namesubcmdr$keys_posrrGrget_keysRsR                 zCommandsParser.get_keysc Gs||dt|dd}z|jd|}WnHtk rv}z*|}d|ksXd|krbWY 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 GETKEYSInvalid arguments The command has no key arguments)rU)rIrexecute_commandr __str__)rrNrpiecesremessagerrrrKs  z!CommandsParser._get_moveable_keysN)r/r0r1__doc__r6r5rTrKrrrrr28s  Ar2c@speZdZdZdZddddZdedddd d Zeee e d fd d dZ eee e d fd ddZ dS)AsyncCommandsParsera% 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. )r4nodeN)returncCs i|_dSr3)r4)rrrrr6szAsyncCommandsParser.__init__r )r_r`cs4|r ||_|jdIdH}dd|D|_dS)NZCOMMANDcSsi|]\}}||qSr)r=)rr?rrrr sz2AsyncCommandsParser.initialize..)r_rXitemsr4)rr_r4rrrr5szAsyncCommandsParser.initialize.)rr`c stdkrdSd}||jkr|}|d}||jkrV|tddn*|IdH||jkrt|d|j|}d|dkr|j IdH}nd|dks|d dkr|j }n|d dkrV|d dkrV|d dkrVd }d|krL|dd}|dD]&}t |d|kr$| |}d}q$|sVdS|d } | dkrxtt | } tt|d | d|d } fdd| D}|S)r@rNrrrArBr rCrr&r"r$Fr*rDTcsg|] }|qSrrrErGrrr sz0AsyncCommandsParser.get_keys..rH) rrr-rOrrrPrQrRr$rSrrGrrTsR                zAsyncCommandsParser.get_keysc shz|jjd|IdH}WnHtk rb}z*|}d|ksDd|krNWY dS|W5d}~XYnX|S)NrUrVrW)rU)r_rXr rY)rrrr[r\rrrrK s z&AsyncCommandsParser._get_moveable_keys)N) r/r0r1r] __slots__r6rr5rrstrrTrKrrrrr^s  Ar^N)typingrrrrrrZredis.exceptionsrr Z redis.utilsr Zredis.asyncio.clusterr r r2r^rrrrs   .u