9i(ddlmZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZmZddlmZmZmZmZmZmZdd lmZGd d eeeZy) ) annotationsN)Iterable)ceilfloor) ModuleType)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 tjj|}g|_|jj j dD]?}|s|j d\}}|jj|t|fA|jr*|js|js|jg|_|jdj|_ t|jdd|_t|jdd|_t%t&t(t*t,ttfgt.f|jd|j0|_||_t7|j st9d |j t;j<|_d |j>_ tB||| 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)#urllibparseurlparsehostsnetlocstripsplitappendintpathport dependenciesmodule dependencystrpoprrrr rlisttupler get_clientroptionsrr threadinglocal local_storagestoragesuper__init__) selfurirr1parsedlochostr( __class__s K/var/www/html/venv/lib/python3.12/site-packages/limits/storage/memcached.pyr7zMemcachedStorage.__init__'s$&&s+ ==&&(..s3 +C3JD$ JJ  tSY/ 0 +{{6==$kk] ++L9@@7;;y2EFG " KK)+C D " j$uS#X"78:JJ K KK 9  dll+$G ~V '__.%)" o>c.|jjSN)r+ MemcacheErrorr8s r>base_exceptionsz MemcachedStorage.base_exceptionsZs,,,r?c 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)r8r*r!kwargss r>r0zMemcachedStorage.get_client`sY u:>"!!%262  )V((%:6:   r?cd|vr@tj|}d|jvs|js|j d||i|S)Nnoreply)inspectgetfullargspecargsvarkwr-)r8funcrOrJargspecs r>call_memcached_funcz$MemcachedStorage.call_memcached_funcssJ  ,,T2G- 9%T$V$$r?ct|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 r5rFrzUnable to import ) hasattrr4r5rrGr!rrrrr1rr)r8r+s r>r5zMemcachedStorage.storage~s **I64;M;M;U;U'(+DJJ!(;$$J(+rXzMemcachedStorage.gets 4<<##C-..r?c8|jj|S)z Return multiple counters at once :param keys: the keys to get the counter values for :meta private: )r5get_many)r8keyss r>r\zMemcachedStorage.get_manys||$$T**r?c:|jj|y)z> :param key: the key to clear rate limits for N)r5deleterYs r>clearzMemcachedStorage.clears C r?c|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. FrL)expirerL)rSr5incraddrset_expiration_keytime)r8rZexpiryamountset_expiration_keyvalues r>rdzMemcachedStorage.incrs"-- !!3. E   L++   #vtF|U,||((f5??%,, ((,,S1,#F| % -Mr?ct|jj|j|xst jS)z; :param key: the key to get the expiry for )floatr5rXrgrhrYs r> get_expiryzMemcachedStorage.get_expirys4 T\\%%d&:&:3&?@ODIIKPPr?c |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/expiresrYs r>rgz MemcachedStorage._expiration_keysZr?c` |j|jjdy#YyxYw)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` z limiter-checkTF)rSr5rXrCs r>checkzMemcachedStorage.checks.    $ $T\\%5%5 G s&)-ctrA)NotImplementedErrorrCs r>resetzMemcachedStorage.resets!!r?c||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)rjrkrTrb) rhsliding_window_keys_get_sliding_window_inforrdminrSr5decr)r8rZlimitrirjrx previous_key current_keyprevious_count previous_ttl current_count_weighted_countactualised_previous_ttls r>acquire_sliding_window_entryz-MemcachedStorage.acquire_sliding_window_entrys  E>iik$($<$get_sliding_windowz#MemcachedStorage.get_sliding_windows@iik$($<$clear_sliding_windowz%MemcachedStorage.clear_sliding_windows>iik$($<$r{z)MemcachedStorage._get_sliding_window_info#s k:;  <+ ,  ;* +& Q  8L3<6"9Q!>?6ILS6\Q./69FB |]KGGr?)F)r9r,rboolr1z$str | Callable[[], MemcachedClientP]returnNone)rz-type[Exception] | tuple[type[Exception], ...])r*rr!zlist[tuple[str, int]]rJr,rr)rQzCallable[P, R]rOzP.argsrJzP.kwargsrr)rr)rZr,rr&)r]z Iterable[str]rzdict[str, Any])rZr,rr)rFT) rZr,rirnrjr&rkrrr&)rZr,rrn)rZr,rr,)rr)rz int | None)rF) rZr,r~r&rir&rjr&rr)rZr,rir&rtuple[int, float, int, float])rZr,rir&rr) rr,rr,rir&rxrnrr)__name__ __module__ __qualname____doc__STORAGE_SCHEME DEPENDENCIESr7propertyrDr0rSr5rXr\r`rdrorgrsrvrrrr{ __classcell__)r=s@r>rrs "]N* >L !&1? 1?1?8 1?  1?f- 6--    )> JM  & %" %+1 %=E %  %BB$/ +!#' % %% % ! % %NQ  " ' '' '  '  'RUU #U &U HH.1H;>HEJH &Hr?r) __future__rrMr2rh urllib.parsercollections.abcrmathrrtypesr limits.errorsrlimits.storage.baser r r limits.typingr r rrrr limits.utilrrrqr?r>rsO" $, 'UHw ;=UUHr?