9i)8UdZddlmZddlZddlmZmZddlmZm Z ddl m Z ddl m Z dd lmZdd lmZmZmZdd lmZdd lmZGd deZGddeZGddeZe dGddeZeeeezeezZeeedZded<y)z Rate limiting strategies ) annotationsN)ABCMetaabstractmethod)floorinf) versionadded)SlidingWindowCounterSupport) RateLimitItem)MovingWindowSupportStorage StorageTypes)cast) WindowStatsc^eZdZd dZeddd dZeddd dZed dZd dZy) RateLimiterc6t|tsJ||_yN) isinstancer storage)selfrs D/var/www/html/venv/lib/python3.12/site-packages/limits/strategies.py__init__zRateLimiter.__init__s'7+++ ' r costctS 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 NotImplementedErrorritemr identifierss rhitzRateLimiter.hit "!rct)aT Check the rate limit without consuming from it. :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 r r"s rtestzRateLimiter.test(r&rct)  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) r rr#r$s rget_window_statszRateLimiter.get_window_stats6s "!rcR|jj|j|Sr)rclearkey_forr+s rr.zRateLimiter.clearBs"||!!,$,, "<==rNrrr#r r$strrintreturnboolr#r r$r2r4rr#r r$r2r4None) __name__ __module__ __qualname__rrr%r(r,r.rrrrsN(FG " "GH " " " ">rr) metaclasscHeZdZdZdfd Zddd dZddd dZd 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__KsI G_ -BU1V%",,-/  !rr rctt|jj|j||j |j |Sr)amount)rr rrAr/rJ get_expiryr"s rr%zMovingWindowRateLimiter.hitUsH'6DD DLL+ & T__5FtE  rctt|jj|j||j |j d|j |z kSaK 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 r )rr rrBr/rJrKr"s rr(zMovingWindowRateLimiter.testes_ $dll 3 E E k* !    {{T!  " rctt|jj|j||j |j \}}||j z}t||j |z S)a returns the number of requests remaining within this limit. :param item: The rate limit item :param identifiers: variable list of strings to uniquely identify this instance of the limit :return: tuple (reset time, remaining) )rr rrBr/rJrKr)rr#r$ window_start window_itemsresets rr,z(MovingWindowRateLimiter.get_window_statszsl&* &  LDLL+6 T__EV W # lt005$++ "<==rr0r1r6) r9r:r;__doc__rr%r(r, __classcell__rEs@rr?r?Fs*"GH HI *>rr?c4eZdZdZddddZddddZd dZy) FixedWindowRateLimiterz3 Reference: :ref:`strategies:fixed window` r rc|jj|j||j||jkSrI)rincrr/rKrJr"s rr%zFixedWindowRateLimiter.hitsI LL   k*!   {{   rcx|jj|j||j|z dzkSrM)rgetr/rJr"s rr(zFixedWindowRateLimiter.tests8||  k :;dkkD>PST>TTTrc td|j|jj|j|z }|jj |j|}t ||S)r*r)maxrJrrZr/rKr)rr#r$ remainingrQs rr,z'FixedWindowRateLimiter.get_window_statss_4;;)9)9,$,, :T)UUV  ''  k(BC5),,rNr1r6)r9r:r;rRr%r(r,r<rrrVrVs%GH *HI U -rrVz4.1)versioncleZdZdZd fd Z d dZddd dZddd dZd 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-WWrr rctt|jj|j||j |j |Sr)rr rrcr/rJrKr"s rr%z#SlidingWindowCounterRateLimiter.hitsG ' & & DLL+ & KK OO     rctt|jj|j||j \}}}}|j |||||j|z dzkSrM)rr rrbr/rKrhrJ)rr#rr$rerfrg_s rr(z$SlidingWindowCounterRateLimiter.tests|AE 'A  \T\\;79J K >+]A  n&9= kkD 1$ % rc tt|jj|j||j \}}}}t d|jt|j||||z }tj}|s|s t||S|j } tt} } |r|| |z z} |r|| z} t|t| | z|S)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: WindowStats(reset time, remaining) r)rr rrbr/rKr\rJrrhtimerrmin) rr#r$rerfrgcurrent_expires_inr]nowexpiryprevious_reset_incurrent_reset_ins rr,z0SlidingWindowCounterRateLimiter.get_window_statssRV 'R  \T\\;79J K O+]SS DLL+ &(9  rr0) r#r rer3rffloatrgr3r4rvr1r6r7) r9r:r;rRrrhr%r(r,r.rSrTs@rr`r`sl" X X X# X  X  XGH (HI ,$VL rr`)zsliding-window-counterz fixed-windowz moving-windowzdict[str, KnownStrategy] STRATEGIES) rR __future__rrmabcrrmathrrdeprecated.sphinxrlimits.storage.baser limitsr rr r rtypingrutilrrr?rVr`type KnownStrategyrw__annotations__r<rrrs# '*;!??.>G.>bB>kB>J4-[4-nen kn n d () !"# "#$>*,( $r