o riT(@sddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZmZmZmZmZmZdd lmZGd d d eeeZdS) ) annotationsN)Iterable)ceilfloor) ModuleType)parse_storage_uri)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 st|}||jr|jg|_n|j|_|jdj|_t|dd|_ t|dd|_ t t t ttttfttBgtf|d|j|_||_t|j sWtd|j t|_d|j_tj||d dS) 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)rpathhosts locations dependenciesmodule dependencyrpoprrrr rlisttupleintr get_clientr rrr threadinglocal local_storagestoragesuper__init__)selfrrrstorage_uri_options __class__W/var/www/html/pca-backend/venv/lib/python3.10/site-packages/limits/storage/memcached.pyr1's,  "    zMemcachedStorage.__init__-type[Exception] | tuple[type[Exception], ...]cCs|jjSN)r& MemcacheErrorr2r6r6r7base_exceptionsSsz MemcachedStorage.base_exceptionsr%rr"list[tuple[str, int]]kwargsrcKs6ttt|dkr|j|fi|S|j|i|S)z returns a memcached client. :param module: the memcached module :param hosts: list of memcached hosts )rrlen HashClient PooledClient)r2r%r"r>r6r6r7r+Ys  zMemcachedStorage.get_clientfuncCallable[P, R]argsP.argsP.kwargsrcOs:d|vrt|}d|jvs|js|d||i|S)Nnoreply)inspectgetfullargspecrEvarkwr')r2rCrEr>argspecr6r6r7call_memcached_funcls   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 r/r?rzUnable to import ) hasattrr.r/rr@r"rrrr rrr)r2r&r6r6r7r/ws zMemcachedStorage.storagekeyr*cCst|j|dS)zB :param key: the key to get the counter value for 0)r*r/getr2rOr6r6r7rQszMemcachedStorage.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: )r/get_many)r2rSr6r6r7rVs zMemcachedStorage.get_manycCs|j|dS)z> :param key: the key to clear rate limits for N)r/deleterRr6r6r7clearszMemcachedStorage.clearr?Texpiryfloatamountset_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. FrHN)expirerH)rMr/incraddrset_expiration_keytime)r2rOrYr[r\valuer6r6r7r_s&  zMemcachedStorage.incrcCst|j||p tS)z; :param key: the key to get the expiry for )rZr/rQrbrcrRr6r6r7 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/expiresr6rRr6r6r7rbsz MemcachedStorage._expiration_keycCs$z ||jjdWdSYdS)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` z limiter-checkTF)rMr/rQr;r6r6r7checks zMemcachedStorage.check int | NonecCstr9)NotImplementedErrorr;r6r6r7resetszMemcachedStorage.resetlimitcCs||krdSt}||||\}}|j||||d\}} } } || || } t| ||kr2dS|j|d||dd} td| t|} || || } t| |krc|j|jj||dddSdS)NF)now)r[r\rTr]) rcsliding_window_keys_get_sliding_window_inforr_minrMr/decr)r2rOrjrYr[rk previous_key current_keyprevious_count previous_ttl current_count_weighted_countactualised_previous_ttlr6r6r7acquire_sliding_window_entrys2  z-MemcachedStorage.acquire_sliding_window_entrytuple[int, float, int, float]cCs*t}||||\}}|||||Sr9)rcrmrnr2rOrYrkrqrrr6r6r7get_sliding_windowsz#MemcachedStorage.get_sliding_windowcCs2t}||||\}}||||dSr9)rcrmrXr{r6r6r7clear_sliding_windows z%MemcachedStorage.clear_sliding_windowrqrrrkc Cs~|||g}t||dt||d}}|dkr!td}n d|||d|}d||d||} |||| fS)Nrr?)rVr*rQrZ) r2rqrrrYrkresultrsrurt current_ttlr6r6r7rns  z)MemcachedStorage._get_sliding_window_info)F)rrrrrrrr)rr8)r%rr"r=r>rrr)rCrDrErFr>rGrr)rr)rOrrr*)rSrTrrU)rOrrr)r?T) rOrrYrZr[r*r\rrr*)rOrrrZ)rOrrr)rr)rrg)r?) rOrrjr*rYr*r[r*rr)rOrrYr*rrz)rOrrYr*rr) rqrrrrrYr*rkrZrrz)__name__ __module__ __qualname____doc__STORAGE_SCHEME DEPENDENCIESr1propertyr<r+rMr/rQrVrXr_rerbrfriryr|r}rn __classcell__r6r6r4r7rs8,         '   ) r) __future__rrIr,rccollections.abcrmathrrtypesrlimits._storage_schemer limits.errorsrlimits.storage.baser r r limits.typingr r rrrr limits.utilrrr6r6r6r7s