o KDi(@sddlmZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZmZddlmZmZmZmZmZmZdd lmZGd d d eeeZdS) ) annotationsN)Iterable)ceilfloor) ModuleType)ConfigurationError)SlidingWindowCounterSupportStorageTimestampedSlidingWindow)AnyCallableMemcachedClientPPRcast)get_dependencycseZdZdZdgZ dgZ dMdNfd d ZedOddZdPddZ dQd d!Z edRd"d#Z dSd&d'Z dTd+d,Z dUd-d.Z / 0dVdWd5d6ZdXd7d8ZdYd9d:ZdZd;d<Zd[d>d?Z /d\d]dAdBZd^dDdEZd_dFdGZd`dKdLZZS)aMemcachedStoragez[ Rate limit storage with memcached as backend. Depends on :pypi:`pymemcache`. memcached pymemcacheFuristrwrap_exceptionsbooloptions$str | Callable[[], MemcachedClientP]returnNonec stj|}g|_|jdD]}|sq|d\}}|j|t|fq|j r6|js6|j s6|j g|_|j dj |_ t|dd|_t|dd|_tttttttfgtf|d|j|_||_t|jsvtd |jt|_d |j_ t!j"||d d S) an :param uri: memcached location of the form ``memcached://host:port,host:port``, ``memcached:///var/tmp/path/to/sock`` :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:`pymemcache.client.base.PooledClient` or :class:`pymemcache.client.hash.HashClient` (if there are more than one hosts specified) :raise ConfigurationError: when :pypi:`pymemcache` is not available ,:rlibraryzpymemcache.clientcluster_libraryzpymemcache.client.hash client_getterz5memcached prerequisite not available. please install N)r)#urllibparseurlparsehostsnetlocstripsplitappendintpathport dependenciesmodule dependencyrpoprr rr rlisttupler get_clientr!rrr threadinglocal local_storagestoragesuper__init__)selfrrrparsedlochostr, __class__Q/var/www/Datamplify/venv/lib/python3.10/site-packages/limits/storage/memcached.pyr9's4       zMemcachedStorage.__init__-type[Exception] | tuple[type[Exception], ...]cCs|jjSN)r/ MemcacheErrorr:r@r@rAbase_exceptionsZsz MemcachedStorage.base_exceptionsr.rr%list[tuple[str, int]]kwargsr cKs6ttt|dkr|j|fi|S|j|i|S)z returns a memcached client. :param module: the memcached module :param hosts: list of memcached hosts )rr len HashClient PooledClient)r:r.r%rHr@r@rAr3`s  zMemcachedStorage.get_clientfuncCallable[P, R]argsP.argsP.kwargsrcOs:d|vrt|}d|jvs|js|d||i|S)Nnoreply)inspectgetfullargspecrOvarkwr0)r:rMrOrHargspecr@r@rAcall_memcached_funcss   z$MemcachedStorage.call_memcached_funccCstt|jdr |jjs3tt|jdkr|jn|jd}|s%td|j|j ||jfi|j |j_t t |jjS)zQ lazily creates a memcached client instance using a thread local r7rIrzUnable to import ) hasattrr6r7rrJr%r rrr!rrr )r:r/r@r@rAr7~s zMemcachedStorage.storagekeyr*cCst|j|dS)zB :param key: the key to get the counter value for 0)r*r7getr:rYr@r@rAr[szMemcachedStorage.getkeys Iterable[str]dict[str, Any]cCs |j|S)z Return multiple counters at once :param keys: the keys to get the counter values for :meta private: )r7get_many)r:r]r@r@rAr`s zMemcachedStorage.get_manycCs|j|dS)z> :param key: the key to clear rate limits for N)r7deleter\r@r@rAclearszMemcachedStorage.clearrITexpiryfloatamountset_expiration_keycCs|j|jj||dd}dur|S|j|jj||t|dds(|j||p'|S|r?|j|jj|||tt|dd|S)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 window every hit. :param amount: the number to increment by :param set_expiration_key: set the expiration key with the expiration time if needed. If set to False, the key will still expire, but memcached cannot provide the expiration time. FrRN)expirerR)rWr7incraddrset_expiration_keytime)r:rYrcrerfvaluer@r@rAris&  zMemcachedStorage.incrcCst|j||p tS)z; :param key: the key to get the expiry for )rdr7r[rlrmr\r@r@rA 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/expiresr@r\r@r@rArlsz MemcachedStorage._expiration_keycCs$z ||jjdWdSYdS)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` z limiter-checkTF)rWr7r[rEr@r@rAchecks zMemcachedStorage.check int | NonecCstrC)NotImplementedErrorrEr@r@rAresetszMemcachedStorage.resetlimitcCs||krdSt}||||\}}|j||||d\}} } } || || } t| ||kr2dS|j|d||dd} td| t|} || || } t| |krc|j|jj||dddSdS)NF)now)rerfrTrg) rmsliding_window_keys_get_sliding_window_inforriminrWr7decr)r:rYrtrcreru previous_key current_keyprevious_count previous_ttl current_count_weighted_countactualised_previous_ttlr@r@rAacquire_sliding_window_entrys2  z-MemcachedStorage.acquire_sliding_window_entrytuple[int, float, int, float]cCs*t}||||\}}|||||SrC)rmrwrxr:rYrcrur{r|r@r@rAget_sliding_windowsz#MemcachedStorage.get_sliding_windowcCs2t}||||\}}||||dSrC)rmrwrbrr@r@rAclear_sliding_windows z%MemcachedStorage.clear_sliding_windowr{r|ruc Cs~|||g}t||dt||d}}|dkr!td}n d|||d|}d||d||} |||| fS)NrrI)r`r*r[rd) r:r{r|rcruresultr}rr~ current_ttlr@r@rArx#s  z)MemcachedStorage._get_sliding_window_info)F)rrrrrrrr)rrB)r.rr%rGrHrrr )rMrNrOrPrHrQrr)rr )rYrrr*)r]r^rr_)rYrrr)rIT) rYrrcrdrer*rfrrr*)rYrrrd)rYrrr)rr)rrq)rI) rYrrtr*rcr*rer*rr)rYrrcr*rr)rYrrcr*rr) r{rr|rrcr*rurdrr)__name__ __module__ __qualname____doc__STORAGE_SCHEME DEPENDENCIESr9propertyrFr3rWr7r[r`rbrirorlrprsrrrrx __classcell__r@r@r>rArs83         '   ) r) __future__rrSr4rm urllib.parser"collections.abcrmathrrtypesr limits.errorsrlimits.storage.baserr r limits.typingr r r rrr limits.utilrrr@r@r@rAs