o ;h'@sUdZddlmZddlZddlmZmZddlmZm Z ddl m Z ddl m Z dd lmZdd lmZmZmZdd lmZdd lmZGd ddedZGdddeZGdddeZe ddGdddeZeeeeBeeBZeeedZded<dS)z Rate limiting strategies ) annotationsN)ABCMetaabstractmethod)floorinf) versionadded)SlidingWindowCounterSupport) RateLimitItem)MovingWindowSupportStorage StorageTypes)cast) WindowStatsc@sVeZdZdddZedddddZedddddZedddZdddZdS) RateLimiterstorager cCst|tsJ||_dSN) isinstancer rselfrrD/var/www/html/venv/lib/python3.10/site-packages/limits/strategies.py__init__s zRateLimiter.__init__r costitemr identifiersstrrintreturnboolcGt 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 NotImplementedErrorrrrrrrrhit zRateLimiter.hitcGr!)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 r$r&rrrtest&r(zRateLimiter.testrcGr!)  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) r$rrrrrrget_window_stats2r(zRateLimiter.get_window_statsNonecGs|j|j|Sr)rclearkey_forr+rrrr.>szRateLimiter.clearNrr rr rrrrrr rr rrrrrr rrrr-) __name__ __module__ __qualname__rrr'r)r,r.rrrrrs    r) metaclasscsLeZdZdZdfdd ZdddddZdddddZdddZZS)MovingWindowRateLimiterz4 Reference: :ref:`strategies:moving window` rr cs4t|dst|dstd|jt|dS)N acquire_entryget_moving_windowz@MovingWindowRateLimiting is not implemented for storage of type hasattrr% __class__superrrr=rrrGsz MovingWindowRateLimiter.__init__r rrr rrrrrr 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 rr9r/rA 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 rr:r/rArBr&rrrr)`s zMovingWindowRateLimiter.testrcGsBtt|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 rr:r/rArBr)rrrZ window_startZ window_itemsresetrrrr,ss  z(MovingWindowRateLimiter.get_window_statsr0r1r2) r4r5r6__doc__rr'r)r, __classcell__rrr?rr8Bs  r8c@s:eZdZdZdddd d ZdddddZdddZdS)FixedWindowRateLimiterz3 Reference: :ref:`strategies:fixed window` r rrr rrrrrr cGs"|jj|j|||d|jkS)r#r@)rincrr/rBrAr&rrrr's zFixedWindowRateLimiter.hitcGs |j|j||j|dkSrC)rgetr/rAr&rrrr)s zFixedWindowRateLimiter.testrcGs:td|j|j|j|}|j|j|}t||S)r*r)maxrArrIr/rBr)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_countrcCs||||S)zu Return the approximated by weighting the previous window count and adding the current window count. )rB)rrrPrQrSrrr_weighted_counts z/SlidingWindowCounterRateLimiter._weighted_countr rrrrr cGs$tt|j|j||j||Sr")rrrrOr/rArBr&rrrr's z#SlidingWindowCounterRateLimiter.hitcGsDtt|j|j||\}}}}||||||j|dkSrC)rrrrNr/rBrTrA)rrrrrPrQrS_rrrr)s   z$SlidingWindowCounterRateLimiter.testrc Gstt|j|j||\}}}}td|jt| ||||}t }|s0|s0t ||S|} t t } } |rA|| |} |rG|| } 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)rrrrNr/rBrJrArrTtimerrmin) rrrrPrQrSZcurrent_expires_inrKnowZexpiryZprevious_reset_inZcurrent_reset_inrrrr,s2      z0SlidingWindowCounterRateLimiter.get_window_statsr-cGstt|j|j||Sr)rrrZclear_sliding_windowr/rBr+rrrr.s z%SlidingWindowCounterRateLimiter.clearr0) rr rPrrQrRrSrrrRr1r2r3) r4r5r6rErrTr'r)r,r.rFrrr?rrMs   &rM)zsliding-window-counterz fixed-windowz moving-windowzdict[str, KnownStrategy] STRATEGIES) rE __future__rrVabcrrmathrrZdeprecated.sphinxrZlimits.storage.baserZlimitsr rr r r typingrutilrrr8rGrMtypeZ KnownStrategyrY__annotations__rrrrs4      -B3n