
    Ji                        d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	  edd       ed	d
       edd       G d de	                           Z
y)    )annotations)versionchanged)Version)parse_storage_uri)RedisStoragez3.14.0z_
Dropped support for the :pypi:`redis-py-cluster` library
which has been abandoned/deprecated.
)versionreasonz2.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`.
z4.3ztAdded support for using the redis client from :pypi:`valkey` if :paramref:`uri` has the ``valkey+cluster://`` schemac                       e Zd ZU dZddgZ	 ddiZded<   	  ed       ed	      d
Ze	j                  df	 	 	 	 	 	 	 	 	 d fdZddZ x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+clusterzvalkey+clustermax_connectionsi  zdict[str, float | str | bool]DEFAULT_OPTIONSz4.2.0z6.0)redisvalkeyFc                .   t        |      }i }|j                  d|j                        x}r||d<   |j                  d|j                        x}r||d<   |j                  }	|| _        d| _        |j                  d      rdnd| _        | j                  | j                     j                  | _        |	D 
cg c]%  }
 | j                  j                  j                  |
 ' }}
i | j                  d|i||}| j                  dk(  r, | j                  j                  j                  di || _        n+ | j                  j                  j                   di || _        | j                  sJ | j#                  |       t%        t&        | R  ||fi | yc c}
w )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.
        usernamepasswordNr   r   startup_nodes )r   getr   r   	locations
key_prefixstorage
startswithtarget_serverdependenciesmodule
dependencyclusterClusterNoder   RedisClusterValkeyClusterinitialize_storagesuperr   __init__)selfurir   wrap_exceptionsoptionsstorage_uri_optionsparsed_authr   r   cluster_hostscr   merged_options	__class__s                `/var/www/html/asim.com/backend/venv/lib/python3.12/site-packages/limits/storage/redis_cluster.pyr$   zRedisClusterStorage.__init__7   s   * 04{{:/B/K/KLL8L&.K
#{{:/B/K/KLL8L&.K
#+55$),)AXw++D,>,>?FFJWX-Q<00<<a@-X
""
.
 
 	
 (?4??22??Q.QDL@4??22@@R>RDL|||$lD*3K7K Ys   5*Fc                @   | j                  d      }d}| j                  j                         D ]h  }| j                  j                  |      }|j	                  |      }|t        |D cg c]"  }|j                  |j                  d            $ c}      z  }j |S c c}w )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*r   zutf-8)prefixed_keyr   get_primariesget_redis_connectionkeyssumdeletedecode)r%   prefixcountprimarynoder5   ks          r/   resetzRedisClusterStorage.reseti   s     ""3'||113G<<44W=D99V$DS$G$Q$++ahhw&78$GHHE 4  Hs   ''B)
r&   strr   r?   r'   boolr(   zfloat | str | boolreturnNone)rA   z
int | None)__name__
__module____qualname____doc__STORAGE_SCHEMEr   __annotations__r   DEPENDENCIESr   PREFIXr$   r>   __classcell__)r.   s   @r/   r   r   
   s    2 &'78N. 	46O2  I !%.L '-- %	0L0L 0L 	0L
 &0L 
0Ld    r   N)
__future__r   deprecated.sphinxr   packaging.versionr   limits._storage_schemer   limits.storage.redisr   r   r   rL   r/   <module>rR      sn    " , % 4 -  	 	CY, Y	0YrL   