+hJ{ddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZmZddlmZmZmZmZmZmZddlmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+dd l,m-Z-dd l.m/Z/erdd l0m1Z1e2gd Z3Gd deZ4Gdde4Z5Gdde'Z6Gdde6e!Z7Gdde$Z8Gdde8eZ9Gdde4e6ee)e8e+e%e&e(e/ Z:Gdde5e7ee9e#ee e" Z;y)N) TYPE_CHECKINGAny AsyncIteratorDictIterableIteratorListMappingNoReturnOptionalUnion)Literal)key_slot)RedisClusterException RedisError)AnyKeyTClusterCommandsProtocol EncodableTKeysTKeyTPatternT) ACLCommandsAsyncACLCommandsAsyncDataAccessCommandsAsyncFunctionCommandsAsyncGearsCommandsAsyncManagementCommandsAsyncModuleCommandsAsyncScriptCommandsDataAccessCommandsFunctionCommands GearsCommandsManagementCommandsModuleCommandsPubSubCommands ResponseTScriptCommands) list_or_args)RedisModuleCommands) TargetNodesT))BITCOUNTBITPOSEVAL_RO EVALSHA_ROEXISTSGEODISTGEOHASHGEOPOS GEORADIUSGEORADIUSBYMEMBERGETGETBITGETRANGEHEXISTSHGETHGETALLHKEYSHLENHMGETHSTRLENHVALSKEYSLINDEXLLENLRANGEMGETPTTL RANDOMKEYSCARDSDIFFSINTER SISMEMBERSMEMBERS SRANDMEMBERSTRLENSUNIONTTLZCARDZCOUNTZRANGEZSCOREc XeZdZdZdeedeeeeffdZ de e e fdeeee ffdZ dede eee fdeefd Zdeede eee fd eedeefd Zded edeeefd Zde e e fdeefdZdededefdZdedefdZdedefdZdedefdZdedefdZy)ClusterMultiKeyCommandsC A class containing commands that handle more than one key keysreturnci}|D]G}t|jj|}|j|gj |I|S)z@Split keys into a dictionary that maps a slot to a list of keys.)rencoderencode setdefaultappend)selfrX slots_to_keyskeyslots I/var/www/html/venv/lib/python3.12/site-packages/redis/commands/cluster.py_partition_keys_by_slotz/ClusterMultiKeyCommands._partition_keys_by_slotjsT  ;CDLL//45D  $ $T2 . 5 5c : ;mappingci}|jD]J}t|jj|d}|j |gj |L|S)zBSplit pairs into a dictionary that maps a slot to a list of pairs.r)itemsrr[r\r]extend)r_rfslots_to_pairspairrbs rc_partition_pairs_by_slotz0ClusterMultiKeyCommands._partition_pairs_by_slotts^ MMO =DDLL//Q89D  % %dB / 6 6t < =recommand slots_to_argsc |jxr|tv}|j}|jDcgc]6\}}|j|g|d|j j ||gi8c}}|jScc}}wN target_nodes)read_from_replicas READ_COMMANDSpipelinerhexecute_command nodes_managerget_node_from_slotexecuter_rmrnrrpiperb slot_argss rc_execute_pipeline_by_slotz1ClusterMultiKeyCommands._execute_pipeline_by_slots"44QM9Q}}$1#6#6#8  i !D   &&99$@RS  ||~ s;B  responsesc t|j|Dcic]\}}t||D]\}}|| }}}}}|D cgc]} ||  c} Scc}}}}wcc} wN)zipvalues) r_rXrnr} slot_valuesresponsekvresultsras rc_reorder_keys_by_commandz0ClusterMultiKeyCommands._reorder_keys_by_commands*-]-A-A-CY)O  % XK2 1 qD   )-- --  .s !A A!argsct||}|j|}|jd|}|j|||S)A Splits the keys into different slots and then calls MGET for the keys of every slot. This operation will not be atomic if keys belong to more than one slot. Returns a list of values ordered identically to ``keys`` For more information see https://redis.io/commands/mget rEr)rdr|rr_rXrr`ress rcmget_nonatomicz&ClusterMultiKeyCommands.mget_nonatomicsMD$'44T: ,,V]C,,T=#FFrecH|j|}|jd|S) Sets key/values based on a mapping. Mapping is a dictionary of key/value pairs. Both keys and values should be strings or types that can be cast to a string via str(). Splits the keys into different slots and then calls MSET for the keys of every slot. This operation will not be atomic if keys belong to more than one slot. For more information see https://redis.io/commands/mset MSETrlr|r_rfrjs rcmset_nonatomicz&ClusterMultiKeyCommands.mset_nonatomics)66w?--fnEErecZ|j|}t|j||S)v Runs the given command once for the keys of each slot. Returns the sum of the return values. rdsumr|r_rmrXr`s rc_split_command_across_slotsz3ClusterMultiKeyCommands._split_command_across_slotss.44T: 411'=IJJrec(|jdg|S)z Returns the number of ``names`` that exist in the whole cluster. The keys are first split up into slots and then an EXISTS command is sent for every slot For more information see https://redis.io/commands/exists r0rr_rXs rcexistszClusterMultiKeyCommands.existss0t//@4@@rec(|jdg|S)a< Deletes the given keys in the cluster. The keys are first split up into slots and then an DEL command is sent for every slot Non-existant keys are ignored. Returns the number of keys that were deleted. For more information see https://redis.io/commands/del DELrrs rcdeletezClusterMultiKeyCommands.deletes0t//===rec(|jdg|S)aa Updates the last access time of given keys across the cluster. The keys are first split up into slots and then an TOUCH command is sent for every slot Non-existant keys are ignored. Returns the number of keys that were touched. For more information see https://redis.io/commands/touch TOUCHrrs rctouchzClusterMultiKeyCommands.touchs0t//?$??rec(|jdg|S)aM Remove the specified keys in a different thread. The keys are first split up into slots and then an TOUCH command is sent for every slot Non-existant keys are ignored. Returns the number of keys that were unlinked. For more information see https://redis.io/commands/unlink UNLINKrrs rcunlinkzClusterMultiKeyCommands.unlinks0t//@4@@reN)__name__ __module__ __qualname____doc__rrrintr rdr rrrlstrrr|rrr rboolrrr'rrrrrercrVrVesHTNtCdO?T w 23  c4 ## $ +238L3L+M c" .tn .sHZ$889 .C= . c .G5GG$x}:MG.Fggz.A&BFtDzF& K3 Kt K KADAYA >D >Y > @4 @I @ AD AY ArerVceZdZdZdededeeefdZ de e e fdee fdZdededefd Zded e eee fdeefd Zy ) AsyncClusterMultiKeyCommandsrWrXrrYcKt||}|j|}|jd|d{}|j|||S7w)rrENrrs rcrz+AsyncClusterMultiKeyCommands.mget_nonatomicsZD$'44T: 226=II,,T=#FFJs3AA ArfcdK|j|}|jd|d{S7w)rrNrrs rcrz+AsyncClusterMultiKeyCommands.mset_nonatomic%s366w?33FNKKKKs '0.0rmcvK|j|}t|j||d{S7w)rNrrs rcrz8AsyncClusterMultiKeyCommands._split_command_across_slots8s844T: 77OOPPOs ,97 9rnc K|jr|jd{|jxr|tv}|j }|j Dcgc]6\}}|j |g|d|jj||gi8c}}|jd{S7cc}}w7 wrp) _initialize initializerrrsrtrhrurvrwrxrys rcr|z6AsyncClusterMultiKeyCommands._execute_pipeline_by_slotDs   //# # #!44QM9Q}}$1#6#6#8  i !D   &&99$@RS  \\^## $ $s- CB7=C ;B9C2B?3C9CN)rrrrrrr r rrr rrrrrrrrr|rrercrr sGGtGXc]@SG.LGGZ4G,HLTRVZL& Q QT Qc Q$$+238L3L+M$ c$rerc eZdZdZdefdZdefdZdefdZdddefdZ ddd e defd Z ddd e defd Z d e defd ZdedefdZd e deefdZd e defdZ d3dddeedefdZd3deddefdZdedefdZ d3dede deddefdZdefdZdddedefdZ d4dededdefdZ d3deddefd Zd!e d"e defd#Z d3d$e deddefd%Zddded e d&edef d'Z d e defd(Z! d3dededdefd)Z"d3deddefd*Z#d3d+Z$d3d,Z%dddefd-Z&d3dedddfd.Z'd3dedddfd/Z(d3deddefd0Z)d3deddefd1Z*defd2Z+y)5ClusterManagementCommands A class for Redis Cluster management commands The class inherits from Redis's core ManagementCommands class and do the required adjustments to work with cluster mode rYctd)z Make the server a replica of another instance, or promote it as master. For more information see https://redis.io/commands/slaveof z(SLAVEOF is not supported in cluster moderr_rkwargss rcslaveofz!ClusterManagementCommands.slaveof`s $$NOOrectd)z Make the server a replica of another instance, or promote it as master. For more information see https://redis.io/commands/replicaof z*REPLICAOF is not supported in cluster moderrs rc replicaofz#ClusterManagementCommands.replicaofhs $$PQQrectd)zo Swaps two Redis databases. For more information see https://redis.io/commands/swapdb z'SWAPDB is not supported in cluster moderrs rcswapdbz ClusterManagementCommands.swapdbps $$MNNre target_noder+c(|jd|S)z Returns the node's id. :target_node: 'ClusterNode' The node to execute the command on For more information check https://redis.io/commands/cluster-myid/ z CLUSTER MYIDrqrur_rs rc cluster_myidz&ClusterManagementCommands.cluster_myidxs##N#MMreslotsc.|jdg|d|iS)z Assign new hash slots to receiving node. Sends to specified node. :target_node: 'ClusterNode' The node to execute the command on For more information see https://redis.io/commands/cluster-addslots zCLUSTER ADDSLOTSrqrr_rrs rccluster_addslotsz*ClusterManagementCommands.cluster_addslotss,$t##  !& 5@  rec.|jdg|d|iS)a Similar to the CLUSTER ADDSLOTS command. The difference between the two commands is that ADDSLOTS takes a list of slots to assign to the node, while ADDSLOTSRANGE takes a list of slot ranges (specified by start and end slots) to assign to the node. :target_node: 'ClusterNode' The node to execute the command on For more information see https://redis.io/commands/cluster-addslotsrange zCLUSTER ADDSLOTSRANGErqrrs rccluster_addslotsrangez/ClusterManagementCommands.cluster_addslotsranges,$t## # &+ :E  reslot_idc&|jd|S)z Return the number of local keys in the specified hash slot Send to node based on specified slot_id For more information see https://redis.io/commands/cluster-countkeysinslot zCLUSTER COUNTKEYSINSLOTrr_rs rccluster_countkeysinslotz1ClusterManagementCommands.cluster_countkeysinslots##$=wGGrenode_idc&|jd|S)z Return the number of failure reports active for a given node Sends to a random node For more information see https://redis.io/commands/cluster-count-failure-reports zCLUSTER COUNT-FAILURE-REPORTSr)r_rs rccluster_count_failure_reportz6ClusterManagementCommands.cluster_count_failure_reports##$CWMMrecL|Dcgc]}|jd|c}Scc}w) Set hash slots as unbound in the cluster. It determines by it self what node the slot is in and sends it there Returns a list of the results for each processed slot. For more information see https://redis.io/commands/cluster-delslots CLUSTER DELSLOTSr)r_rrbs rccluster_delslotsz*ClusterManagementCommands.cluster_delslotss(LQQ4$$%7>QQQs!c(|jdg|S)aQ Similar to the CLUSTER DELSLOTS command. The difference is that CLUSTER DELSLOTS takes a list of hash slots to remove from the node, while CLUSTER DELSLOTSRANGE takes a list of slot ranges to remove from the node. For more information see https://redis.io/commands/cluster-delslotsrange zCLUSTER DELSLOTSRANGErr_rs rccluster_delslotsrangez/ClusterManagementCommands.cluster_delslotsranges$t##$;DeDDreNoptionc|r4|jdvrtd||jd||S|jd|S)a  Forces a slave to perform a manual failover of its master Sends to specified node :target_node: 'ClusterNode' The node to execute the command on For more information see https://redis.io/commands/cluster-failover )FORCETAKEOVERz-Invalid option for CLUSTER FAILOVER command: zCLUSTER FAILOVERr)upperrru)r_rrs rccluster_failoverz*ClusterManagementCommands.cluster_failoversg ||~%:: CF8L++&[,''(:'U Urerqc(|jd|S)z Provides info about Redis Cluster node state. The command will be sent to a random node in the cluster if no target node is specified. For more information see https://redis.io/commands/cluster-info z CLUSTER INFOrrr_rqs rc cluster_infoz&ClusterManagementCommands.cluster_infos##N#NNrerac&|jd|S)z Returns the hash slot of the specified key Sends to random node in the cluster For more information see https://redis.io/commands/cluster-keyslot zCLUSTER KEYSLOTr)r_ras rccluster_keyslotz)ClusterManagementCommands.cluster_keyslots##$5s;;rehostportc,|jd|||S)z Force a node cluster to handshake with another node. Sends to specified node. For more information see https://redis.io/commands/cluster-meet z CLUSTER MEETrr)r_rrrqs rc cluster_meetz&ClusterManagementCommands.cluster_meets%## D$\$  rec$|jdS)z Get Cluster config for the node. Sends to random node in the cluster For more information see https://redis.io/commands/cluster-nodes z CLUSTER NODESr)r_s rc cluster_nodesz'ClusterManagementCommands.cluster_nodess##O44rec*|jd||S)z Reconfigure a node as a slave of the specified master node For more information see https://redis.io/commands/cluster-replicate zCLUSTER REPLICATErr)r_rqrs rccluster_replicatez+ClusterManagementCommands.cluster_replicate s### |$  resoftc>|jd|rd|Sd|S)z Reset a Redis Cluster node If 'soft' is True then it will send 'SOFT' argument If 'soft' is False then it will send 'HARD' argument For more information see https://redis.io/commands/cluster-reset z CLUSTER RESETsSOFTsHARDrr)r_rrqs rc cluster_resetz'ClusterManagementCommands.cluster_resets9## W $  29 $  rec(|jd|S)z Forces the node to save cluster state on disk For more information see https://redis.io/commands/cluster-saveconfig zCLUSTER SAVECONFIGrrrs rccluster_save_configz-ClusterManagementCommands.cluster_save_config(s##$8|#TTrerbnum_keysc(|jd||S)z Returns the number of keys in the specified cluster slot For more information see https://redis.io/commands/cluster-getkeysinslot zCLUSTER GETKEYSINSLOTr)r_rbrs rccluster_get_keys_in_slotz2ClusterManagementCommands.cluster_get_keys_in_slot2s ##$;T8LLreepochc*|jd||S)z Set the configuration epoch in a new node For more information see https://redis.io/commands/cluster-set-config-epoch zCLUSTER SET-CONFIG-EPOCHrr)r_rrqs rccluster_set_config_epochz2ClusterManagementCommands.cluster_set_config_epoch:s### &L$  restatec|jdvr|jd||||S|jdk(r tdtd|)z Bind an hash slot to a specific node :target_node: 'ClusterNode' The node to execute the command on For more information see https://redis.io/commands/cluster-setslot ) IMPORTINGNODE MIGRATINGCLUSTER SETSLOTrSTABLEz4For "stable" state please use cluster_setslot_stablezInvalid slot state: )rrur)r_rrrrs rccluster_setslotz)ClusterManagementCommands.cluster_setslotFse ;;=> >''!7E7( [[]h &VW W3E7;< '+D ###J\#JJrec6d|_|jd|S)z Disables read queries. The command will be sent to the default cluster node if target_nodes is not specified. For more information see https://redis.io/commands/readwrite F READWRITErrrs rc readwritez#ClusterManagementCommands.readwrites"#(##Kl#KKrec &|jdi|S)zl On an OSS cluster, before executing any gears function, you must call this command. # noqa )zREDISGEARS_2.REFRESHCLUSTERr)r_rs rcgears_refresh_clusterz/ClusterManagementCommands.gears_refresh_clusters$t##LVLLrer)TN),rrrrr rrrr'rrrrrrrrr rrrr rrrrrrrrrrrrrr r r rrrrrr rrercrrXsBP(PRHROO N N9 N  )  3=     ) 3=  $HsHyHNCNIN Rz Rd4j R EJ E9 EDHV)V3;C=V V0O.)AOYO<3<9<NR    "  2:>2J    5y5  *  58    KO    /7/G    8<U$^4U UMSMCMIMDH    (0(@    =)=47=BE=NQ= =(JcJiJFJ    *2>*B    P(>*BPiPQT O OI O x/G SW  h~.F RV KXn%= K K Lh~&> L) LMMrerc&eZdZdZdedeefdZy)AsyncClusterManagementCommandsrrrYcVKtjfd|Dd{S7w)rc3hK|])}tjjd|+yw)rN)asyncio create_taskru).0rbr_s rc zBAsyncClusterManagementCommands.cluster_delslots..s2##D$8$89KT$RSs/2N)r%gatherrs` rcrz/AsyncClusterManagementCommands.cluster_delslotss1^^!    s )')N)rrrrrr rrrrercr"r"s Z DJ rer"ceZdZdZ ddeddededeededfd ed ed ee d ed e ffd Z ddee dee dee d efdZxZS)ClusterDataAccessCommands A class for Redis Cluster Data Access Commands The class inherits from Redis's core DataAccessCommand class and do the required adjustments to work with cluster mode stringsalgoLCSvalue1value2specific_argumentrXlenidx minmatchlen withmatchlenrYc | jdd} |dk(r| d} | jd| it | ||||||||fi| S)a Implements complex algorithms that operate on strings. Right now the only algorithm implemented is the LCS algorithm (longest common substring). However new algorithms could be implemented in the future. ``algo`` Right now must be LCS ``value1`` and ``value2`` Can be two strings or two keys ``specific_argument`` Specifying if the arguments to the algorithm will be keys or strings. strings is the default. ``len`` Returns just the len of the match. ``idx`` Returns the match positions in each string. ``minmatchlen`` Restrict the list of matches to the ones of a given minimal length. Can be provided only when ``idx`` set to True. ``withmatchlen`` Returns the matches with the len of the match. Can be provided only when ``idx`` set to True. For more information see https://redis.io/commands/stralgo rqNr-z default-node)popupdatesuperstralgo) r_r.r0r1r2r3r4r5r6rrq __class__s rcr;z!ClusterDataAccessCommands.stralgosk>zz.$7  )l.B)L ~|45w            rematchcount_typec +BK|jd|||d|\}}|Ed{|jDcic]\}}|dk7s ||}}}|r|jDcic]}||j|} }|j dd|ru|jD]5\}}|jd||||| |d|\} }|Ed{| |||<7|jDcic]\}}|dk7s ||}}}|rtyyy7cc}}wcc}w7Jcc}}wwN)r=r>r?r) node_namerq)cursorr=r>r?rqrscanrhrXget_noder8) r_r=r>r?rcursorsdatanamerCnodescurs rc scan_iterz#ClusterDataAccessCommands.scan_itersN" RU%R6R 4;MMOSLD&vQR{4<SS EL\\^TTT4==4=88TET JJ~t ,$+MMO .LD& ) !%###%*4[ ! ! !IC $OO$'IGDM .6=]]_%1T6RS D&L  SU$s]DD D D D  D#D=AD D !D. D<DDD DD)r-FFNFNNN)rrrrrrr rr rr'r;rrrrL __classcell__)r<s@rcr+r+sIR%)"- en- -  - !!3WV_!DE -  - - c]- -  - b%)## !!!}!} !  !rer+c BeZdZdZ ddeedeedeedefdZ y) AsyncClusterDataAccessCommandsr,Nr=r>r?rYc lK|jd|||d|d{\}}|D]}| |jD cic]\}} | dk7s || }}} |r|jDcic]}||j|} }|j dd|r|jD]?\}} |jd| |||| |d|d{\} }|D]}| | |||<A|jD cic]\}} | dk7s || }}} |r~yyy7cc} }wcc}w7Zcc} }wwrArD) r_r=r>r?rrGrHvaluerIrCrJrKs rcrLz(AsyncClusterDataAccessCommands.scan_iter?sw(diiXe5XQWXX  EK 5rcs    !>$"-2*, ^aA5aAHL$#:L$^dM 2dMN  6 6X 2Xv-6-`@ ""re