keJiMo.ddlmZddlmZmZmZmZmZmZm Z ddl m Z m Z m Z ddlmZerddlmZGddeZGdd eZGd d ZeeeeeffZGd d ZGddeZGddeZdS))Enum) TYPE_CHECKINGAny AwaitableDictOptionalTupleUnion)IncorrectPolicyType RedisError ResponseError str_if_bytes) ClusterNodec.eZdZdZdZdZdZdZdZdZ dZ d S) 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_NODE]/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/redis/_parsers/commands.pyrr s6IJ!LKG'O#M!LLLr'rc6eZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) 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*s@#M#M'O%NGGGG'O#MMMr'r*c8eZdZejejfdedefdZdS)CommandPoliciesrequest_policyresponse_policyc"||_||_dSNr;r<)selfr;r<s r(__init__zCommandPolicies.__init__$s -.r'N)rrrrr#r*rAr&r'r(r:r:#sM)6(E*8*H//%/(//////r'r:ceZdZdZdZdS)AbstractCommandsParserc0t|dkrdSd|D}|d}d}|dkr)|d}|dvr |dd}n)|dvrt|dd}n |d vr |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 Nc,g|]}t|Sr&r).0args r( z;AbstractCommandsParser._get_pubsub_keys..;s 222c S!!222r'rPUBSUB)CHANNELSNUMSUB SHARDCHANNELS SHARDNUMSUB) SUBSCRIBE PSUBSCRIBE UNSUBSCRIBE PUNSUBSCRIBE)PUBLISHSPUBLISH)lenupperlist)r@argscommandkeys pubsub_types r(_get_pubsub_keysz'AbstractCommandsParser._get_pubsub_keys1s t99q==422T222q'--// h  q'--//KTTTABBx R R RQRR>>DD / / /G9D r'c Bi}t|d}||d<t|d|d<d|dD|d<|d|d <|d |d <|d |d <t|dkr!|d|d<|d|d<|d|d<|S)NrnamerKarityc,g|]}t|Sr&r)rGflags r(rIz;AbstractCommandsParser.parse_subcommand..Ss GGGD\$//GGGr'rEflags first_key_pos last_key_pos step_counttipskey_specifications subcommands)rintrV)r@rZoptionscmd_dictcmd_names r(parse_subcommandz'AbstractCommandsParser.parse_subcommandNs ++# OOGGGAJGGG$+AJ!#*1: !( w<)commandsredis_connection initialize)r@rys r(rAzCommandsParser.__init__gs, 0 -.....r'c|}g}|D]0}td|Dr||1|D],}||||<-||_dS)Nc3>K|]}|VdSr>)isupper)rGxs r( z,CommandsParser.initialize..ps*,,1199;;,,,,,,r')rZanyappendpoplowerrx)r@rrxuppercase_commandscmds r(rzzCommandsParser.initializels99;; / /C,,,,,,, /"))#...% 6 6C$,LL$5$5HSYY[[ ! !  r'ctdkrdSd}||jvr|}|d}||jvr|t ddznB||||jvr$t |d|j|}d|dvr|j |gR}nd|dvs |d dkr |j }n|d dkr|d dkr||d dkrpd }d|vrf|dd}|dD]>} t| d|kr#| | }|d dkrd}?|sdS|d } | dkrtt| z } t t|d | dz|d } fd| 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. rENrrK( command doesn't exist in Redis commands movablekeysrcpubsubr_rirergFro|Tc g|] }| Sr&r&rGposrYs r(rIz+CommandsParser.get_keys..222#DI222r'rVrrxsplitrXrzr rWget_get_moveable_keysr]rrtabsrange) r@ redis_connrYrscmd_name_splitrZr[ is_subcmd subcmd_namesubcmdrgkeys_poss ` r(get_keyszCommandsParser.get_keysysr t99q==47==?? 4= ( (&^^--N%a(H4=((%T!""X6 +++4=00$#>>++UUU-##H-- GG, , ,*4*:====DD ) ) )WV_-H-H(4($/DD %**O,11N+q00!  G++%-"A"AQ "A"AK")-"811'q 22kAA&*&;&;F&C&CG&7!;;,0 ! 4">2La"4yy3|+<+<< go. q0@',BWXXH3222222D r'c|dt|ddz} |jdg|R}n6#t$r)}|}d|vsd|vrYd}~dS|d}~wwxYw|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. rrKNCOMMAND GETKEYSInvalid arguments The command has no key arguments)rrXexecute_commandr __str__)r@rrYpiecesr[emessages r(rz!CommandsParser._get_moveable_keyssa4QRR>>1 -:-.?I&IIIDD   iikkG#w..5@@ttttt  sA A6 A1/A11A6N command_namesubcommand_namereturncf|rt|j|dD]>}t|d|kr#||}|ddkcS?t d|d||j|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. rorrez Subcommand z not found in command NzCommand z not found in commandsrxrrrt ValueErrorr@rr subcommand parsed_subcmdcommand_detailss r(_is_keyless_commandz"CommandsParser._is_keyless_commands6  N"m// ==mL ? ?  1 ../AA$($9$9*$E$EM(9Q>>>>BSoSS\SS #m// dCCO*&71<<L LLLMM Mr'cifd|jD]\}}||}|rtj}t j}ntj}t j}|d}t|dkr|d}|d}n d}|d} |d|t||i|<nt||||<| d} | d } | r | ||| r| D]} | d} t| tr| } ||| }|rtj}t j}ntj}t j}| d 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. crt|ttfr4t|}|ds|dr|dr]|dd} t|||_n #t$rtd|wxYw|dr`|dd} t|||_ dS#t$rtd|wxYwdSdSt|tr|D]}|||dSt|tr$|D]}|||dSdSa 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<:rKzIncorrect request policy type: z Incorrect response policy type: N isinstancestrbytesrdecode startswithrrr;rr r*r<rXdictvalues data module_namerpolicy policy_typeitemvaluecommand_with_policiesextract_policiess r(rz=CommandsParser.get_command_policies..extract_policiesB$e --( G%dkkmm44$$%566&:K:K%;;(()9:: &,ll3&7&7&: 0=[/I/I2+> ,,n)"5 O+ O O## (():;; &,ll3&7&7&: 1?{0K0K2+> ,-oo)"5 P; P P##-  D$'' G FFD$$T; EEEEFFD$'' G![[]]GGE$$UKFFFF G GGG B<<C D//E .rKrcoreNr?rkror rxitemsrrr#r*r$rrVrr:rrrreplacer@rZdetails is_keylessdefault_request_policydefault_response_policy split_namerrrkrosubcommand_detailsrsubcommand_detailrrs @@r(get_command_policiesz#CommandsParser.get_command_policiess !#0 G0 G0 G0 G0 G0 Gd!% 3 3 5 5F VF V GW11'::J G)6)F&*8*H'')6)D&*8*F'!s++J:""(m )!} $ )!} %((d;;C /'=(?###6%k22DS#9$;DDD%k2<@ ;;v&&D!++m44K B  {LAAA V*5VV&"4Q"7K!+u55;&1&8&8&:&: "&!9!9';!O!OJ!O1>1N.2@2P//1>1L.2@2N/"-"5"5c3"?"?KGV'=(?GGG)+6{C .@-CVV)(():KUUUUV%$r'r>)rrr__doc__rArzrrrrboolr PolicyRecordsrr&r'r(rvrv^s/// !!!AAAF4CG(N(N(N2:3-(N (N(N(N(NTL%mL%L%L%L%L%L%r'rvceZdZdZdZddZddedddfdZd edee e d ffd Z d edee e d ffd Z dd e dee de fdZdeefdZdS)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. )rxnoderNci|_dSr>)rx)r@s r(rAzAsyncCommandsParser.__init__s ?A r'rrcK|r||_|jdd{V}d|D|_dS)NCOMMANDc>i|]\}}||Sr&)r)rGrrZs r( z2AsyncCommandsParser.initialize..s&SSS,#wgSSSr')rrrrx)r@rrxs r(rzzAsyncCommandsParser.initializes_  DI229========SS(..BRBRSSS r'rY.c0KtdkrdSd}||jvr|}|d}||jvr|t ddznG|d{V||jvr$t |d|j|}d|dvr|j d{V}nd|dvs |d dkr |j }n|d dkr|d dkr||d dkrpd }d|vrf|dd}|dD]>}t|d|kr#| |}|d dkrd}?|sdS|d } | dkrtt| z } t t|d | dz|d } fd| D}|S)rrENrrKrrrcrr_rirergFrorTc g|] }| Sr&r&rs r(rIz0AsyncCommandsParser.get_keys..rr'r) r@rYrsrrZr[rrrrgrs ` r(rzAsyncCommandsParser.get_keyss t99q==47==?? 4= ( (&^^--N%a(H4=((%T!""X6oo'''''''''4=00$#>>++UUU-##H-- GG, , ,00$7777777DD ) ) )WV_-H-H(4($/DD %**O,11N+q00!  G++%-"A"AQ "A"AK")-"811'q 22kAA&*&;&;F&C&CG&7!;;,0 ! 4">2La"4yy3|+<+<< go. q0@',BWXXH3222222D r'cK |jjdg|Rd{V}n6#t$r)}|}d|vsd|vrYd}~dS|d}~wwxYw|S)Nrrr)rrr r)r@rYr[rrs r(rz&AsyncCommandsParser._get_moveable_keyss 223DLtLLLLLLLLLDD   iikkG#w..5@@ttttt  s AA  A  ArrcjK|rt|j|dD]>}t|d|kr#||}|ddkcS?t d|d||j|d}| |ddkSt d|drrrs r(rz'AsyncCommandsParser._is_keyless_commands6  N"m// ==mL ? ?  1 ../AA$($9$9*$E$EM(9Q>>>>BSoSS\SS #m// dCCO*&71<<L LLLMM Mr'c$Kifd|jD]\}}||d{V}|rtj}t j}ntj}t j}|d}t|dkr|d}|d}n d}|d} |d|t||i|<nt||||<| d} | d } | r | ||| r| D]} | d} t| tr| } ||| d{V}|rtj}t j}ntj}t j}| d d } t|||| <| ddD]} | || ͐S) rcrt|ttfr4t|}|ds|dr|dr]|dd} t|||_n #t$rtd|wxYw|dr`|dd} t|||_ dS#t$rtd|wxYwdSdSt|tr|D]}|||dSt|tr$|D]}|||dSdSrrrs r(rzBAsyncCommandsParser.get_command_policies..extract_policies:rrNrrKrrr?rkrorrrrs @@r(rz(AsyncCommandsParser.get_command_policies(s !#0 G0 G0 G0 G0 G0 Gd!% 3 3 5 5F VF V GW#77@@@@@@@@J G)6)F&*8*H'')6)D&*8*F'!s++J:""(m )!} $ )!} %((d;;C /'=(?###6%k22DS#9$;DDD%k2<@ ;;v&&D!++m44K B  {LAAA V*5VV&"4Q"7K!+u55;&1&8&8&:&: (,'?'?'U'U!U!U!U!U!U!UJ!O1>1N.2@2P//1>1L.2@2N/"-"5"5c3"?"?KGV'=(?GGG)+6{C .@-CVV)(():KUUUUV%$r')rNr>)rrrr __slots__rArrzrr rrrrrrrrr&r'r(rrsA  %IBBBBTTXm%<TTTTTACAHU38_,EAAAAF c huS#X6O    CG(N(N(N2:3-(N (N(N(N(NTL%Im,DL%L%L%L%L%L%r'rN)enumrtypingrrrrrr r redis.exceptionsr r r redis.utilsrredis.asyncio.clusterrrr*r:rrrrCrvrr&r'r(rsNNNNNNNNNNNNNNNNNNKKKKKKKKKK$$$$$$2111111"""""D""" $ $ $ $ $T $ $ $////////S$sO3445 ++++++++\m%m%m%m%m%+m%m%m%` f%f%f%f%f%0f%f%f%f%f%r'