ZeIi(ddlmZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZmZddlmZmZmZmZmZmZdd lmZGd d eeeZdS) ) annotationsN)Iterable)ceilfloor) ModuleType)ConfigurationError)SlidingWindowCounterSupportStorageTimestampedSlidingWindow)AnyCallableMemcachedClientPPRcast)get_dependencyceZdZdZdgZ dgZ ddZd?dZ d@dZ edAdZ dBd!Z dCd%Z dDd&Z dEdFd-ZdGd.ZdHd/ZdId0ZdJd2Z dKdLd4ZdMd6ZdNd7ZdOd;ZxZS)PMemcachedStoragez[ Rate limit storage with memcached as backend. Depends on :pypi:`pymemcache`. memcached pymemcacheFuristrwrap_exceptionsbooloptions$str | Callable[[], MemcachedClientP]returnNonec tj|}g|_|jdD]F}|s|d\}}|j|t|fG|j r|js|j s |j g|_|j dj |_ t|dd|_t|dd|_t%t&t(t*t,ttfgt.f|d|j|_||_t7|jst9d |jt;j|_d |j_ tC"|| 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 dependencyrpopr"r#rr rlisttupler get_clientr$rrr threadinglocal local_storagestoragesuper__init__) selfrrrparsedlochostr/ __class__s ]/var/www/html/MCyber-Diagnostic/venv/lib/python3.11/site-packages/limits/storage/memcached.pyr<zMemcachedStorage.__init__'s$&&s++ =&&((..s33 + +C 3JD$ J  tSYY/ 0 0 0 0{ +6= + +$k] +L9@7;;y2EFFGG " KK)+C D D  " j$uS#X"78:JJ K KK 9 9   dl++ $V VV '_..%)" o>>>>>-type[Exception] | tuple[type[Exception], ...]c|jjSN)r2 MemcacheErrorr=s rBbase_exceptionsz MemcachedStorage.base_exceptionsZs,,rCr1rr(list[tuple[str, int]]kwargsrc ttt|dkr|j|fi|n |j|i|S)z returns a memcached client. :param module: the memcached module :param hosts: list of memcached hosts )rrlen HashClient PooledClient)r=r1r(rKs rBr6zMemcachedStorage.get_client`s\ u::>>"!%226222(V(%:6::    rCfuncCallable[P, R]argsP.argsP.kwargsrcd|vr9tj|}d|jvs|js|d||i|S)Nnoreply)inspectgetfullargspecrSvarkwr3)r=rQrSrKargspecs rBcall_memcached_funcz$MemcachedStorage.call_memcached_funcssZ   ,T22G--- 9%%%tT$V$$$rCclt|jdr |jjsutt |jdkr|jn|jd}|std|j|j ||jfi|j |j_tt|jjS)zQ lazily creates a memcached client instance using a thread local r:rMrzUnable to import ) hasattrr9r:rrNr(r#r"rr$rrr)r=r2s rBr:zMemcachedStorage.storage~s *I66 4;M;U '(+DJ!(;(;$$J U()ST=Q)S)STTT););DJ***.,**D  &$d&8&@AAArCkeyr-cRt|j|dS)zB :param key: the key to get the counter value for 0)r-r:getr=r_s rBrbzMemcachedStorage.gets$4<##C--...rCkeys Iterable[str]dict[str, Any]c6|j|S)z Return multiple counters at once :param keys: the keys to get the counter values for :meta private: )r:get_many)r=rds rBrhzMemcachedStorage.get_manys|$$T***rCc:|j|dS)z> :param key: the key to clear rate limits for N)r:deletercs rBclearzMemcachedStorage.clears  C     rCrMTexpiryfloatamountset_expiration_keyc||jj||dx} |S||jj||t |ds|j||p|S|rX||jj|||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. FrWN)expirerW)r\r:incraddrset_expiration_keytime)r=r_rlrnrovalues rBrszMemcachedStorage.incrs"-- !3. E   L++  #vtF||U, |((f55??%,, (,,S11,#F|| % -MrCct|j||pt jS)z; :param key: the key to get the expiry for )rmr:rbrvrwrcs rB get_expiryzMemcachedStorage.get_expirys; T\%%d&:&:3&?&?@@ODIKKPPPrCc |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/expiresrcs rBrvz MemcachedStorage._expiration_keysZrCcX ||jjddS#YdSxYw)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` z limiter-checkTF)r\r:rbrHs rBcheckzMemcachedStorage.checks7    $ $T\%5 G G G4 55s $) int | NonectrF)NotImplementedErrorrHs rBresetzMemcachedStorage.resets!!rClimitc||krdStj}||||\}}|||||\}} } } || z|z | z} t| |z|krdS||d|z|d} t d| tj|z z } || z|z | z} t| |kr%||jj||ddSdS)NF)now)rnrorTrq) rwsliding_window_keys_get_sliding_window_inforrsminr\r:decr)r=r_rrlrnr previous_key current_keyprevious_count previous_ttl current_count_weighted_countactualised_previous_ttls rBacquire_sliding_window_entryz-MemcachedStorage.acquire_sliding_window_entrysK E>>5ikk$($<$?6ILS6\Q./69FB |]KGGrC)F)rrrrrrrr)rrD)r1rr(rJrKrrr)rQrRrSrTrKrUrr)rr)r_rrr-)rdrerrf)r_rrr)rMT) r_rrlrmrnr-rorrr-)r_rrrm)r_rrr)rr)rr)rM) r_rrr-rlr-rnr-rr)r_rrlr-rr)r_rrlr-rr) rrrrrlr-rrmrr)__name__ __module__ __qualname____doc__STORAGE_SCHEME DEPENDENCIESr<propertyrIr6r\r:rbrhrkrsrzrvr~rrrrr __classcell__)rAs@rBrrs "]N* >L !&1?1?1?1?1?1?1?f---X-     & % % % %BBBXB$//// ++++!!!!#' %%%%%NQQQQ        """" '''''RUUUU    HHHHHHHHrCr) __future__rrXr7rw urllib.parser%collections.abcrmathrrtypesr limits.errorsrlimits.storage.baser r r limits.typingr r rrrr limits.utilrrr|rCrBrss"""""" $$$$$$,,,,,, '&&&&&UHUHUHUHUHw ;=UUHUHUHUHUHrC