ZeIi)6dZddlmZddlZddlmZmZddlmZm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZd d lmZmZd dlmZGddeZGddeZGddeZe dGddeZeeedZdS)z' Asynchronous rate limiting strategies ) annotationsN)ABCabstractmethod)floorinf) versionadded) RateLimitItem) StorageTypes)cast) WindowStats)MovingWindowSupportStorage)SlidingWindowCounterSupportcreZdZddZeddddZeddddZeddZddZdS) RateLimiterstorager cBt|tsJ||_dSN) isinstancerr)selfrs Z/var/www/html/MCyber-Diagnostic/venv/lib/python3.11/site-packages/limits/aio/strategies.py__init__zRateLimiter.__init__s#'7+++++ ' rcostitemr identifiersstrrintreturnboolcKt)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 NotImplementedErrorrrrrs rhitzRateLimiter.hit"!rcKt): 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*rr cKt)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&rrrs rget_window_statszRateLimiter.get_window_stats6r*rNonecVK|j|j|d{VSr)rclearkey_forr/s rr3zRateLimiter.clearDs7\''  k(BCCCCCCCCCrNrr rr rr rr!r"r#rr rr r"r rr rr r"r1) __name__ __module__ __qualname__rrr)r-r0r3rrrrs((((LM " " " " "^ "MN " " " " "^ " " " "^ "DDDDDDrrcHeZdZdZdfd ZddddZddddZddZxZS)MovingWindowRateLimiterz4 Reference: :ref:`strategies:moving window` rr r"r1ct|ds't|dstd|jt|dS)N acquire_entryget_moving_windowz@MovingWindowRateLimiting is not implemented for storage of type hasattrr' __class__superrrrrDs rrz MovingWindowRateLimiter.__init__Msq G_ - - 18BU1V1V &/",//  !!!!!rrrrr rr rr!r#cKtt|j|j||j||d{VSr%)amountN)r rrr@r4rI get_expiryr(s rr)zMovingWindowRateLimiter.hitWso-t|<<JJ DL+ & T__5F5FtK         rcKtt|j|j||j|d{V}|d}||j|z kSr,Nr)r rrrAr4rIrJ)rrrrresrIs rr-zMovingWindowRateLimiter.testgs,dl;;MM DL+ & K OO           Qt+++rr cKtt|j|j||j|d{V\}}||z}t||j|z S)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 rrrAr4rIrJr )rrr window_start window_itemsresets rr0z(MovingWindowRateLimiter.get_window_stats{s,0 , ,  LDL+6 T__EVEV W W&X&X&X&X&X&X" lt0005$+ "<===r)rr r"r1r6r7) r9r:r;__doc__rr)r-r0 __classcell__rDs@rr>r>Hs""""""MN      NO,,,,,,(>>>>>>>>rr>c6eZdZdZdddd Zdddd ZddZdS)FixedWindowRateLimiterz3 Reference: :ref:`strategies:fixed window` rrrr rr rr!r"r#cK|j|j|||d{V|jkSrH)rincrr4rJrIr(s rr)zFixedWindowRateLimiter.hitsm,## k*!!$       {   rctK|j|j|d{V|j|z dzkSrL)rgetr4rIr(s rr-zFixedWindowRateLimiter.testsM,""<4<#=>> > > > > > >tASVWAW W rr c Ktd|j|j|j|d{Vz }|j|j|d{V}t ||S)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)maxrIrrZr4rJr )rrr remainingrQs rr0z'FixedWindowRateLimiter.get_window_statss K 0 0{1K L LLLLLLL L  l--ldlK.HIIIIIIII5),,,rNr6r7)r9r:r;rRr)r-r0r<rrrVrVs|MN      *NO      ------rrVz4.1)versioncXeZdZdZdfd Zdd ZddddZddddZddZddZ xZ S)SlidingWindowCounterRateLimiterz= Reference: :ref:`strategies:sliding window counter` rr ct|drt|dstd|jt|dS)Nget_sliding_windowacquire_sliding_window_entryzHSlidingWindowCounterRateLimiting is not implemented for storage of type rBrFs rrz(SlidingWindowCounterRateLimiter.__init__svw 455 W 3> >  &/",//  !!!!!rrr previous_countr!previous_expires_infloat current_countr"c<||z|z |zS)zu Return the approximated by weighting the previous window count and adding the current window count. )rJ)rrrdrergs r_weighted_countz/SlidingWindowCounterRateLimiter._weighted_counts$ 33doo6G6GG-WWrrrrr rr#cKtt|j|j||j||d{VS)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 rrrcr4rIrJr(s rr)z#SlidingWindowCounterRateLimiter.hitsp '  & & DL+ & K OO             rcKtt|j|j||d{V\}}}}||||||j|z dzkS)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 rrrbr4rJrirI)rrrrrdrerg_s rr-z$SlidingWindowCounterRateLimiter.testsGK 'G G  \T\;79J9J K KALALALALALAL=+]A  n&9=  kD 1$ % rr c Ktt|j|j||d{V\}}}}t d|jt| ||||z }tj }|s|st||S|} 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: (reset time, remaining) Nr)r rrrbr4rJr\rIrritimer rmin) rrrrdrergcurrent_expires_inr]nowexpiryprevious_reset_incurrent_reset_ins rr0z0SlidingWindowCounterRateLimiter.get_window_statssJ"2DLAATT DL+ &(9(9              K$$.*=}   ikk /- /sI.. ."".13+  P 3v7N O   ;1F: 3%68H!I!II9UUUrr1cKtt|j|j||d{VSr)r rrclear_sliding_windowr4rJr/s rr3z%SlidingWindowCounterRateLimiter.clearAsg '   |t|[94??;L;L M MNNNNNN Nrr5) rr rdr!rerfrgr!r"rfr6r7r8) r9r:r;rRrrir)r-r0r3rSrTs@rr`r`s"""""" X X X XMN      (NO      .,V,V,V,V\NNNNNNNNrr`)zsliding-window-counterz fixed-windowz moving-window)rR __future__rrnabcrrmathrrdeprecated.sphinxrlimitsr rr typingr utilr rr storage.baserrr>rVr` STRATEGIESr<rrrs#""""" ########******""""""""""""111111115555550D0D0D0D0D#0D0D0DfC>C>C>C>C>kC>C>C>L;-;-;-;-;-[;-;-;-|ewNwNwNwNwNkwNwNwNv>*, r