a a*j*@s dZddlmZddlZddlmZmZddlmZm Z ddl m Z m Z ddl mZdd lmZdd lmZdd lmZd d lmZmZd dlmZGdddeZGdddeZGdddeZe ddGdddeZe ddGdddeZeeeedZdS)z' Asynchronous rate limiting strategies ) annotationsN)ABCabstractmethod)floorinf) deprecated versionadded) RateLimitItem) StorageTypes)cast) WindowStats)MovingWindowSupportStorage)SlidingWindowCounterSupportc@s~eZdZddddZeddddd d d d d Zeddddd d d ddZeddddddZddddddZdS) RateLimiterr storagecCst|tsJ||_dSN) isinstancerrselfrrQ/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/aio/strategies.py__init__szRateLimiter.__init__rcostr strintboolitem identifiersrreturncstdS) 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 NNotImplementedErrorrr"rr#rrrhits zRateLimiter.hitcstdS) 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 Nr&r(rrrtest&s zRateLimiter.testr r"r#r$cstdS)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)) Nr&rr"r#rrrget_window_stats2s zRateLimiter.get_window_statsNonecs|j|j|IdHSr)rclearkey_forr-rrrr0@szRateLimiter.clearN) __name__ __module__ __qualname__rrr)r+r.r0rrrrrs   rcsneZdZdZdddfdd Zddd d d d d ddZddd d d d d ddZd d ddddZZS)MovingWindowRateLimiterz4 Reference: :ref:`strategies:moving window` r r/)rr$cs2t|ds"t|ds"td|jt|dS)N acquire_entryget_moving_windowzBMovingWindowRateLimiting is not implemented for storage of type %shasattrr' __class__superrrr:rrrIsz MovingWindowRateLimiter.__init__rrr rrr r!cs,tt|jj|j||j||dIdHS)r%)amountN)r rrr6r1r= get_expiryr(rrrr)Ss zMovingWindowRateLimiter.hitcs>tt|j|j||j|IdH}|d}||j|kSr*Nr)r rrr7r1r=r>)rr"rr#resr=rrrr+as  zMovingWindowRateLimiter.testr r,csHtt|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 rrr7r1r=r>r )rr"r#Z window_startZ window_itemsresetrrrr.ss  z(MovingWindowRateLimiter.get_window_stats) r2r3r4__doc__rr)r+r. __classcell__rrr<rr5Ds  r5c@sVeZdZdZdddddddd d Zdddddddd d Zddd dddZdS)FixedWindowRateLimiterz3 Reference: :ref:`strategies:fixed window` rrr rrr r!cs*|jj|j||d|dIdH|jkS)r%FZelastic_expiryr=Nrincrr1r>r=r(rrrr)s  zFixedWindowRateLimiter.hitcs&|j|j|IdH|j|dkSr?)rgetr1r=r(rrrr+s $zFixedWindowRateLimiter.testr r,csFtd|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=rrHr1r>r )rr"r# remainingrArrrr.s z'FixedWindowRateLimiter.get_window_statsN)r2r3r4rBr)r+r.rrrrrDsrDz4.1)versioncseZdZdZddfdd Zdddddd d d Zd d dddddddZd d dddddddZddddddZZ S)SlidingWindowCounterRateLimiterz= Reference: :ref:`strategies:sliding window counter` r rcs2t|drt|ds"td|jt|dS)Nget_sliding_windowacquire_sliding_window_entryzJSlidingWindowCounterRateLimiting is not implemented for storage of type %sr8rr<rrrs z(SlidingWindowCounterRateLimiter.__init__r rfloat)r"previous_countprevious_expires_in current_countr$cCs||||S)zu Return the approximated by weighting the previous window count and adding the current window count. )r>)rr"rPrQrRrrr_weighted_counts z/SlidingWindowCounterRateLimiter._weighted_countrrrr r!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 rrrNr1r=r>r(rrrr)s z#SlidingWindowCounterRateLimiter.hitcsJtt|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 rrrMr1r>rSr=)rr"rr#rPrQrR_rrrr+s  z$SlidingWindowCounterRateLimiter.testr r,c stt|j|j||IdH\}}}}td|jt| ||||}t }|sf|sft ||S|} t t } } |r|| |} |r|| } 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 rrrMr1r>rIr=rrStimer rmin) rr"r#rPrQrRZcurrent_expires_inrJnowZexpiryZprevious_reset_inZcurrent_reset_inrrrr.s8     z0SlidingWindowCounterRateLimiter.get_window_stats) r2r3r4rBrrSr)r+r.rCrrr<rrLs   rLc@s*eZdZdZdddddddd d Zd S) #FixedWindowElasticExpiryRateLimiterzG Reference: :ref:`strategies:fixed window with elastic expiry` rrr rrr r!cs.|jj|j||d|dIdH}||jkS)a Consume the rate limit :param item: a :class:`limits.limits.RateLimitItem` instance :param identifiers: variable list of strings to uniquely identify the limit :param cost: The cost of this hit, default 1 TrENrF)rr"rr#r=rrrr)9s  z'FixedWindowElasticExpiryRateLimiter.hitN)r2r3r4rBr)rrrrrX3srX)zsliding-window-counterz fixed-windowzfixed-window-elastic-expiryz moving-window)rB __future__rrUabcrrmathrrZdeprecated.sphinxrrZlimitsr rr typingr utilr rrZ storage.baserrr5rDrLrXZ STRATEGIESrrrrs.      /B;q