U ]g@sddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZedd d Gd d d e ZdS) N)cast)versionchanged)Version)ConfigurationError) RedisStorage)DictListOptionalTupleUnionz2.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`. )versionreasoncseZdZUdZdgZddiZeeee ee ffe d<e de ddZ eee ee fd d fd d Zeeeefee ee fd d ddZeeeefee ee fd d ddZeedddZZS)RedisClusterStoragezY Rate limit storage with redis cluster as backend Depends on :pypi:`redis`. z redis+clusterZmax_connectionsiDEFAULT_OPTIONSz4.2.0z2.0.0)redis redisclusterN)urioptionsreturnc  stj|}i}|jr |j|d<|jr0|j|d<|jdd}g}|j|ddD]$}|d\}} ||t | fqXd|_ d|_ |j |f|j |||j st||tt|j|f|dS) a :param uri: url of the form ``redis+cluster://[:password]@host:port,host:port`` :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,:F)urllibparseurlparserrnetlocfindsplitappendintstorageusing_redis_py"_RedisClusterStorage__pick_storagerAssertionErrorZinitialize_storagesuperr__init__) selfrrparsedZ parsed_authsep cluster_hostslochostport __class__@/tmp/pip-unpacked-wheel-kizsipjx/limits/storage/redis_cluster.pyr(+s*      zRedisClusterStorage.__init__)r,rrc sxzB|jdjfdd|D}jjfd|i||_d|_WdStk rr|j|f||jsntdYnXdS)Nrcsg|]}jj|qSr2)clusterZ ClusterNode.0cZredis_pyr2r3 Psz6RedisClusterStorage.__pick_storage.. startup_nodesTztUnable to find an implementation for redis cluster Cluster support requires either redis-py>=4.2 or redis-py-cluster) dependenciesmoduler4 RedisClusterr#r$r7_RedisClusterStorage__use_legacy_cluster_implementation)r)r,rr:r2r8r3Z__pick_storageKs   z"RedisClusterStorage.__pick_storagecKs:|jdj}td|jfddd|Di||_dS)Nra/Using redis-py-cluster is deprecated as the library has been absorbed by redis-py (>=4.2). The support will be eventually removed from the limits library and is no longer tested against since version: 2.6. To get rid of this warning, uninstall redis-py-cluster and ensure redis-py>=4.2.0 is installedr:cSsg|]}|d|ddqS)rr)r.r/r2r5r2r2r3r9oszKRedisClusterStorage.__use_legacy_cluster_implementation..)r;r<warningswarnr=r#)r)r,rZ redis_clusterr2r2r3Z#__use_legacy_cluster_implementationas   z7RedisClusterStorage.__use_legacy_cluster_implementation)rcsd}jrXd}jD]4}j||}|tfdd|D7}q|Sj|}tttfdd|DSdS)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|]}|dqSzutf-8)deletedecoder6k)noder2r3r9sz-RedisClusterStorage.reset..csg|]}j|dqSrB)r#rCrDrE)r)r2r3r9sN) Z prefixed_keyr$r#Z get_primariesZget_redis_connectionkeyssumrr")r)prefixcountZprimaryrHr2)rGr)r3resetss    zRedisClusterStorage.reset)__name__ __module__ __qualname____doc__ZSTORAGE_SCHEMErrstrr floatbool__annotations__rZ DEPENDENCIESr(rr r"r%r>r rL __classcell__r2r2r0r3r s$  !    r)rr?typingrZdeprecated.sphinxrZpackaging.versionrZ limits.errorsrZlimits.storage.redisrZ limits.typingrrr r r rr2r2r2r3s