ZeIi ddlmZddlZddlmZddlmZddlmZedded d ed d Gd deZ dS)) annotationsN)versionchanged)Version) 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 j d fdfd Z ddZ 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)redisvalkeyFuristr key_prefixwrap_exceptionsbooloptionsfloat | str | boolreturnNonec ftj|}i}|jr |j|d<|jr |j|d<|jddz}g}|j|ddD]>} | d\} } || t| f?|_ d_ | drdnd _ ij||} jj j_fd |D} j d krjjjd d | i| _ njjjd d | i| _ j sJ|t-t.j||fi|dS) 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. usernamepassword@N,:rr c8g|]}jjj|S) dependencycluster ClusterNode).0cselfs a/var/www/html/MCyber-Diagnostic/venv/lib/python3.11/site-packages/limits/storage/redis_cluster.py z0RedisClusterStorage.__init__..`s)XXXQ<0?FXXXX-XXX   ( (?4?2?+/=DLLA4?2@+/=DL| $$$*lD!!*3KK7KKKKKr) int | Nonec|d}d}|jD]O}|j||}|t fd|Dz }P|S)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*rc`g|]*}|d+S)zutf-8)deletedecode)r$knodes r'r(z-RedisClusterStorage.reset..s1GGGQ$++ahhw&7&788GGGr)) prefixed_keyr3 get_primariesget_redis_connectionkeyssum)r&prefixcountprimaryrQrMs @r'resetzRedisClusterStorage.resetns""3''|1133 I IG<44W==D99V$$D SGGGG$GGGHH HEE r)) rrrrrrrrrr)rrF)__name__ __module__ __qualname____doc__STORAGE_SCHEMEr __annotations__r DEPENDENCIESrPREFIXr<rV __classcell__)rEs@r'r r s2&'78N. 46OI!!'%..L'- % 4L4L4L4L4L4L4Llr)r ) __future__rr+deprecated.sphinxrpackaging.versionrlimits.storage.redisrr r r)r'rds """""" ,,,,,,%%%%%%------         C]]]]],]]  0]]]r)