a a*j*@s4UdZddlmZddlZddlmZmZddlmZm Z ddl m Z m Z ddl mZdd lmZdd lmZmZmZdd lmZmZdd lmZGd ddedZGdddeZGdddeZe ddGdddeZe dddGdddeZeeeeeeeeefZ eeeedZ!de"d<dS)z Rate limiting strategies ) annotationsN)ABCMetaabstractmethod)floorinf) deprecated versionadded)SlidingWindowCounterSupport) RateLimitItem)MovingWindowSupportStorage StorageTypes)Unioncast) WindowStatsc@s~eZdZddddZeddddd d d d d Zeddddd d d ddZeddddddZddddddZdS) RateLimiterrstoragecCst|tsJ||_dSN) isinstancer rselfrrM/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/strategies.py__init__szRateLimiter.__init__r costr strintboolitem identifiersrreturncGstdS) 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 NNotImplementedErrorrr"rr#rrrhits zRateLimiter.hitcGstdS)a 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 Nr&r(rrrtest&s zRateLimiter.testrr"r#r$cGstdS)  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&rr"r#rrrget_window_stats2s zRateLimiter.get_window_statsNonecGs|j|j|Sr)rclearkey_forr-rrrr0>szRateLimiter.clearN) __name__ __module__ __qualname__rrr)r*r.r0rrrrrs   r) metaclasscsleZdZdZddfdd Zdddd d d d d dZdddd d d d ddZdd ddddZZS)MovingWindowRateLimiterz4 Reference: :ref:`strategies:moving window` rrcs2t|ds"t|ds"td|jt|dS)N acquire_entryget_moving_windowzBMovingWindowRateLimiting is not implemented for storage of type %shasattrr' __class__superrrr;rrrGsz MovingWindowRateLimiter.__init__r rr rrr r!cGs&tt|jj|j||j||dS)a 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: (reset time, remaining) )amount)rr rr7r1r> get_expiryr(rrrr)Qs zMovingWindowRateLimiter.hitcGs0tt|j|j||j|d|j|kSa 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 r )rr rr8r1r>r?r(rrrr*`s zMovingWindowRateLimiter.testrr+cGsBtt|j|j||j|\}}||}t||j|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 rr8r1r>r?r)rr"r#Z window_startZ window_itemsresetrrrr.ss  z(MovingWindowRateLimiter.get_window_stats) r2r3r4__doc__rr)r*r. __classcell__rrr=rr6Bs  r6c@sVeZdZdZdddddddd d Zdddddddd d Zddd dddZdS)FixedWindowRateLimiterz3 Reference: :ref:`strategies:fixed window` r rr rrr r!cGs$|jj|j||d|d|jkS)r%FZelastic_expiryr>rincrr1r?r>r(rrrr)s zFixedWindowRateLimiter.hitcGs |j|j||j|dkSr@)rgetr1r>r(rrrr*s zFixedWindowRateLimiter.testrr+cGs:td|j|j|j|}|j|j|}t||S)r,r)maxr>rrHr1r?r)rr"r# remainingrArrrr.s z'FixedWindowRateLimiter.get_window_statsN)r2r3r4rBr)r*r.rrrrrDs rDz4.1)versioncseZdZdZddfdd Zdddddd d d Zd d dddddddZd d dddddddZddddddZZ S)SlidingWindowCounterRateLimiterz= Reference: :ref:`strategies:sliding window counter` rrcs2t|drt|ds"td|jt|dS)Nget_sliding_windowacquire_sliding_window_entryzJSlidingWindowCounterRateLimiting is not implemented for storage of type %sr9rr=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_countr rrr r!cGs$tt|j|j||j||S)r%)rr rrNr1r>r?r(rrrr)s z#SlidingWindowCounterRateLimiter.hitcGsDtt|j|j||\}}}}||||||j|dkSr@)rr rrMr1r?rSr>)rr"rr#rPrQrR_rrrr*s   z$SlidingWindowCounterRateLimiter.testrr+c Gstt|j|j||\}}}}td|jt| ||||}t }|s`|s`t ||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: WindowStats(reset time, remaining) r)rr rrMr1r?rIr>rrStimerrmin) rr"r#rPrQrRZcurrent_expires_inrJnowZexpiryZprevious_reset_inZcurrent_reset_inrrrr.s2      z0SlidingWindowCounterRateLimiter.get_window_stats) r2r3r4rBrrSr)r*r.rCrrr=rrLs   rLalways)rKactionc@s*eZdZdZdddddddd d Zd S) #FixedWindowElasticExpiryRateLimiterzG Reference: :ref:`strategies:fixed window with elastic expiry` r rr rrr r!cGs$|jj|j||d|d|jkS)r%TrErFr(rrrr)'s z'FixedWindowElasticExpiryRateLimiter.hitN)r2r3r4rBr)rrrrrZ!srZ)zsliding-window-counterz fixed-windowzfixed-window-elastic-expiryz moving-windowzdict[str, KnownStrategy] STRATEGIES)#rB __future__rrUabcrrmathrrZdeprecated.sphinxrrZlimits.storage.baser Zlimitsr rr r rtypingrrutilrrr6rDrLrZtypeZ KnownStrategyr[__annotations__rrrrs:    -B4h