o ;h @spddlmZddlZddlmZddlmZddlmZeddded d ded d dGd ddeZ 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://`` schemacsdeZdZUdZddgZ ddiZded< eded d Ze j d fdfdd Z dddZ Z 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+clusterZmax_connectionsizdict[str, float | str | bool]DEFAULT_OPTIONSz4.2.0z6.0)redisvalkeyFuristr key_prefixwrap_exceptionsbooloptionsfloat | str | boolreturnNonec sDtj|}i}|jr|j|d<|jr|j|d<|jdd}g}|j|ddD]} | d\} } || t | fq,|_ d_ | drLdnd _ ij||} jj j_fd d |D} j d kr|jjjd d | i| _ n jjjd d | i| _ j sJ|ttj||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,:r r csg|] }jjj|qS) dependencyclusterZ ClusterNode).0cselfrO/var/www/html/venv/lib/python3.10/site-packages/limits/storage/redis_cluster.py `sz0RedisClusterStorage.__init__.. startup_nodesr)urllibparseurlparserrnetlocfindsplitappendintrstorage startswithZ target_serverr dependenciesmodulerrZ RedisClusterZ ValkeyClusterZinitialize_storagesuperr__init__)r!r rrrparsedZ parsed_authsepZ cluster_hostslochostportZmerged_optionsr$ __class__r r"r28s>        zRedisClusterStorage.__init__ int | NonecsR|d}d}|jD]}|j||}|tfdd|D7}q |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*rcsg|] }|dqS)zutf-8)deletedecode)rknoderr"r#sz-RedisClusterStorage.reset..)Z prefixed_keyr-Z get_primariesZget_redis_connectionkeyssum)r!prefixcountprimaryrArr?r"resetns   zRedisClusterStorage.reset) r r rr rrrrrr)rr:)__name__ __module__ __qualname____doc__ZSTORAGE_SCHEMEr __annotations__rZ DEPENDENCIESrPREFIXr2rF __classcell__rrr8r"r s  6r) __future__rr%Zdeprecated.sphinxrZpackaging.versionrZlimits.storage.redisrrrrrr"s$