^g& ddlmZddlZddlZddlmZmZddlmZddl m Z ddl Z ddl mZmZmZddlmZGdd Ze d Gd d eeeeZy)) annotationsN)Counter defaultdict)floor) versionadded)MovingWindowSupportSlidingWindowCounterSupportStorage)TimestampedSlidingWindowceZdZddZy)Entryc^tj|_|j|z|_yN)timeatimeexpiry)selfrs `/var/www/html/suriana-translation/venv/lib/python3.12/site-packages/limits/aio/storage/memory.py__init__zEntry.__init__sYY[ jj6) N)rintreturnNone)__name__ __module__ __qualname__rrrr r s*rr z2.1)versioncPeZdZdZdgZ d dfd ZddZddZddZddZ e ddZ d dd Z ddd Z d d Zd!d Z d d"d Zd#dZd$dZ d%dZ d d"dZ d&dZ d'dZd(dZd)dZxZS)* MemoryStoragez rate limit storage using :class:`collections.Counter` as an in memory storage for fixed and elastic window strategies, and a simple list to implement moving window strategy. z async+memoryc t|_ttj|_i|_i|_d|_t|(|fd|i|y)Nwrap_exceptions) rstoragerasyncioLocklocks expirationseventstimersuperr)rurir"_ __class__s rrzMemoryStorage.__init__)sK4;9 5@5N -/.0 04  CoCCrcF|jj}|d=|d=|S)Nr)r&)__dict__copyrstates r __getstate__zMemoryStorage.__getstate__3s' ""$ 'N 'N rc|jj|d|_ttj |_t j|jyr) r/updater)rr$r%r& ensure_future_MemoryStorage__schedule_expiryr1s r __setstate__zMemoryStorage.__setstate__9sB U#  . d4467rcK|jjD]}|j|4d{t|j|D]T}|jt j ks%||j|vs7|j|j |V|jj|ds8|jj|d|jj|ddddd{t|jjD]{}|j|t j ks(|jj|d|jj|d|jj|d}y7|7#1d{7swYxYwwr) r(keysr&listrrremovegetpopr'r#)rkeyevents r__expire_eventszMemoryStorage.__expire_events?si;;##% .Czz# . .!$++c"237E||tyy{2u C@P7P C(//67{{sD1KKOOC.JJNN3-  . . . .((--/0 *C$ 3   d+  $$S$/ sD)  * . . . . .sY1G F/G =F45F4A4F4; G F2AG AG 2G 4G :F=;G G cK|jr|jjr)tj|j |_yywr)r)doner$ create_task_MemoryStorage__expire_eventsrs r__schedule_expiryzMemoryStorage.__schedule_expiryOs:zzTZZ__. ,,T-A-A-CDDJ/sAActSr) ValueErrorrFs rbase_exceptionszMemoryStorage.base_exceptionsSs rcK|j|d{|jd{|j|4d{|j|xx|z cc<|s|j||k(r$t j|z|j |<dddd{|jj||S7777&#1d{7swY6xYww)aD 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 N)r=r7r&r#rr')rr?relastic_expiryamounts rincrzMemoryStorage.incrYshhsm$$&&&::c? = = LL  ' c!2f!<(, f(<  %  = = ||V,, & = = = = =shC$CC$C C$C C$ AC C$&C '!C$ C$ C$ C$C!C C!C$cK|j|d{|jd{|j|4d{t|j||z d|j|<dddd{|jj||S77s7]7&#1d{7swY6xYww)z decrements the counter for a given rate limit key. 0 is the minimum allowed value. :param amount: the number to increment by Nr)r=r7r&maxr#)rr?rMs rdecrzMemoryStorage.decros hhsm$$&&&::c? C C #DLL$5$> BDLL  C C||V,, & C C C C CsgB>B!B>B#B>B%B> *B)5 B>B'!B>#B>%B>'B>)B;/B2 0B;7B>cLK|jj|dtjkrT|jj |d|jj |d|j j |d|jj|dSw)zB :param key: the key to get the counter value for rN)r'r=rr#r>r&rr?s rr=zMemoryStorage.get|s|     Q '499; 6 LL  S$ '    d + JJNN3 %||Q''sB"B$cK|jj|d|jj|d|jj|d|jj|dyw)z> :param key: the key to clear rate limits for N)r#r>r'r(r&rSs rclearzMemoryStorage.clearsV d# S$' T" sD!sA2A4c<K||kDry|jd{|j|4d{|jj|gt j} |j|||z }|r$|j ||z k\r dddd{yt|Dcgc] }t|c}|j|dd dddd{y77#t $rd}Y{wxYw7^cc}w7!#1d{7swYyxYww)z :param key: rate limit key to acquire an entry in :param limit: amount of entries allowed :param expiry: expiry of the entry :param amount: the number of entries to acquire FNrT) r7r&r( setdefaultr IndexErrorrranger )rr?limitrrM timestampentryr,s r acquire_entryzMemoryStorage.acquire_entrys E>$$&&&::c?   KK " "3 + I &*kk#&6uv~&F F(::   @EV}'M!f 'M C !$    '     (N    sDC)DC+D1D*C-?D DC> D% D2DD D#D$D+D- C;8D:C;;D>DDDD D DDcfK|jj|tjSw)z; :param key: the key to get the expiry for )r'r=rrSs r get_expiryzMemoryStorage.get_expirys' ##C55s/1cKtj}|jj|rFt|jj|gDcgc]}|j||z k\s|c}SdScc}ww)z returns the number of entries already acquired :param key: rate limit key to acquire an entry in :param expiry: expiry of the entry r)rr(r=lenr)rr?rr[ks rget_num_acquiredzMemoryStorage.get_num_acquiredskIIK {{s# DKKOOC4Vq9vCU8UV W  VsAA?A:-A:1A?cKtj}|j||d{}|jj|gdddD]$}|j||z k\s|j|fcS||fS7Qw)z returns the starting point and the number of entries in the moving window :param key: rate limit key :param expiry: expiry of entry :return: (start of window, number of acquired entries) N)rrcr(r=r)rr?rZrr[acquireditems rget_moving_windowzMemoryStorage.get_moving_windowsIIK ..sF;;KKOOC,TrT2 ,DzzY//zz8++ ,("" ;B)BcK||kDrytj}|j|||\}}|j||||d{\}} } } || z|z | z} t| |z|kDry|j |d|z|d{} || z|z | z} t| |kDr|j ||d{yy7z7;7 w)NF)rMT)rsliding_window_keys_get_sliding_window_inforrNrQ) rr?rZrrMnow previous_key current_keyprevious_count previous_ttl current_countr,weighted_counts racquire_sliding_window_entryz*MemoryStorage.acquire_sliding_window_entrys E>iik$($<$C ?C C  C cKtj}|j|||\}}|j||||d{S7wr)rrkrl)rr?rrmrnros rget_sliding_windowz MemoryStorage.get_sliding_windowsRiik$($<$?6ILS6\Q./69FB |]KGG63sA2A.A2A0>A20A2c Kyw)z- check if storage is healthy TrrFs rcheckzMemoryStorage.checks scBKtt|jt|j}|jj |j j |jj |j j |Swr)rPrar#r(rUr'r&)r num_itemss rresetzMemoryStorage.resetslDLL)3t{{+;<      sBB)NF)r+z str | Noner"boolr,strrr)rdict[str, limits.typing.Any])r2rrr)rr)rz-type[Exception] | tuple[type[Exception], ...])Frx) r?rrryrLrrMrrr)rx)r?rrMrrr)r?rrr)r?rrr) r?rrZrrrrMrrr)r?rrry)r?rrrrr)r?rrZrrrrztuple[float, int])r?rrrrtuple[int, float, int, float]) rnrrorrrrmryrr)rr)rz int | None)rrr__doc__STORAGE_SCHEMErr3r8rEr7propertyrJrNrQr=rUr]r_rcrhrtrvrlr|r __classcell__)r-s@rr r s %%N ?DDD7;DJMD D 8 * E 6 TU-- %-7;-MP- -, - ("@A",/9< 86  ##"#,/# #2         D  # & HHH H  H ' H rr ) __future__rr$r collectionsrrmathrdeprecated.sphinxr limits.typinglimitslimits.aio.storage.baserr r limits.storage.baser r r rrrrs`" ,* 9** eF "=?WFFr