o ;h(@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`. Z 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 dependencyrpoprrrr rlisttupler get_clientr rrr threadinglocal local_storagestoragesuper__init__)selfrrrparsedlochostr+ __class__K/var/www/html/venv/lib/python3.10/site-packages/limits/storage/memcached.pyr8's4       zMemcachedStorage.__init__-type[Exception] | tuple[type[Exception], ...]cCs|jjSN)r.Z MemcacheErrorr9r?r?r@base_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 lenZ HashClientZ PooledClient)r9r-r$rFr?r?r@r2`s  zMemcachedStorage.get_clientfuncCallable[P, R]argsP.argsP.kwargsrcOs:d|vrt|}d|jvs|js|d||i|S)Nnoreply)inspectgetfullargspecrKvarkwr/)r9rIrKrFZargspecr?r?r@call_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 r6rGrzUnable to import ) hasattrr5r6rrHr$rrrr rrr )r9r.r?r?r@r6~s zMemcachedStorage.storagekeyr)cCst|j|dS)zB :param key: the key to get the counter value for 0)r)r6getr9rTr?r?r@rVszMemcachedStorage.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: )r6get_many)r9rXr?r?r@r[s zMemcachedStorage.get_manycCs|j|dS)z> :param key: the key to clear rate limits for N)r6deleterWr?r?r@clearszMemcachedStorage.clearrGTexpiryfloatamountset_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. FrNN)ZexpirerN)rRr6incraddrset_expiration_keytime)r9rTr^r`ravaluer?r?r@rcs&  zMemcachedStorage.incrcCst|j||p tS)z; :param key: the key to get the expiry for )r_r6rVrfrgrWr?r?r@ 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?rWr?r?r@rfsz MemcachedStorage._expiration_keycCs$z ||jjdWdSYdS)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` z limiter-checkTF)rRr6rVrCr?r?r@checks zMemcachedStorage.check int | NonecCstrB)NotImplementedErrorrCr?r?r@resetszMemcachedStorage.resetlimitcCs||krdSt}||||\}}|j||||d\}} } } || || } t| ||kr2dS|j|d||dd} td| t|} || || } t| |krc|j|jj||dddSdS)NF)now)r`rarTrb) rgsliding_window_keys_get_sliding_window_inforrcminrRr6Zdecr)r9rTrnr^r`ro previous_key current_keyprevious_count previous_ttl current_count_Zweighted_countZactualised_previous_ttlr?r?r@acquire_sliding_window_entrys2  z-MemcachedStorage.acquire_sliding_window_entrytuple[int, float, int, float]cCs*t}||||\}}|||||SrB)rgrqrrr9rTr^rortrur?r?r@get_sliding_windowsz#MemcachedStorage.get_sliding_windowcCs2t}||||\}}||||dSrB)rgrqr]r|r?r?r@clear_sliding_windows z%MemcachedStorage.clear_sliding_windowrtruroc Cs~|||g}t||dt||d}}|dkr!td}n d|||d|}d||d||} |||| fS)NrrG)r[r)rVr_) r9rtrur^roresultrvrxrwZ current_ttlr?r?r@rr#s  z)MemcachedStorage._get_sliding_window_info)F)rrrrrrrr)rrA)r-rr$rErFrrr )rIrJrKrLrFrMrr)rr )rTrrr))rXrYrrZ)rTrrr)rGT) rTrr^r_r`r)rarrr))rTrrr_)rTrrr)rr)rrk)rG) rTrrnr)r^r)r`r)rr)rTrr^r)rr{)rTrr^r)rr) rtrrurr^r)ror_rr{)__name__ __module__ __qualname____doc__ZSTORAGE_SCHEMEZ DEPENDENCIESr8propertyrDr2rRr6rVr[r]rcrirfrjrmrzr}r~rr __classcell__r?r?r=r@rs83         '   ) r) __future__rrOr3rg urllib.parser!collections.abcrmathrrtypesrZ limits.errorsrZlimits.storage.baserr r Z limits.typingr r r rrrZ limits.utilrrr?r?r?r@s