jModdlmZddlmZmZmZmZmZmZm Z ddl m Z m Z m Z ddlmZerddlmZGddeZGdd eZGd d ZeeeeeffZGd d ZGddeZGddeZy))Enum) TYPE_CHECKINGAny AwaitableDictOptionalTupleUnion)IncorrectPolicyType RedisError ResponseError) str_if_bytes) ClusterNodec,eZdZdZdZdZdZdZdZdZ dZ y ) RequestPolicy all_nodes all_shards all_replicas multi_shardspecialdefault_keyless default_keyed default_nodeN) __name__ __module__ __qualname__ ALL_NODES ALL_SHARDS ALL_REPLICAS MULTI_SHARDSPECIALDEFAULT_KEYLESS DEFAULT_KEYED DEFAULT_NODEa/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/redis/_parsers/commands.pyrr s*IJ!LKG'O#M!Lr&rc4eZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) ResponsePolicy one_succeeded all_succeededagg_logical_andagg_logical_oragg_minagg_maxagg_sumrrrN) rrr ONE_SUCCEEDED ALL_SUCCEEDEDAGG_LOGICAL_ANDAGG_LOGICAL_ORAGG_MINAGG_MAXAGG_SUMr!r"r#r%r&r'r)r)s4#M#M'O%NGGGG'O#Mr&r)cJeZdZejejfdedefdZy)CommandPoliciesrequest_policyresponse_policyc ||_||_yNr:r;)selfr:r;s r'__init__zCommandPolicies.__init__$s -.r&N)rrrrr"r)r@r%r&r'r9r9#s-)6(E(E*8*H*H/%/(/r&r9ceZdZdZdZy)AbstractCommandsParserc t|dkry|Dcgc] }t|}}|dj}d}|dk(r|dj}|dvr|dd}|S|dvrt|dd}|S|dvr|dg}|Scc}w) 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 NrPUBSUB)CHANNELSNUMSUB SHARDCHANNELS SHARDNUMSUB) SUBSCRIBE PSUBSCRIBE UNSUBSCRIBE PUNSUBSCRIBE)PUBLISHSPUBLISH)lenrupperlist)r?argsargcommandkeys pubsub_types r'_get_pubsub_keysz'AbstractCommandsParser._get_pubsub_keys1s t9q=-12Tc S!T2q'--/ h q'--/KTTABx R RQR>D  / /G9D #3sBc i}t|d}||d<t|d|d<|dDcgc] }t|c}|d<|d|d<|d |d <|d |d <t|d kDr|d |d<|d|d<|d|d<|Scc}w)NrnamerFarityrDflags first_key_pos last_key_pos step_counttipskey_specifications subcommands)rintrQ)r?rVoptionscmd_dictcmd_nameflags r'parse_subcommandz'AbstractCommandsParser.parse_subcommandNs +# Oz,CommandsParser.initialize..ps,1199;s)rVanyappendpoplowerrs)r?rrsuppercase_commandscmds r'ruzCommandsParser.initializelse99;C,,,"))#.&C$,LL$5HSYY[ !&  r&ct|dkry|dj}||jvrp|j}|d}||jvr|t |ddz}n;|j |||jvrt |jd|jj|}d|dvr|j|g|}|Sd|dvs|d dk(r|j|}|S|d dk(ri|d dk(ra|d dk(rYd }d|vrP|d|dj}|dD]0} t| d|k(s|j| }|d dkDs/d}2|sy|d } | dkrt|t| z } t t|d | dz|d } | D cgc]} ||  }} |Scc} wap 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. rDNrrFz( command doesn't exist in Redis commands movablekeysr]pubsubr[rcr_raFri|TrQrrssplitrSrur rRget_get_moveable_keysrYrroabsrange) r? redis_connrTrmcmd_name_splitrVrW is_subcmd subcmd_namesubcmdrakeys_posposs r'get_keyszCommandsParser.get_keysys t9q=7==? 4== (&^^-N%a(H4==(%T!"X6 +4==0$#>>+,,TU--##H- GG, ,*4**:==D< ;) )WV_-H(4(($/D8 3 %*O,1N+q0!  G+%-JaQ /@"AK")-"8'q 2kA&*&;&;F&CG&7!;,0 #9!">2La"4y3|+<< go. q0@',BWXH*22#DID2 3s- F=c|djt|ddz} |jdg|}|S#t$r$}|j }d|vsd|vrYd}~y|d}~wwxYw)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. rrFNCOMMAND GETKEYSInvalid arguments The command has no key arguments)rrSexecute_commandr __str__)r?rrTpiecesrWemessages r'rz!CommandsParser._get_moveable_keyss{a4QR>1 -:--.?I&ID  iikG#w.5@ s9 A&A!A!!A&N command_namesubcommand_namereturnc0|r`|jj|dD].}t|d|k(s|j|}|ddkcSt d|d||jj|d}||ddkSt d|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. rirr_z Subcommand z not found in command NzCommand z not found in commandsrsrrro ValueErrorr?rr subcommand parsed_subcmdcommand_detailss r'_is_keyless_commandz"CommandsParser._is_keyless_commands6 "mm// =mL  1 ./A$($9$9*$EM(9Q>>Mo..D\NS #mm// dCO*&71<<x ~5KLM Mr&cifd|jjD]\}}|j|}|r!tj}t j}n tj }t j }|jd}t|dkDr |d}|d}nd}|d}j|d|t||i|<nt||||<|jd} |jd } | r | ||| s| D]} | d} t| tr| j} |j|| }|r!tj}t j}n tj }t j }| jd d } t|||| <| ddD] } | || S)  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. ct|ttfrt|j }|j ds|j drz|j dr+|j dd} t|||_|j dr,|j dd} t|||_ yyyt|tr|D] }|||yt|tr |jD] }|||yy#t$rtd|wxYw#t$rtd|wxYwa 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 r:r;:rFzIncorrect request policy type: z Incorrect response policy type: N isinstancestrbytesrdecode startswithrrr:rr r)r;rSdictvalues data module_namerpolicy policy_typeitemvaluecommand_with_policiesextract_policiess r'rz=CommandsParser.get_command_policies..extract_policies|$e -%dkkm4$$%56&:K:K%;(()9:&,ll3&7&: 0=[/I2+> ,, (():;&,ll3&7&: 1?{0K2+> ,- <;6D$' D$T; E!D$'![[]E$UKF+(- *"5"A+ O# *"5"B;- P#9D!5D<!D9<E.rFrcoreNr>rerir rsitemsrrr"r)r#rrQrr9rrrreplacer?rVdetails is_keylessdefault_request_policydefault_response_policy split_namerrrerisubcommand_detailsrsubcommand_detailrrs @@r'get_command_policiesz#CommandsParser.get_command_policiess !#0 Gd!% 3 3 5 GW11':J)6)F)F&*8*H*H')6)D)D&*8*F*F'!s+J:"(m )!} $ )!} %((d;C /'=(?#6%k2DS#9$;D%k2<@ ;;v&D!++m4K {LA*5&"4Q"7K!+u5&1&8&8&: "&!9!9';!OJ!1>1N1N.2@2P2P/1>1L1L.2@2N2N/"-"5"5c3"?KGV'=(?G)+6{C .@-C)():KU.D3+6Y!6P%$r&r=)rrr__doc__r@rurrrrboolr PolicyRecordsrr%r&r'rqrq^sX/ !AF4CG(N(N2:3-(N (NTL%mL%r&rqceZdZdZdZddZddedddfdZd edee e d ffd Z d edee e d ffd Z dd e dee de fdZdeefdZy)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. )rsnoderNci|_yr=)rs)r?s r'r@zAsyncCommandsParser.__init__s ?A r&rrcK|r||_|jjdd{}|jDcic]\}}|j|c}}|_y7Ss!(A/A'A/A) A/)A/rT.cKt|dkry|dj}||jvrw|j}|d}||jvr|t |ddz}nB|j d{||jvrt |jd|jj|}d|dvr|j|d{}|Sd|dvs|d dk(r|j|}|S|d dk(ri|d dk(ra|d dk(rYd }d|vrP|d|dj}|dD]0}t|d|k(s|j|}|d dkDs/d}2|sy|d } | dkrt|t| z } t t|d | dz|d } | D cgc]} ||  }} |S7O7cc} wwrr) r?rTrmrrVrWrrrrarrs r'rzAsyncCommandsParser.get_keyss. t9q=7==? 4== (&^^-N%a(H4==(%T!"X6oo'''4==0$#>>+,,TU--##H- GG, ,000$77D< ;) )WV_-H(4(($/D8 3 %*O,1N+q0!  G+%-JaQ /@"AK")-"8'q 2kA&*&;&;F&CG&7!;,0 #9!">2La"4y3|+<< go. q0@',BWXH*22#DID2 M(883sEA9G;G >Mo..D\NS #mm// dCO*&71<<x ~5KLM Ms 5BA"BcKifd|jjD]\}}|j|d{}|r!tj}t j}n tj }t j }|jd}t|dkDr |d}|d}nd}|d}j|d|t||i|<nt||||<|jd} |jd } | r | ||| s| D]} | d} t| tr| j} |j|| d{}|r!tj}t j}n tj }t j }| jd d } t|||| <| ddD] } | || S77w) rct|ttfrt|j }|j ds|j drz|j dr+|j dd} t|||_|j dr,|j dd} t|||_ yyyt|tr|D] }|||yt|tr |jD] }|||yy#t$rtd|wxYw#t$rtd|wxYwrrrs r'rzBAsyncCommandsParser.get_command_policies..extract_policies:rrNrrFrrr>rerirrrrs @@r'rz(AsyncCommandsParser.get_command_policies(s! !#0 Gd!% 3 3 5 GW#77@@J)6)F)F&*8*H*H')6)D)D&*8*F*F'!s+J:"(m )!} $ )!} %((d;C /'=(?#6%k2DS#9$;D%k2<@ ;;v&D!++m4K {LA*5&"4Q"7K!+u5&1&8&8&: (,'?'?'U!UJ!1>1N1N.2@2P2P/1>1L1L.2@2N2N/"-"5"5c3"?KGV'=(?G)+6{C .@-C)():KU.D3+6Y!6P%$MAb"Vs+>G0G+C G0$?G0#G.$BG0.G0)rNr=)rrrr __slots__r@rrurr rrrrrrrrr%r&r'rrs %IBTXm%<TTACAHU38_,EAF c huS#X6O CG(N(N2:3-(N (NTL%Im,DL%r&rN)enumrtypingrrrrrr r redis.exceptionsr r r redis.utilsrredis.asyncio.clusterrrr)r9rrrrBrqrr%r&r'rsNNNKK$1"D" $T $//S$sO3445 ++\m%+m%` f%0f%r&