o iMo@sddlmZddlmZmZmZmZmZmZm Z ddl m Z m Z m Z ddlmZer0ddlmZGdddeZGdd d eZGd d d ZeeeeeffZGd d d ZGdddeZGdddeZdS))Enum) TYPE_CHECKINGAny AwaitableDictOptionalTupleUnion)IncorrectPolicyType RedisError ResponseError str_if_bytes) ClusterNodec@s,eZdZdZdZdZdZdZdZdZ dZ d S) RequestPolicyZ all_nodesZ all_shardsZ all_replicasZ multi_shardspecialdefault_keyless default_keyed default_nodeN) __name__ __module__ __qualname__Z ALL_NODESZ ALL_SHARDSZ ALL_REPLICASZ MULTI_SHARDSPECIALDEFAULT_KEYLESS DEFAULT_KEYEDZ DEFAULT_NODErrN/home/app/Keep/.python/lib/python3.10/site-packages/redis/_parsers/commands.pyr src@s4eZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) ResponsePolicyZ one_succeededZ all_succeededZagg_logical_andZagg_logical_orZagg_minZagg_maxZagg_sumrrrN) rrrZ ONE_SUCCEEDEDZ ALL_SUCCEEDEDZAGG_LOGICAL_ANDZAGG_LOGICAL_ORZAGG_MINZAGG_MAXZAGG_SUMrrrrrrrrsrc@s(eZdZejejfdedefddZdS)CommandPoliciesrequest_policyresponse_policycCs||_||_dSNrr )selfrr rrr__init__$s zCommandPolicies.__init__N)rrrrrrr$rrrrr#src@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|qSrr ).0argrrr ;z;AbstractCommandsParser._get_pubsub_keys..rZPUBSUB)ZCHANNELSZNUMSUBZ SHARDCHANNELSZ SHARDNUMSUB)Z SUBSCRIBEZ PSUBSCRIBEZ UNSUBSCRIBEZ PUNSUBSCRIBE)ZPUBLISHZSPUBLISH)lenupperlist)r#argscommandkeysZ pubsub_typerrr_get_pubsub_keys1s      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)Nrnamer,aritycSr'rr )r(flagrrrr*Sr+z;AbstractCommandsParser.parse_subcommand..r&flags first_key_pos last_key_pos step_counttipsZkey_specifications subcommands)rintr-)r#r1optionsZcmd_dictcmd_namerrrparse_subcommandNs        z'AbstractCommandsParser.parse_subcommandN)rrrr3rFrrrrr%0s r%c@s\eZdZdZddZddZddZdd Z dd ed e ed e fddZ d e fddZ d 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|_||_||jdSr!)commandsredis_connection initialize)r#rIrrrr$gszCommandsParser.__init__cCsX|}g}|D]}tdd|Dr||q|D] }||||<q||_dS)Ncss|]}|VqdSr!)isupper)r(xrrr psz,CommandsParser.initialize..)r1anyappendpoplowerrH)r#rrHZuppercase_commandscmdrrrrJls  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 dkrd}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. r&Nrr,( command doesn't exist in Redis commands movablekeysr7pubsubr4r=r9r;FrB|Tcg|]}|qSrrr(posr0rrr*r+z+CommandsParser.get_keys..r-rQrHsplitr/rJr r.get_get_moveable_keysr3rrFabsrange) r# redis_connr0rEcmd_name_splitr1r2 is_subcmd subcmd_namesubcmdr;keys_posrr\rget_keysysT               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. rr,NCOMMAND GETKEYSInvalid arguments The command has no key arguments)r^r/execute_commandr __str__)r#rcr0piecesr2emessagerrrr`s  z!CommandsParser._get_moveable_keysN command_namesubcommand_namereturncCs|r,|j|dD]}t|d|kr!||}|ddkSq td|d||j|d}|dur=|ddkStd|d a Determines whether a given command or subcommand is considered "keyless". A keyless command does not operate on specific keys, which is determined based on the first key position in the command or subcommand details. If the command or subcommand's first key position is zero or negative, it is treated as keyless. Parameters: command_name: str The name of the command to check. subcommand_name: Optional[str], default=None The name of the subcommand to check, if applicable. If not provided, the check is performed only on the command. Returns: bool True if the specified command or subcommand is considered keyless, False otherwise. Raises: ValueError If the specified subcommand is not found within the command or the specified command does not exist in the available commands. rBrr9z Subcommand z not found in command NzCommand z not found in commandsrHr_rrF ValueErrorr#rrrs subcommandZ parsed_subcmdZcommand_detailsrrr_is_keyless_commands  z"CommandsParser._is_keyless_commandcstifdd|jD]\}}||}|r tj}tj}ntj}tj}|d}t|dkr:|d}|d}nd}|d} |ddurS|t ||di|<n t ||d||<| d } | d } | ro| ||| r| D]C} | d} t | t r| } ||| }|rtj}tj}ntj}tj}| d d } t ||d|| <| ddD]} | || qqsqS)  Retrieve and process the command policies for all commands and subcommands. This method traverses through commands and subcommands, extracting policy details from associated data structures and constructing a dictionary of commands with their associated policies. It supports nested data structures and handles both main commands and their subcommands. Returns: PolicyRecords: A collection of commands and subcommands associated with their respective policies. Raises: IncorrectPolicyType: If an invalid policy type is encountered during policy extraction. c t|ttfrht|}|ds|drd|dr=|dd}z t|||_Wnt y<t d|w|drf|dd}z t |||_ WdSt yct d|wdSdSt|t rz|D]}|||qodSt|tr|D] }|||qdSdSa Recursively extract policies from nested data structures. Args: data: The data structure to search (can be list, dict, str, bytes, etc.) command_name: The command name to associate with found policies rr :r,zIncorrect request policy type: z Incorrect response policy type: N isinstancestrbytesrdecode startswithr^rrrwr rr r/dictvaluesdata module_namerrpolicyZ policy_typeitemvalueZcommand_with_policiesextract_policiesrrrT        z=CommandsParser.get_command_policies..extract_policies.r,rcoreNr"r?rBrX rHitemsrzrrrrr^r-r_rrrrreplacer#r1detailsZ is_keylessZdefault_request_policyZdefault_response_policyZ split_namerrrr?rBZsubcommand_detailsrfZsubcommand_detailrrrget_command_policiessd2           z#CommandsParser.get_command_policiesr!)rrr__doc__r$rJrir`rrboolrz PolicyRecordsrrrrrrG^s C *rGc@seZdZdZdZdddZddedddfd d Zd edee e d ffd dZ d edee e d ffddZ dde dee de fddZdeefddZdS)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. )rHnodertNcCs i|_dSr!)rH)r#rrrr$s zAsyncCommandsParser.__init__rrcs6|r||_|jdIdH}dd|D|_dS)NZCOMMANDcSsi|] \}}||qSr)rQ)r(rSr1rrr sz2AsyncCommandsParser.initialize..)rrmrrH)r#rrHrrrrJs zAsyncCommandsParser.initializer0.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 dkrd}q|sdS|d } | dkrtt | } tt|d | d|d } fdd| D}|S)rTr&Nrr,rUrVr7rWr4r=r9r;FrBrXTcrYrrrZr\rrr*r+z0AsyncCommandsParser.get_keys..r]) r#r0rErdr1r2rerfrgr;rhrr\rrisV              zAsyncCommandsParser.get_keysc sfz|jjdg|RIdH}W|Sty2}z|}d|vs%d|vr,WYd}~dS|d}~ww)Nrjrkrl)rrmr rn)r#r0r2rprqrrrr`s z&AsyncCommandsParser._get_moveable_keysrrrscs|r-|j|dD]}t|d|kr"||}|ddkSq td|d||j|d}|dur>|ddkStd|drurvrxrrrrzs  z'AsyncCommandsParser._is_keyless_commandcsifdd|jD]\}}||IdH}|r$tj}tj}ntj}tj}|d}t|dkr>|d}|d}nd}|d} |ddurW|t ||di|<n t ||d||<| d } | d } | rs| ||| r| D]F} | d} t | t r| } ||| IdH}|rtj}tj}ntj}tj}| d d } t ||d|| <| ddD]} | || qqwqS) r{cr|r}rrrrrr:rzBAsyncCommandsParser.get_command_policies..extract_policiesNrr,rrr"r?rBrXrrrrrrr(sf2         z(AsyncCommandsParser.get_command_policies)rtNr!)rrrr __slots__r$rrJrrrrir`rrzrrrrrrrrs   C *rN)enumrtypingrrrrrrr Zredis.exceptionsr r r Z redis.utilsrZredis.asyncio.clusterrrrrrrrr%rGrrrrrs $     .2