a a*j(@sddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z mZddlmZddlmZmZmZmZmZe d d Gd d d ee eZdS) ) annotationsN)Iterable)ceilfloor) versionadded)SlidingWindowCounterSupportStorage)TimestampedSlidingWindow)EmcacheClientPItemPOptionalTypeUnionz2.1)versioncs6eZdZdZdgZdgZd:ddddd fd d Zed d ddZdd ddZ dddddZ dddddZ dddddZ d;dddddd d!Z dMemcachedStoragezW Rate limit storage with memcached as backend. Depends on :pypi:`emcache` zasync+memcachedemcacheFstrboolzUnion[float, str, bool]None)uriwrap_exceptionsoptionsreturnc stj|}g|_dd|jdDD]\}}|j|t|fq,||_ d|_ t j |fd|i||j dj|_dS)a :param uri: memcached location of the form ``async+memcached://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:`emcache.Client` :raise ConfigurationError: when :pypi:`emcache` is not available css |]}|r|dVqdS):N)stripsplit).0locrX/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/aio/storage/memcached.py .sz,MemcachedStorage.__init__..,Nrr)urllibparseurlparsehostsnetlocrrappendint_options_storagesuper__init__ dependenciesmodule dependency)selfrrrparsedhostport __class__rrr,s zMemcachedStorage.__init__z3Union[Type[Exception], tuple[Type[Exception], ...]])rcCs|jj|jjfSN)r/ZClusterNoAvailableNodes CommandErrorr0rrrbase_exceptions8sz MemcachedStorage.base_exceptionsr csDjs4jjfddjDfijIdH_js>JjS)Ncsg|]\}}j||qSr)r/ZMemcachedHostAddress)rhpr8rr Dz0MemcachedStorage.get_storage..)r*r/Z create_clientr%r)r8rr8r get_storageAs zMemcachedStorage.get_storager()keyrcs2|IdH|dIdH}|r.t|jp0dS)zB :param key: the key to get the counter value for Nutf-8r)r>getencoder(value)r0r?itemrrrrAJs zMemcachedStorage.getz Iterable[str]zdict[bytes, ItemP])keysrcs$|IdHdd|DIdHS)zo Return multiple counters at once :param keys: the keys to get the counter values for NcSsg|]}|dqS)r@)rB)rkrrrr<Yr=z-MemcachedStorage.get_many..)r>get_many)r0rErrrrGRs zMemcachedStorage.get_manycs$|IdH|dIdHdS)z> :param key: the key to clear rate limits for Nr@)r>deleterBr0r?rrrclear\szMemcachedStorage.clear)r?amountnoreplyrcsV|IdH}|d}z|j|||dIdHp2d}Wn|jjyPd}Yn0|S)a- decrements the counter for a given rate limit key retursn 0 if the key doesn't exist or if noreply is set to True :param key: the key to decrement :param amount: the number to decrement by :param noreply: set to True to ignore the memcached response Nr@rMr)r>rBZ decrementr/NotFoundCommandError)r0r?rLrMstorage limit_keyrCrrrdecrbs   zMemcachedStorage.decrTfloat)r?expiryelastic_expiryrLset_expiration_keyrc s|IdH}|d}||}d} zj|||IdHp@|} |r|j|t|dIdH|r|j|t|tdt|ddIdH| WS|j j y|IdH}z\|j ||t|dIdH|r |j|t|tdt|ddIdH|} Wn|j j y|IdH}|||IdHpJ|} |r|j|t|dIdH|r|j|t|tdt|ddIdHYn0| YS0dS)a increments the counter for a given rate limit key :param key: the key to increment :param expiry: amount in seconds for the key to expire in :param elastic_expiry: whether to keep extending the rate limit window every hit. :param amount: the number to increment by :param set_expiration_key: if set to False, the expiration time won't be stored but the key will still expire Nr@)exptimeF)rWrM) r>rB_expiration_key incrementtouchrsetrtimer/rOaddZNotStoredStorageCommandError) r0r?rTrUrLrVrPrQZ expire_keyrCrrrincrtsR    zMemcachedStorage.incrcs@|IdH}|||dIdH}|r8t|jp>tS)z; :param key: the key to get the expiry for Nr@)r>rArXrBrSrCr\)r0r?rPrDrrr get_expiryszMemcachedStorage.get_expirycCs|dS)z Return the expiration key for the given counter key. Memcached doesn't natively return the expiration time or TTL for a given key, so we implement the expiration time on a separate key. z/expiresrrIrrrrXsz MemcachedStorage._expiration_keycs8z$|IdH}|dIdHWdSYdS0dS)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` Ns limiter-checkTF)r>rA)r0rPrrrchecks zMemcachedStorage.checkz Optional[int]cstdSr6)NotImplementedErrorr8rrrresetszMemcachedStorage.reset)r?limitrTrLrcs||kr dSt}||||\}}|||||IdH\}} } } t} || || } t| ||krpdS|j|d||ddIdH} t}td| || }|||| } t| |kr|j||ddIdHdSdSdS)NF)rLrVrTrN)r\sliding_window_keys_get_sliding_window_inforr^maxrR)r0r?rcrTrLnow previous_key current_keyprevious_count previous_ttl current_count_t0Zweighted_countt1Zactualised_previous_ttlrrracquire_sliding_window_entrys2  z-MemcachedStorage.acquire_sliding_window_entryztuple[int, float, int, float])r?rTrcs0t}||||\}}|||||IdHSr6)r\rerf)r0r?rTrhrirjrrrget_sliding_windows z#MemcachedStorage.get_sliding_window)rirjrTrhrc s|||gIdH}||d}||d}|rBt|jpDd}|rTt|jpVd} | dkrjtd} nd|||d|} d||d||} | | || fS)Nr@rrK)rGrArBr(rCrS) r0rirjrTrhresultZraw_previous_countZraw_current_countrmrkrlZ current_ttlrrrrfs z)MemcachedStorage._get_sliding_window_info)F)rKF)FrKT)rK)__name__ __module__ __qualname____doc__ZSTORAGE_SCHEMEZ DEPENDENCIESr,propertyr9r>rArGrJrRr^r_rXr`rbrqrrrf __classcell__rrr4rrs0  >   ) r) __future__rr\ urllib.parser"collections.abcrmathrrZdeprecated.sphinxrZlimits.aio.storage.baserrZlimits.storage.baser Z limits.typingr r r r rrrrrrs