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 ).0argrrI/var/www/html/env/lib/python3.10/site-packages/redis/_parsers/commands.py z;AbstractCommandsParser._get_pubsub_keys..rPUBSUB)CHANNELSNUMSUB SHARDCHANNELS SHARDNUMSUB) SUBSCRIBE PSUBSCRIBE UNSUBSCRIBE PUNSUBSCRIBE)PUBLISHSPUBLISH)lenupperlist)selfargscommandkeys 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)NrnameraritycSrrr )rflagrrrr-rz;AbstractCommandsParser.parse_subcommand..rflags first_key_pos last_key_pos step_counttipskey_specifications subcommands)r intr")r%r'optionscmd_dictcmd_namerrrparse_subcommand(s        z'AbstractCommandsParser.parse_subcommandN)__name__ __module__ __qualname__r*r?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)r%redis_connectionrrr__init__AszCommandsParser.__init__cCsX|}g}|D]}tdd|Dr||q|D] }||||<q||_dS)Ncss|]}|VqdSrD)isupper)rxrrr Isz,CommandsParser.initialize..)r'anyappendpoplowerrE)r%rrEuppercase_commandscmdrrrrFEs  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.pubsubr+r4r0r2Fr:|Tcg|]}|qSrrrposr&rrrrz+CommandsParser.get_keys..r"rOrEsplitr$rFrr#get_get_moveable_keysr*r r?absrange) r% redis_connr&r>cmd_name_splitr'r( is_subcmd subcmd_namesubcmdr2keys_posrr[rget_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)r]r$execute_commandr __str__)r%rbr&piecesr(emessagerrrr_s  z!CommandsParser._get_moveable_keysN)r@rArB__doc__rHrFrhr_rrrrrC8s  ArCc@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. )rEnodereturnNcCs i|_dSrD)rE)r%rrrrHs zAsyncCommandsParser.__init__rsr cs6|r||_|jdIdH}dd|D|_dS)NCOMMANDcSsi|] \}}||qSr)rO)rrRr'rrr sz2AsyncCommandsParser.initialize..)rsrlitemsrE)r%rsrErrrrFs 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)rSrNrrrTrUr.rVr+r4r0r2Fr:rWTcrXrrrYr[rrr rz0AsyncCommandsParser.get_keys..r\) r%r&r>rcr'r(rdrerfr2rgrr[rrhsT             zAsyncCommandsParser.get_keysc sfz|jjdg|RIdH}W|Sty2}z|}d|vs%d|vr,WYd}~dS|d}~ww)Nrirjrk)rsrlr rm)r%r&r(rorprrrr_ s z&AsyncCommandsParser._get_moveable_keys)rtNrD) r@rArBrq __slots__rHrrFrrstrrhr_rrrrrrs  "ArrN)typingrrrrrrredis.exceptionsrr redis.utilsr redis.asyncio.clusterr r rCrrrrrrs   .u