o ri@stddlmZddlmZddlmZddlmZddlm Z eddded d ded d dGd dde Z dS)) 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://`` 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+clustermax_connectionsizdict[str, float | str | bool]DEFAULT_OPTIONSz4.2.0z6.0)redisvalkeyFuristr key_prefixwrap_exceptionsbooloptionsfloat | str | boolreturnNonec  s t|}i}|d|j}r||d<|d|j}r ||d<|j} |_d_|dr0dnd_j jj _ fdd| D} ij d| i||} jdkrbj j jd i| _n j j jd i| _jsrJ|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. usernamepasswordNr r csg|] }jjj|qS) dependencycluster ClusterNode).0cselfr[/var/www/html/pca-backend/venv/lib/python3.10/site-packages/limits/storage/redis_cluster.py Ysz0RedisClusterStorage.__init__.. startup_nodesr)rgetrr locationsrstorage startswith target_server dependenciesmodulerr r RedisCluster ValkeyClusterinitialize_storagesuperr__init__) r rrrrstorage_uri_options parsed_authrr cluster_hostsr#merged_options __class__rr!r/7s6   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"zsz-RedisClusterStorage.reset..) prefixed_keyr& get_primariesget_redis_connectionkeyssum)r prefixcountprimaryr@rr;r!resetis   zRedisClusterStorage.reset) rrrrrrrrrr)rr6)__name__ __module__ __qualname____doc__STORAGE_SCHEMEr __annotations__r DEPENDENCIESrPREFIXr/rE __classcell__rrr4r!r s  2r N) __future__rdeprecated.sphinxrpackaging.versionrlimits._storage_schemerlimits.storage.redisrr rrrr!s$