o ri@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZer3 ed d d Gd d d e ZdS)) annotations) TYPE_CHECKING)versionchanged)Version)parse_storage_uri)ConfigurationError) RedisStorage) RedisClientz4.3zuAdded support for using the redis client from :pypi:`valkey` if :paramref:`uri` has the ``valkey+sentinel://`` schema)versionreasoncsreZdZdZddgZ ededededdZdddejd fd!fdd Z e d"ddZ d#d$dd Z Z S)%RedisSentinelStoragez Rate limit storage with redis sentinel as backend Depends on :pypi:`redis` package (or :pypi:`valkey` if :paramref:`uri` starts with ``valkey+sentinel://``) zredis+sentinelzvalkey+sentinelz3.0z6.0)rediszredis.sentinelvalkeyzvalkey.sentinelNTFuristr service_name str | None use_replicasboolsentinel_kwargs$dict[str, float | str | bool] | None key_prefixwrap_exceptionsoptionsfloat | str | boolreturnNonec s2tt|j|fd|i|t|}g} |r|ni} i} |d|j} r+| | d<|d|j} r8| | d<| |j ||_ |j rK|j ddn||_ |j durWtd|dr^dnd |_|j|jd j}|j| fd i| | ii| ||_|j|j |_|j|j |_||_||dS) a :param uri: url of the form ``redis+sentinel://host:port,host:port/service_name`` If the uri scheme is ``valkey+sentinel`` the implementation used will be from :pypi:`valkey`. :param service_name: sentinel service name (if not provided in :attr:`uri`) :param use_replicas: Whether to use replicas for read only operations :param sentinel_kwargs: kwargs to pass as :attr:`sentinel_kwargs` to :class:`redis.sentinel.Sentinel` :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.sentinel.Sentinel` :raise ConfigurationError: when the redis library is not available or if the redis master host cannot be pinged. rusernamepassword/Nz'service_name' not providedrr z .sentinelr)superr__init__rcopygetrrextend locationsrpathreplacerr startswith target_server dependenciesmoduleSentinelsentinel master_forstorage slave_for storage_slaverinitialize_storage)selfrrrrrrrstorage_uri_optionssentinel_configurationsentinel_options parsed_authrr sentinel_dep __class__\/var/www/html/pca-backend/venv/lib/python3.10/site-packages/limits/storage/redis_sentinel.pyr"*sH     zRedisSentinelStorage.__init__-type[Exception] | tuple[type[Exception], ...]cCs&|jdkr |jdjjS|jdjjS)Nr r)r*r+r, RedisError ValkeyError)r4r<r<r=base_exceptionsns   z$RedisSentinelStorage.base_exceptionsreadonlyr cCs|r|jr|jS|jS)N)rr2r0)r4rBr<r<r=get_connectionxsz#RedisSentinelStorage.get_connection)rrrrrrrrrrrrrrrr)rr>)F)rBrrr )__name__ __module__ __qualname____doc__STORAGE_SCHEMEr DEPENDENCIESrPREFIXr"propertyrArC __classcell__r<r<r:r=r s$ D  r N) __future__rtypingrdeprecated.sphinxrpackaging.versionrlimits._storage_schemer limits.errorsrlimits.storage.redisr limits.typingr r r<r<r<r=s