o ;h'@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) isinstancerrselfrrH/var/www/html/venv/lib/python3.10/site-packages/limits/aio/strategies.py__init__s zRateLimiter.__init__r costitemr identifiersstrrintreturnboolct) 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 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 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_stats2s zRateLimiter.get_window_statsNonecs|j|j|IdHSr)rclearkey_forr+rrrr.@szRateLimiter.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<rrrIsz MovingWindowRateLimiter.__init__r rrr rrrrr!cs.tt|jj|j||j||dIdHSr#)amountN)r rrr8r/r@ get_expiryr&rrrr'Ss  zMovingWindowRateLimiter.hitcs@tt|j|j||j|IdH}|d}||j|kSr)Nr )r rrr9r/r@rA)rrrrresr@rrrr*as  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 )rrrZ window_startZ window_itemsresetrrrr,ss  z(MovingWindowRateLimiter.get_window_stats)rr r r-r1r2) r4r5r6__doc__rr'r*r, __classcell__rrr>rr7Ds  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@rrIr/rAr )rrr remainingrDrrrr,s  z'FixedWindowRateLimiter.get_window_statsNr1r2)r4r5r6rEr'r*r,rrrrrGs rGz4.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)rrrPrQrSrrr_weighted_counts z/SlidingWindowCounterRateLimiter._weighted_countr rrrrr!cs,tt|j|j||j||IdHS)z 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 N)r rrrOr/r@rAr&rrrr's z#SlidingWindowCounterRateLimiter.hitcsLtt|j|j||IdH\}}}}||||||j|dkS)a 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 Nr )r rrrNr/rArTr@)rrrrrPrQrS_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 rrrNr/rArJr@rrTtimer rmin) rrrrPrQrSZcurrent_expires_inrKnowZexpiryZprevious_reset_inZcurrent_reset_inrrrr,s:     z0SlidingWindowCounterRateLimiter.get_window_statsr-cs&tt|j|j||IdHSr)r rrZclear_sliding_windowr/rAr+rrrr.1s z%SlidingWindowCounterRateLimiter.clearr0) rr rPrrQrRrSrr rRr1r2r3) r4r5r6rErrTr'r*r,r.rFrrr>rrMs   .rM)zsliding-window-counterz fixed-windowz moving-window)rE __future__rrVabcrrmathrrZdeprecated.sphinxrZlimitsr rr typingr utilr rrZ storage.baserrr7rGrMZ STRATEGIESrrrrs*       /B:w