jddlmZddlmZddlmZddlmZddlm Z edded d ed d Gd de Z y)) annotations)versionchanged)Version)parse_storage_uri) RedisStoragez3.14.0z_ Dropped support for the :pypi:`redis-py-cluster` library which has been abandoned/deprecated. )versionreasonz2.5.0aa Cluster support was provided by the :pypi:`redis-py-cluster` library which has been absorbed into the official :pypi:`redis` client. By default the :class:`redis.cluster.RedisCluster` client will be used however if the version of the package is lower than ``4.2.0`` the implementation will fallback to trying to use :class:`rediscluster.RedisCluster`. z4.3ztAdded support for using the redis client from :pypi:`valkey` if :paramref:`uri` has the ``valkey+cluster://`` schemaceZdZUdZddgZ ddiZded< eded d Ze jd f dfd Z dd Z xZ S)RedisClusterStoragez Rate limit storage with redis cluster as backend Depends on :pypi:`redis` (or :pypi:`valkey` if :paramref:`uri` starts with ``valkey+cluster://``). z redis+clusterzvalkey+clustermax_connectionsizdict[str, float | str | bool]DEFAULT_OPTIONSz4.2.0z6.0)redisvalkeyFc .t|}i}|jd|jx}r||d<|jd|jx}r||d<|j} ||_d|_|jdrdnd|_|j|jj|_ | D cgc]%} |jjj| '} } i|jd| i||} |jdk(r,|jjjdi| |_n+|jjj di| |_|j sJ|j#|t%t&|R||fi|ycc} w)a :param uri: url of the form ``redis+cluster://[:password]@host:port,host:port`` If the uri scheme is ``valkey+cluster`` the implementation used will be from :pypi:`valkey`. :param key_prefix: the prefix for each key created in redis :param wrap_exceptions: Whether to wrap storage exceptions in :exc:`limits.errors.StorageError` before raising it. :param options: all remaining keyword arguments are passed directly to the constructor of :class:`redis.cluster.RedisCluster` :raise ConfigurationError: when the :pypi:`redis` library is not available or if the redis cluster cannot be reached. usernamepasswordNrr startup_nodes)rgetrr locations key_prefixstorage startswith target_server dependenciesmodule dependencycluster ClusterNoder RedisCluster ValkeyClusterinitialize_storagesuperr__init__)selfurirwrap_exceptionsoptionsstorage_uri_options parsed_authrr cluster_hostscrmerged_options __class__s f/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/limits/storage/redis_cluster.pyr$zRedisClusterStorage.__init__7s*04 {{:/B/K/KL L8 L&.K #{{:/B/K/KL L8 L&.K #+55 $ ),)AXw++D,>,>?FFJWX-Q<00<RDL||| $ lD*3K7KYs5*Fc @|jd}d}|jjD]h}|jj|}|j |}|t |Dcgc]"}|j |jd$c}z }j|Scc}w)a Redis Clusters are sharded and deleting across shards can't be done atomically. Because of this, this reset loops over all keys that are prefixed with :paramref:`RedisClusterStorage.prefix` and calls delete on them one at a time. .. warning:: This operation was not tested with extremely large data sets. On a large production based system, care should be taken with its usage as it could be slow on very large data sets*rzutf-8) prefixed_keyr get_primariesget_redis_connectionkeyssumdeletedecode)r%prefixcountprimarynoder5ks r/resetzRedisClusterStorage.resetis""3'||113G<<44W=D99V$D S$G$Q$++ahhw&78$GH HE4 Hs''B ) r&strrr?r'boolr(zfloat | str | boolreturnNone)rAz int | None)__name__ __module__ __qualname____doc__STORAGE_SCHEMEr __annotations__r DEPENDENCIESrPREFIXr$r> __classcell__)r.s@r/r r s2&'78N. 46O2I!%.L'-- % 0L 0L0L 0L & 0L  0Ldr N) __future__rdeprecated.sphinxrpackaging.versionrlimits._storage_schemerlimits.storage.redisrr rrLr/rRsn",%4-       CY,Y 0YrL