j"ddlmZddlZddlZddlZddlmZmZddlm Z ddl Z ddl m Z mZmZmZGddZGdd ee eeZy) ) annotationsN)Counter defaultdict)floor)MovingWindowSupportSlidingWindowCounterSupportStorageTimestampedSlidingWindowceZdZddZy)Entryc^tj|_|j|z|_yN)timeatimeexpiry)selfrs _/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/limits/storage/memory.py__init__zEntry.__init__sYY[ jj6) N)rfloatreturnNone)__name__ __module__ __qualname__rrrr r s*rr ceZdZdZdgZddfd ZddZddZddZddZ e ddZ ddd Z ddd Z dd Zd d Zdd!d Zd"dZd#dZ d d!dZ d$dZ d%dZd&dZd'dZd(dZxZS)) MemoryStoragez rate limit storage using :class:`collections.Counter` as an in memory storage for fixed and sliding window strategies, and a simple list to implement moving window strategy. memoryc t|_ttj|_i|_i|_tjd|j|_ |jjt|4|fd|i|y)N{Gz?wrap_exceptions)rstorager threadingRLocklocks expirationseventsTimer_MemoryStorage__expire_eventstimerstartsuperr)rurir"_ __class__s rrzMemoryStorage.__init__$sj3:9 8CIOO8T -/.0 &/oodDz/MemoryStorage.__expire_events..>s  rkey) listr(keysr&getbisect bisect_leftrpopr'r#)rr@r(oldests r__expire_eventszMemoryStorage.__expire_events9s+ ((*+CC![[__S"5565#// 2MF(,{{3'7'@DKK${{sD1JJNN3-!,((--/0C$ 3   d+  $$S$/ sD) 1!s B"FF c|jjs@tjd|j|_|jj yyr8)r+is_aliver$r)r*r,rs r__schedule_expiryzMemoryStorage.__schedule_expiryIs?zz""$"t/C/CDDJ JJ   %rctSr) ValueErrorrKs rbase_exceptionszMemoryStorage.base_exceptionsNs rc^|j||j|j|5|j|xx|z cc<|j||k(r$t j|z|j |<ddd|jj|dS#1swY%xYw)z increments the counter for a given rate limit key :param key: the key to increment :param expiry: amount in seconds for the key to expire in :param amount: the number to increment by Nr)rC_MemoryStorage__schedule_expiryr&r#rr')rr@ramounts rincrzMemoryStorage.incrTs    ZZ_ LL  ' ||C F*(, f(<  %||Q'' _s AB##B,c|j||j|j|5t|j||z d|j|<ddd|jj|dS#1swY%xYw)z decrements the counter for a given rate limit key :param key: the key to decrement :param amount: the number to decrement by rN)rCrQr&maxr#)rr@rRs rdecrzMemoryStorage.decrdso    ZZ_ #DLL$5$> BDLL ||Q''_s *A??BcD|jj|dtjkrT|jj |d|jj |d|j j |d|jj|dS)zB :param key: the key to get the counter value for rN)r'rCrr#rFr&rr@s rrCzMemoryStorage.getrsx     Q '499; 6 LL  S$ '    d + JJNN3 %||Q''rc|jj|d|jj|d|jj|d|jj|dy)z> :param key: the key to clear rate limits for N)r#rFr'r(r&rXs rclearzMemoryStorage.clear~sR d# S$' T" sD!rc||kDry|j|j|5|jj|gt j} |j|||z }|r|j ||z k\r dddyt|g|z|j|dd dddy#t $rd}YTwxYw#1swYyxYw)z :param key: rate limit key to acquire an entry in :param limit: amount of entries allowed :param expiry: expiry of the entry :param amount: the number of entries to acquire FNrT)rQr&r( setdefaultr IndexErrorrr )rr@limitrrR timestampentrys r acquire_entryzMemoryStorage.acquire_entrys E>  ZZ_ KK " "3 + I  C(8 F(::_).f '? C !$_    _s51CB4-C C4 C?CCCCc^|jj|tjS)z; :param key: the key to get the expiry for )r'rCrrXs r get_expiryzMemoryStorage.get_expirys# ##C55rctj}|jj|gx}r1tj|||z d}||dz j |fS|dfS)z returns the starting point and the number of entries in the moving window :param key: rate limit key :param expiry: expiry of entry :return: (start of window, number of acquired entries) c|j Sr)r)r`s rr>z1MemoryStorage.get_moving_window..s %++rr?r)rr(rCrDrEr)rr@r^rr_r(rGs rget_moving_windowzMemoryStorage.get_moving_windowsoIIK [[__S"- -6 -'')f,-3MF&1*%++V3 3!|rcV||kDrytj}|j|||\}}|j||||\}} } } || z|z | z} t| |z|kDry|j |d|z|} || z|z | z} t| |kDr|j ||yy)NF)rRT)rsliding_window_keys_get_sliding_window_inforrSrV) rr@r^rrRnow previous_key current_keyprevious_count previous_ttl current_countr/weighted_counts racquire_sliding_window_entryz*MemoryStorage.acquire_sliding_window_entrys E>iik$($<$?6ILS6\Q./69FB |]KGGrc~tj}|j|||\}}|j||||Sr)rrjrkrr@rrlrmrns rget_sliding_windowz MemoryStorage.get_sliding_windows@iik$($<$iik$($<$rsF"  ,** k "=?Wkr