9i)dZddlmZddlZddlmZmZddlmZm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZd d lmZmZd dlmZGddeZGddeZGddeZe dGddeZeeedZy)z' Asynchronous rate limiting strategies ) annotationsN)ABCabstractmethod)floorinf) versionadded) RateLimitItem) StorageTypes)cast) WindowStats)MovingWindowSupportStorage)SlidingWindowCounterSupportcjeZdZd dZeddd dZeddd dZe d dZd dZy) RateLimiterc6t|tsJ||_yN) isinstancerstorage)selfrs H/var/www/html/venv/lib/python3.12/site-packages/limits/aio/strategies.py__init__zRateLimiter.__init__s'7+++ ' rcostcKtw)B Consume the rate limit :param item: the rate limit item :param identifiers: variable list of strings to uniquely identify the limit :param cost: The cost of this hit, default 1 :return: True if ``cost`` could be deducted from the rate limit without exceeding it NotImplementedErrorritemr identifierss rhitzRateLimiter.hit"! cKtw): Check if the rate limit can be consumed :param item: the rate limit item :param identifiers: variable list of strings to uniquely identify the limit :param cost: The expected cost to be consumed, default 1 :return: True if the rate limit is not depleted r r"s rtestzRateLimiter.test(r&r'cKtw)z Query the reset time and remaining amount for the limit :param item: the rate limit item :param identifiers: variable list of strings to uniquely identify the limit :return: (reset time, remaining)) r rr#r$s rget_window_statszRateLimiter.get_window_stats6r&r'cnK|jj|j|d{S7wr)rclearkey_forr,s rr/zRateLimiter.clearDs,\\''  k(BCCCCs ,535Nrr r#r r$strrintreturnboolr#r r$r3r5r r#r r$r3r5None) __name__ __module__ __qualname__rrr%r*r-r/rrrrsg(LM " "MN " " "! "14 "  " "DrrcTeZdZdZdfd Zddd dZddd dZ d dZxZS) MovingWindowRateLimiterz4 Reference: :ref:`strategies:moving window` ct|ds$t|dstd|jt||y)N acquire_entryget_moving_windowz@MovingWindowRateLimiting is not implemented for storage of type hasattrr! __class__superrrrrEs rrz MovingWindowRateLimiter.__init__MsI G_ -BU1V%",,-/  !rrrcKtt|jj|j||j |j |d{S7wr)amountN)r rrrAr0rJ get_expiryr"s rr%zMovingWindowRateLimiter.hitWsV-t||<JJ DLL+ & T__5FtK    sAAAAcKtt|jj|j||j |j d{}|d}||j |z kS7wr)Nr)r rrrBr0rJrK)rr#rr$resrJs rr*zMovingWindowRateLimiter.testgsl,dll;MM DLL+ & KK OO    Qt+++ sAA4A2A4cKtt|jj|j||j |j d{\}}||j z}t||j |z S73w)z returns the number of requests remaining within this limit. :param item: the rate limit item :param identifiers: variable list of strings to uniquely identify the limit :return: (reset time, remaining) N)r rrrBr0rJrKr )rr#r$ window_start window_itemsresets rr-z(MovingWindowRateLimiter.get_window_stats{sz,0 ,  LDLL+6 T__EV W&X" lt005$++ "<== &XsAB B 4B )rr r5r9r2r7) r:r;r<__doc__rr%r*r- __classcell__rEs@rr?r?HsB"MN NO,(>!>14> >rr?c@eZdZdZddddZddddZ d dZy) FixedWindowRateLimiterz3 Reference: :ref:`strategies:fixed window` rrcK|jj|j||j|d{|jkS7wrI)rincrr0rKrJr"s rr%zFixedWindowRateLimiter.hitsW,,## k*!$  {{    s=AAAcK|jj|j|d{|j|z dzkS7wrM)rgetr0rJr"s rr*zFixedWindowRateLimiter.testsD,,""<4<<#=> >tASVWAW W >s,AAAc Ktd|j|jj|j|d{z }|jj |j|d{}t ||S7F7w)z Query the reset time and remaining amount for the limit :param item: the rate limit item :param identifiers: variable list of strings to uniquely identify the limit :return: reset time, remaining rN)maxrJrr[r0rKr )rr#r$ remainingrRs rr-z'FixedWindowRateLimiter.get_window_statssy KK 0 0{1K LL L ll--ldllK.HII5),, MIs"=B B 5B 5B6B B Nr2r7)r:r;r<rSr%r*r-r=rrrWrWs<MN *NO -!-14- -rrWz4.1)versioncxeZdZdZd fd Z d dZddd dZddd dZ d dZdd Z xZ S)SlidingWindowCounterRateLimiterz= Reference: :ref:`strategies:sliding window counter` ct|dr t|dstd|jt||y)Nget_sliding_windowacquire_sliding_window_entryzHSlidingWindowCounterRateLimiting is not implemented for storage of type rCrGs rrz(SlidingWindowCounterRateLimiter.__init__sOw 45W 3> &",,-/  !rc4||z|jz |zS)zu Return the approximated by weighting the previous window count and adding the current window count. )rK)rr#previous_countprevious_expires_in current_counts r_weighted_countz/SlidingWindowCounterRateLimiter._weighted_counts! 33doo6GG-WWrrrcKtt|jj|j||j |j |d{S7w)aS Consume the rate limit :param item: The rate limit item :param identifiers: variable list of strings to uniquely identify this instance of the limit :param cost: The cost of this hit, default 1 :return: True if ``cost`` could be deducted from the rate limit without exceeding it N)r rrrdr0rJrKr"s rr%z#SlidingWindowCounterRateLimiter.hitsU ' & & DLL+ & KK OO       sAAAAcKtt|jj|j||j d{\}}}}|j |||||j|z dzkS70w)aK Check if the rate limit can be consumed :param item: The rate limit item :param identifiers: variable list of strings to uniquely identify this instance of the limit :param cost: The expected cost to be consumed, default 1 :return: True if the rate limit is not depleted Nr)r rrrcr0rKrirJ)rr#rr$rfrgrh_s rr*z$SlidingWindowCounterRateLimiter.testsGK 'G  \T\\;79J KAL=+]A  n&9= kkD 1$ % ALsA A> A< 1A>c Ktt|jj|j||j d{\}}}}t d|jt|j||||z }tj}|s|s t||S|j } tt} } |r|| |z z} |r|| z} t|t| | z|S7w)a  Query the reset time and remaining amount for the limit. :param item: The rate limit item :param identifiers: variable list of strings to uniquely identify this instance of the limit :return: (reset time, remaining) Nr)r rrrcr0rKr]rJrritimer rmin) rr#r$rfrgrhcurrent_expires_inr^nowexpiryprevious_reset_incurrent_reset_ins rr-z0SlidingWindowCounterRateLimiter.get_window_statss"2DLLATT DLL+ &(9        KK$$.*=}  iik-sI. .".13+  3v7N O  1F: 3%68H!II9UU7 sA C5 C3 B(C5cKtt|jj|j||j d{S7wr)r rrclear_sliding_windowr0rKr,s rr/z%SlidingWindowCounterRateLimiter.clearAsH '  |t||[94??;L MN NNsA A A Ar1) r#r rfr4rgfloatrhr4r5rwr2r7r8) r:r;r<rSrrir%r*r-r/rTrUs@rraras" X X X# X  X  XMN (NO .,V!,V14,V ,V\Nrra)zsliding-window-counterz fixed-windowz moving-window)rS __future__rrnabcrrmathrrdeprecated.sphinxrlimitsr rr typingr utilr rr storage.baserrr?rWra STRATEGIESr=rrrs# #*""150D#0DfC>kC>L;-[;-|ewNkwNwNv>*, r