U ]gp@sdZddlmZmZddlmZddlmZddlm Z ddl m Z dd lm Z m Z Gd d d eZGd d d eZGdddeZGdddeZeeedZdS)z' Asynchronous rate limiting strategies )ABCabstractmethod)cast) RateLimitItem) StorageTypes) WindowStats)MovingWindowSupportStoragec@s~eZdZedddZeddeeee dddZ eddeeee dd d Z eeee d d d Z eedd ddZdS) RateLimiter)storagecCst|tst||_dSN) isinstancer AssertionErrorr selfr r9/tmp/pip-unpacked-wheel-kizsipjx/limits/aio/strategies.py__init__szRateLimiter.__init__r costitem 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 NNotImplementedErrorrrrrrrrhits 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 Nrrrrrtests zRateLimiter.testrrrcstdS)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)) Nrrrrrrrget_window_stats+s zRateLimiter.get_window_statsNcs|j|j|IdHSr)r clearkey_forr$rrrr&9szRateLimiter.clear)__name__ __module__ __qualname__rrrrstrintboolr r"rr%r&rrrrr s   r csneZdZdZeddfdd Zddeeee dd d Z ddeeee dd d Z eee d ddZ ZS)MovingWindowRateLimiterz4 Reference: :ref:`strategies:moving window` N)r rcs2t|ds"t|ds"td|jt|dS)N acquire_entryget_moving_windowzBMovingWindowRateLimiting is not implemented for storage of type %s)hasattrr __class__superrrr2rrrBsz MovingWindowRateLimiter.__init__r rrcs,tt|jj|j||j||dIdHS)r)amountN)rr r r/r'r5 get_expiryrrrrr Ls zMovingWindowRateLimiter.hitcs>tt|j|j||j|IdH}|d}||j|kSr!Nr )rr r r0r'r5r6)rrrrresr5rrrr"Zs  zMovingWindowRateLimiter.testr#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)rr r r0r'r5r6r)rrrZ window_startZ window_itemsresetrrrr%ls  z(MovingWindowRateLimiter.get_window_stats)r(r)r*__doc__rrrr+r,r-r r"rr% __classcell__rrr4rr.=s r.c@sVeZdZdZddeeeedddZddeeeedddZ eee d d d Z d S) FixedWindowRateLimiterz3 Reference: :ref:`strategies:fixed window` r rrcs*|jj|j||d|dIdH|jkS)rFZelastic_expiryr5Nr incrr'r6r5rrrrr s  zFixedWindowRateLimiter.hitcs&|j|j|IdH|j|dkSr7)r getr'r5rrrrr"s $zFixedWindowRateLimiter.testr#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)maxr5r r@r'r6r)rrr remainingr9rrrr%s z'FixedWindowRateLimiter.get_window_statsN) r(r)r*r:rr+r,r-r r"rr%rrrrr<sr<c@s*eZdZdZddeeeedddZdS)#FixedWindowElasticExpiryRateLimiterzG Reference: :ref:`strategies:fixed window with elastic expiry` r rrcs.|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 Tr=Nr>)rrrrr5rrrr s  z'FixedWindowElasticExpiryRateLimiter.hitN) r(r)r*r:rr+r,r-r rrrrrCsrC)z fixed-windowzfixed-window-elastic-expiryz moving-windowN)r:abcrrtypingrZlimitsrr rutilrr r r r.r<rCZ STRATEGIESrrrrs    /B;