a a*j@stddlmZddlZddlmZddlmZddlmZddl m Z m Z eddd ed d d Gd d d eZ dS)) annotationsN)versionchanged)Version) RedisStorage)OptionalUnionz3.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`. cs`eZdZUdZdgZddiZded<dediZdd d d d dfdd Z ddddZ Z S)RedisClusterStoragezY Rate limit storage with redis cluster as backend Depends on :pypi:`redis`. z redis+clusterZmax_connectionsiz"dict[str, Union[float, str, bool]]DEFAULT_OPTIONSredisz4.2.0FstrboolzUnion[float, str, bool]None)uriwrap_exceptionsoptionsreturnc  stj|}i}|jr |j|d<|jr0|j|d<|jdd}g}|j|ddD]$}|d\} } || t | fqXd_ ij ||} j dj _fd d |D} jjjfd | i| _ j sJ|ttj||fi|dS) a :param uri: url of the form ``redis+cluster://[:password]@host:port,host:port`` :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 csg|]}jjj|qS) dependencyclusterZ ClusterNode).0cselfrX/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/storage/redis_cluster.py Qz0RedisClusterStorage.__init__.. startup_nodes)urllibparseurlparserrnetlocfindsplitappendintstorager dependenciesmodulerrZ RedisClusterZinitialize_storagesuperr__init__) r rrrparsedZ parsed_authsepZ cluster_hostslochostportZmerged_optionsr$ __class__rr!r10s.      zRedisClusterStorage.__init__z Optional[int])rcsR|d}d}|jD]4}|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 ``self.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"jr#z-RedisClusterStorage.reset..)Z prefixed_keyr-Z get_primariesZget_redis_connectionkeyssum)r prefixcountprimaryr?rr=r!resetYs   zRedisClusterStorage.reset)F) __name__ __module__ __qualname____doc__ZSTORAGE_SCHEMEr __annotations__rZ DEPENDENCIESr1rD __classcell__rrr7r!r s  )r ) __future__rr%Zdeprecated.sphinxrZpackaging.versionrZlimits.storage.redisrZ limits.typingrrr rrrr!s