^gy(ddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z mZddlmZddlmZmZe d Gd d ee eZy) ) annotationsN)Iterable)ceilfloor) versionadded)SlidingWindowCounterSupportStorage)TimestampedSlidingWindow)EmcacheClientPItemPz2.1)versionceZdZdZdgZ dgZ d dfd Ze ddZddZ ddZ ddZ dd Z ddd Z d dd Zdd Zdd Zd dZd!dZ d" d#dZ d$dZ d%dZxZS)&MemcachedStoragezW Rate limit storage with memcached as backend. Depends on :pypi:`emcache` zasync+memcachedemcachec tjj|}g|_d|jj j dDD]+\}}|jj|t|f-||_ d|_ t|0|fd|i||jdj|_y)a :param uri: memcached location of the form ``async+memcached://host:port,host:port`` :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:`emcache.Client` :raise ConfigurationError: when :pypi:`emcache` is not available c3`K|]&}|js|jd(yw):N)stripsplit).0locs c/var/www/html/suriana-translation/venv/lib/python3.12/site-packages/limits/aio/storage/memcached.py z,MemcachedStorage.__init__...s% "#))+CIIcN s..,Nwrap_exceptionsr)urllibparseurlparsehostsnetlocrrappendint_options_storagesuper__init__ dependenciesmodule dependency)selfuriroptionsparsedhostport __class__s rr&zMemcachedStorage.__init__s&&s+  &,mm&9&9&;&A&A#&F  1JD$ JJ  tSY/ 0 1     IoII++I6==cZ|jj|jjfSN)r)ClusterNoAvailableNodes CommandErrorr*s rbase_exceptionsz MemcachedStorage.base_exceptions8s( OO 3 3 OO ( (  r1c <K|jsk|jj|jDcgc]!\}}|jj ||#c}}fi|j d{|_|jsJ|jScc}}w7)wr3)r$r) create_clientrMemcachedHostAddressr#)r*hps r get_storagezMemcachedStorage.get_storageAs}}"?$//"?"?HL S155a;S#--#DM}}}}} Ts2B&B B0B1*BcK|jd{j|jdd{}|xrt|jxsdS7G7#w)zB :param key: the key to get the counter value for Nutf-8r)r=getencoder"value)r*keyitems rr@zMemcachedStorage.getJsQ!,,..33CJJw4GHH'DJJ,1,/HsA"A%A"A "A" A"cK|jd{j|Dcgc]}|jdc}d{S78cc}w7 w)zo Return multiple counters at once :param keys: the keys to get the counter values for Nr?)r=get_manyrA)r*keysks rrFzMemcachedStorage.get_manyRsN !,,..88(, -1QXXg  -   . - s,AAAA A A AAcK|jd{j|jdd{y7+7w)z> :param key: the key to clear rate limits for Nr?)r=deleterAr*rCs rclearzMemcachedStorage.clear\s8%%''// 70CDDD'DsAA%AAAAcK|jd{}|jd} |j|||d{xsd}|S787 #|jj$rd}Y|SwxYww)a- decrements the counter for a given rate limit key retursn 0 if the key doesn't exist or if noreply is set to True :param key: the key to decrement :param amount: the number to decrement by :param noreply: set to True to ignore the memcached response Nr?noreplyr)r=rA decrementr)NotFoundCommandError)r*rCamountrOstorage limit_keyrBs rdecrzMemcachedStorage.decrbs((**JJw'  !++Ivw+OOTSTE +P33 E  sDA5AA5AAA A5AA2.A51A22A5c "K|jd{}|jd}|j|j}d} |j||d{xs|} |ry|j |t |d{|rS|j |t|tjzjdt |dd{| S777_7 #|jj$r|jd{7} |j||jt |d{7|rT|j |t|tjzjdt |dd{7|} n#|jj$r|jd{7}|j||d{7xs|} |r{|j |t |d{7|rT|j |t|tjzjdt |dd{7YnwxYw| cYSwxYww)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 :param elastic_expiry: whether to keep extending the rate limit window every hit. :param amount: the number to increment by :param set_expiration_key: if set to False, the expiration time won't be stored but the key will still expire Nr?)exptimeF)rWrO) r=rA_expiration_key incrementtouchrsetstrtimer)rQaddNotStoredStorageCommandError) r*rCexpiryelastic_expiryrRset_expiration_keyrSrT expire_keyrBs rincrzMemcachedStorage.incrtsI$((**JJw' ))#.557 & !++Iv>>H&EmmItF|mDDD%!++"FTYY[0188A $V % & L+ ?D33  ,,...G kk)x-?-?-A4PV<JJ JJ  J J  JcK|jd{}|j|j|jdd{}|xrt |j xst j S7j75w)z; :param key: the key to get the expiry for Nr?)r=r@rXrAfloatrBr])r*rCrSrDs r get_expiryzMemcachedStorage.get_expirysh((**[[!5!5c!:!A!A'!JKK)djj)8TYY[8+Ks!BB6B B4BBc |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/expiresrKs rrXz MemcachedStorage._expiration_keysZr1cK |jd{}|jdd{y77#YyxYww)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` Ns limiter-checkTF)r=r@)r*rSs rcheckzMemcachedStorage.checksD   ,,..G++./ / // / s/A:6:8:A::>AcKtwr3)NotImplementedErrorr6s rresetzMemcachedStorage.resets !!s cK||kDrytj}|j|||\}}|j||||d{\}} } } tj} || z|z | z} t| |z|kDry|j |d|z|dd{} tj}t d| || z z }||z|z | z} t| |kDr|j ||dd{yy77c7 w)NF)rRrbrTrN)r]sliding_window_keys_get_sliding_window_inforrdmaxrU)r*rClimitr`rRnow previous_key current_keyprevious_count previous_ttl current_count_t0weighted_countt1actualised_previous_ttls racquire_sliding_window_entryz-MemcachedStorage.acquire_sliding_window_entrys9 E>iik$($<$?6ILS6\Q./69FB |]KGGBsCCB8C)F)r+r\rboolr,zfloat | str | boolreturnNone)rz-type[Exception] | tuple[type[Exception], ...])rr )rCr\rr")rGz Iterable[str]rzdict[bytes, ItemP])rCr\rr)rF)rCr\rRr"rOrrr")FrT) rCr\r`rfrarrRr"rbrrr")rCr\rrf)rCr\rr\)rr)rz int | None)r) rCr\rtr"r`r"rRr"rr)rCr\r`r"rtuple[int, float, int, float]) rvr\rwr\r`r"rurfrr)__name__ __module__ __qualname____doc__STORAGE_SCHEME DEPENDENCIESr&propertyr7r=r@rFrLrUrdrgrXrkrnrrrr __classcell__)r0s@rrrsl ((NI;L !&> >>& >  >8 6  - E , %#' < << <  < ! < <|9  " ' '' '  '  'R  # & HH.1H;>HEJH &Hr1r) __future__rr] urllib.parsercollections.abcrmathrrdeprecated.sphinxrlimits.aio.storage.baserr limits.storage.baser limits.typingr r rrir1rrsL" $*H8/eFHw ;=UFHFHr1