o KDi)@sdZddlmZddlZddlmZmZddlmZm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZd d lmZmZd dlmZGdddeZGdddeZGdddeZe ddGdddeZeeedZdS)z' Asynchronous rate limiting strategies ) annotationsN)ABCabstractmethod)floorinf) versionadded) RateLimitItem) StorageTypes)cast) WindowStats)MovingWindowSupportStorage)SlidingWindowCounterSupportc@sVeZdZdddZedddddZedddddZedddZdddZdS) RateLimiterstorager cCst|tsJ||_dSN) isinstancerrselfrrN/var/www/Datamplify/venv/lib/python3.10/site-packages/limits/aio/strategies.py__init__s zRateLimiter.__init__r costitemr identifiersstrrintreturnboolct)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 NotImplementedErrorrrrrrrrhit zRateLimiter.hitcr"): 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&rrrtest(r(zRateLimiter.testr cr")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$rrrrrrget_window_stats6r(zRateLimiter.get_window_statsNonecs|j|j|IdHSr)rclearkey_forr+rrrr.DszRateLimiter.clearNrr rr rrrrr r!rr rrr r rr rrr r-) __name__ __module__ __qualname__rrr'r*r,r.rrrrrs    rcsLeZdZdZdfdd Zdd dddZdd dddZdddZZS)MovingWindowRateLimiterz4 Reference: :ref:`strategies:moving window` rr r r-cs4t|dst|dstd|jt|dS)N acquire_entryget_moving_windowz@MovingWindowRateLimiting is not implemented for storage of type hasattrr% __class__superrrr<rrrMsz MovingWindowRateLimiter.__init__r rrr rrrrr!cs.tt|jj|j||j||dIdHSr#)amountN)r rrr8r/r@ get_expiryr&rrrr'Ws  zMovingWindowRateLimiter.hitcs@tt|j|j||j|IdH}|d}||j|kSr)Nr )r rrr9r/r@rA)rrrrresr@rrrr*gs  zMovingWindowRateLimiter.testr csJtt|j|j||j|IdH\}}||}t||j|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 rrr9r/r@rAr )rrr window_start window_itemsresetrrrr,{s  z(MovingWindowRateLimiter.get_window_stats)rr r r-r1r2) r4r5r6__doc__rr'r*r, __classcell__rrr>rr7Hs  r7c@s:eZdZdZdddd d ZdddddZdddZdS)FixedWindowRateLimiterz3 Reference: :ref:`strategies:fixed window` r rrr rrrrr r!cs*|jj|j|||dIdH|jkSr?)rincrr/rAr@r&rrrr's  zFixedWindowRateLimiter.hitcs(|j|j|IdH|j|dkSrB)rgetr/r@r&rrrr*s$ zFixedWindowRateLimiter.testr csHtd|j|j|j|IdH}|j|j|IdH}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)maxr@rrKr/rAr )rrr remainingrFrrrr,s  z'FixedWindowRateLimiter.get_window_statsNr1r2)r4r5r6rGr'r*r,rrrrrIs rIz4.1)versioncs`eZdZdZd fdd Zd!ddZddd"ddZddd"ddZd#ddZd$ddZ Z S)%SlidingWindowCounterRateLimiterz= Reference: :ref:`strategies:sliding window counter` rr cs4t|dr t|dstd|jt|dS)Nget_sliding_windowacquire_sliding_window_entryzHSlidingWindowCounterRateLimiting is not implemented for storage of type r:rr>rrrs z(SlidingWindowCounterRateLimiter.__init__rr previous_countrprevious_expires_infloat current_countr cCs||||S)zu Return the approximated by weighting the previous window count and adding the current window count. )rA)rrrRrSrUrrr_weighted_counts z/SlidingWindowCounterRateLimiter._weighted_countr rrrrr!cs,tt|j|j||j||IdHS)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 rrrQr/r@rAr&rrrr's z#SlidingWindowCounterRateLimiter.hitcsLtt|j|j||IdH\}}}}||||||j|dkS)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 rrrPr/rArVr@)rrrrrRrSrU_rrrr*s  z$SlidingWindowCounterRateLimiter.testr c stt|j|j||IdH\}}}}td|jt| ||||}t }|s4|s4t ||S|} t t } } |rE|| |} |rK|| } t |t | | |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 rrrPr/rArLr@rrVtimer rmin) rrrrRrSrUcurrent_expires_inrMnowexpiryprevious_reset_incurrent_reset_inrrrr,s:     z0SlidingWindowCounterRateLimiter.get_window_statsr-cs&tt|j|j||IdHSr)r rrclear_sliding_windowr/rAr+rrrr.As z%SlidingWindowCounterRateLimiter.clearr0) rr rRrrSrTrUrr rTr1r2r3) r4r5r6rGrrVr'r*r,r.rHrrr>rrOs   .rO)zsliding-window-counterz fixed-windowz moving-window)rG __future__rrXabcrrmathrrdeprecated.sphinxrlimitsr rr typingr utilr rr storage.baserrr7rIrO STRATEGIESrrrrs*       3F>{