JiT(ddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZmZmZmZmZmZdd lmZGd d eeeZy) ) annotationsN)Iterable)ceilfloor) ModuleType)parse_storage_uri)ConfigurationError)SlidingWindowCounterSupportStorageTimestampedSlidingWindow)AnyCallableMemcachedClientPPRcast)get_dependencycJeZdZdZdgZ dgZ d dfd Ze ddZ ddZ ddZ eddZ dd Z dd Z dd Z d dd Zd d Zd!dZd"dZd#dZ d$ d%dZ d&dZd'dZ d(dZxZS))MemcachedStoragez[ Rate limit storage with memcached as backend. Depends on :pypi:`pymemcache`. memcached pymemcachec t|}||jr|jg|_n|j|_|jdj |_t|jdd|_ t|jdd|_ ttttttt fttzgt"f|jd|j$|_||_t+|jst-d|jt/j0|_d|j2_t6|q|| y) 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)wrap_exceptions)rpathhosts locations dependenciesmodule dependencystrpoprrrrrlisttupleintr get_clientroptionsrr threadinglocal local_storagestoragesuper__init__)selfurirr)storage_uri_options __class__s \/var/www/html/asim.com/backend/venv/lib/python3.12/site-packages/limits/storage/memcached.pyr/zMemcachedStorage.__init__'s1$04  # #-223DJ,66DJ++L9@@7;;y2EFG " KK)+C D " j$uS#X"7$s)"CDFVV W KK 9  dll+$G ~V '__.%)" o>c.|jjSN)r" MemcacheErrorr0s r4base_exceptionsz MemcachedStorage.base_exceptionsSs,,,r5c ttt|dkDr|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)r0r!rkwargss r4r(zMemcachedStorage.get_clientYsY u:>"!!%262  )V((%:6:   r5cd|vr@tj|}d|jvs|js|j d||i|S)Nnoreply)inspectgetfullargspecargsvarkwr$)r0funcrEr@argspecs r4call_memcached_funcz$MemcachedStorage.call_memcached_funclsJ  ,,T2G- 9%T$V$$r5ct|jdr|jjstt |j dkDr |j n |jd}|std|j |j||j fi|j|j_tt|jjS)zQ lazily creates a memcached client instance using a thread local r-r<rzUnable to import ) hasattrr,r-rr=rrrr rr)rr)r0r"s r4r-zMemcachedStorage.storagews **I64;M;M;U;U'(+DJJ!(;$$J(+ :param key: the key to clear rate limits for N)r-deleterOs r4clearzMemcachedStorage.clears C r5c|j|jj||dx} |S|j|jj||t |ds |jj||xs|S|rW|j|jj |j ||tjzt |d|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. FrB)expirerB)rIr-incraddrset_expiration_keytime)r0rPexpiryamountset_expiration_keyvalues r4rZzMemcachedStorage.incrs"-- !!3. E   L++   #vtF|U,||((f5??%,, ((,,S1,#F| % -Mr5ct|jj|j|xst jS)z; :param key: the key to get the expiry for )floatr-rNr]r^rOs r4 get_expiryzMemcachedStorage.get_expirys4 T\\%%d&:&:3&?@ODIIKPPr5c |dzS)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/expiresrOs r4r]z MemcachedStorage._expiration_keysZr5c` |j|jjdy#YyxYw)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` z limiter-checkTF)rIr-rNr9s r4checkzMemcachedStorage.checks.    $ $T\\%5%5 G s&)-ctr7)NotImplementedErrorr9s r4resetzMemcachedStorage.resets!!r5c||kDrytj}|j|||\}}|j||||\}} } } || z|z | z} t| |z|kDry|j |d|z|d} t d| tj|z z } || z|z | z} t| |kDr*|j |jj||dyy)NF)now)r`rarTrX) r^sliding_window_keys_get_sliding_window_inforrZminrIr-decr)r0rPlimitr_r`rn previous_key current_keyprevious_count previous_ttl current_count_weighted_countactualised_previous_ttls r4acquire_sliding_window_entryz-MemcachedStorage.acquire_sliding_window_entrys  E>iik$($<$iik$($<$?6ILS6\Q./69FB |]KGGr5)F)r1r#rboolr)z$str | Callable[[], MemcachedClientP]returnNone)rz-type[Exception] | tuple[type[Exception], ...])r!rrzlist[tuple[str, int]]r@r#rr)rGzCallable[P, R]rEzP.argsr@zP.kwargsrr)rr)rPr#rr')rSz Iterable[str]rzdict[str, Any])rPr#rr)r<T) rPr#r_rdr`r'rarrr')rPr#rrd)rPr#rr#)rr)rz int | None)r<) rPr#rtr'r_r'r`r'rr)rPr#r_r'rtuple[int, float, int, float])rPr#r_r'rr) rur#rvr#r_r'rnrdrr)__name__ __module__ __qualname____doc__STORAGE_SCHEME DEPENDENCIESr/propertyr:r(rIr-rNrRrVrZrer]rirlr}rrrq __classcell__)r3s@r4rrs "]N* >L !&*? *?*?8 *?  *?X- 6--    )> JM  & %" %+1 %=E %  %BB$/ +!#' % %% % ! % %NQ  " ' '' '  '  'RUU #U &U HH.1H;>HEJH &Hr5r) __future__rrCr*r^collections.abcrmathrrtypesrlimits._storage_schemer limits.errorsr limits.storage.baser r r limits.typingr rrrrr limits.utilrrrgr5r4rsO" $4, 'NHw ;=UNHr5