o ;h,+@s~ddlmZmZmZmZmZmZddlmZm Z ddl m Z er&ddl m Z GdddZGdddeZGd d d eZd S) ) TYPE_CHECKINGAnyDictOptionalTupleUnion) RedisError ResponseError str_if_bytes) ClusterNodec@seZdZddZddZdS)AbstractCommandsParsercGst|dkrdSdd|D}|d}d}|dkr-|d}|dvr+|dd}|S|d vr;t|dd}|S|d vrD|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 NcSg|]}t|qSr ).0argrrJ/var/www/html/venv/lib/python3.10/site-packages/redis/_parsers/commands.py z;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|dkrI|d|d<|d|d<|d|d<|S)NrnamerZaritycSrrr )rflagrrrr-rz;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 s r 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] }||||<q||_dS)Ncss|]}|VqdSr4)isupper)rxrrr Isz,CommandsParser.initialize..)ranyappendpoplowerr5)rrr5Zuppercase_commandscmdrrrr6Es  zCommandsParser.initializec stdkrdSd}||jvr>|}|d}||jvr+|tddn||||jvr>t|d|j|}d|dvrU|j |gR}|Sd|dvsa|d dkrh|j }|S|d dkr|d dkr|d dkrd }d|vr|dd}|dD]} t | d|kr| | }d}q|sdS|d } | dkrtt | } 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+|Tcg|]}|qSrrrposrrrrrz+CommandsParser.get_keys..rr>r5splitrr6rrget_get_moveable_keysrr r/absrange) r redis_connrr.cmd_name_splitrr is_subcmd subcmd_namesubcmdr%keys_posrrIrget_keysRsR              zCommandsParser.get_keysc Gsx|dt|dd}z |jdg|R}W|Sty;}z|}d|vs.d|vr5WYd}~dS|d}~ww)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)rKrexecute_commandr __str__)rrPrpiecesremessagerrrrMs  z!CommandsParser._get_moveable_keysN)r0r1r2__doc__r7r6rVrMrrrrr38s  Ar3c@sreZdZdZdZdddZddedddfd d Zd edee e d ffd dZ d edee e d ffddZ 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. )r5nodereturnNcCs i|_dSr4)r5)rrrrr7s zAsyncCommandsParser.__init__rar cs6|r||_|jdIdH}dd|D|_dS)NZCOMMANDcSsi|] \}}||qSr)r>)rr@rrrr sz2AsyncCommandsParser.initialize..)rarZitemsr5)rrar5rrrr6s zAsyncCommandsParser.initializer.c stdkr dSd}||jvrA|}|d}||jvr,|tddn|IdH||jvrAt|d|j|}d|dvrW|j IdH}|Sd|dvsc|d dkrj|j }|S|d dkr|d dkr|d dkrd }d|vr|dd}|dD]}t |d|kr| |}d}q|sdS|d } | dkrtt | } tt|d | d|d } fdd| D}|S)rArNrrrBrCr!rDrr'r#r%Fr+rETcrFrrrGrIrrr rz0AsyncCommandsParser.get_keys..rJ) rrr.rQrrrRrSrTr%rUrrIrrVsT             zAsyncCommandsParser.get_keysc sfz|jjdg|RIdH}W|Sty2}z|}d|vs%d|vr,WYd}~dS|d}~ww)NrWrXrY)rarZr r[)rrrr]r^rrrrM s z&AsyncCommandsParser._get_moveable_keys)rbNr4) r0r1r2r_ __slots__r7rr6rrstrrVrMrrrrr`s  "Ar`N)typingrrrrrrZredis.exceptionsrr Z redis.utilsr Zredis.asyncio.clusterr r r3r`rrrrs   .u