9ir& ddlmZddlZddlZddlZddlmZmZddlm Z ddl m Z ddl Z ddlmZmZmZddlmZGdd Ze d Gd d eeeeZy)) annotationsN)Counter defaultdict)floor) versionadded)MovingWindowSupportSlidingWindowCounterSupportStorage)TimestampedSlidingWindowceZdZddZy)Entryc^tj|_|j|z|_yN)timeatimeexpiry)selfrs L/var/www/html/venv/lib/python3.12/site-packages/limits/aio/storage/memory.py__init__zEntry.__init__sYY[ jj6) N)rintreturnNone)__name__ __module__ __qualname__rrrr r s*rr z2.1)versioncDeZdZdZdgZ d dfd ZddZddZddZddZ e ddZ ddd Z ddd Z d d Zd!d Z d d"d Zd#dZ d$dZ d d"dZ d%dZd&dZ d'dZd(dZd)dZddZxZS)* MemoryStoragez rate limit storage using :class:`collections.Counter` as an in memory storage for fixed & sliding window strategies, and a simple list to implement moving window strategy. z async+memoryc t|_ttj|_i|_i|_d|_t|(|fd|i|y)Nwrap_exceptions) rstoragerasyncioLocklocks expirationseventstimersuperr)rurir"_ __class__s rrzMemoryStorage.__init__*sK4;9 5@5N -/.0 04  CoCCrcF|jj}|d=|d=|S)Nr)r&)__dict__copyrstates r __getstate__zMemoryStorage.__getstate__4s' ""$ 'N 'N rc|jj|d|_ttj |_t j|jyr) r/updater)rr$r%r& ensure_future_MemoryStorage__schedule_expiryr1s r __setstate__zMemoryStorage.__setstate__:sB U#  . d4467rcK tjt|jjD]}|j|4d{t j fd|j|d{}|jj|gr|j|d||j|<|jj|ds8|jj|d|jj|ddddd{t|jjD]{}|j|tjks(|jj|d|jj|d|jj|d}y7v7K7#1d{7swYxYw#t j$rYywxYww)Nc8tj| dS)Nc|j Sr)r)events rzAMemoryStorage.__expire_events....Gs 5<<-rkey)bisect bisect_left)evtsnows rr=z/MemoryStorage.__expire_events..FsV%7%7 3$,G&r) rlistr(keysr&r$ to_threadgetpopr'r#CancelledError)rr?cutoffrCs @r__expire_eventszMemoryStorage.__expire_events@s ))+CDKK,,./ 2::c? 2 2#*#4#4 C( $F {{sB/+/;;s+;GV+D C(;;??35 T2 sD1 2 2 2 2D,,1134 .##C(DIIK7LL$$S$/$$((d3JJNN3-  . 2 2 2 2 2$%%   sG?A G&GG&*GG BG G&G AG&1AG&G?G& GG&G# GG# G&&G<9G?;G<<G?cK|jr|jjr)tj|j |_yywr)r)doner$ create_task_MemoryStorage__expire_eventsrs r__schedule_expiryzMemoryStorage.__schedule_expiryYs:zzTZZ__. ,,T-A-A-CDDJ/sAActSr) ValueErrorrPs rbase_exceptionszMemoryStorage.base_exceptions]s rcK|j|d{|jd{|j|4d{|j|xx|z cc<|j||k(r$t j|z|j |<dddd{|jj||S7777&#1d{7swY6xYww)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 N)rGr7r&r#rr')rr?ramounts rincrzMemoryStorage.incrcshhsm$$&&&::c? = = LL  ' ||C F*(, f(<  % = =||V,, & = = = = =shC"CC"CC"C C" AC  C"$C %!C"C" C" C" CC CC"cK|j|d{|jd{|j|4d{t|j||z d|j|<dddd{|jj||S77s7]7&#1d{7swY6xYww)z decrements the counter for a given rate limit key. 0 is the minimum allowed value. :param amount: the number to increment by Nr)rGr7r&maxr#)rr?rVs rdecrzMemoryStorage.decrss hhsm$$&&&::c? C C #DLL$5$> BDLL  C C||V,, & C C C C CsgB>B!B>B#B>B%B> *B)5 B>B'!B>#B>%B>'B>)B;/B2 0B;7B>cLK|jj|dtjkrT|jj |d|jj |d|j j |d|jj|dSw)zB :param key: the key to get the counter value for rN)r'rGrr#rHr&rr?s rrGzMemoryStorage.gets|     Q '499; 6 LL  S$ '    d + JJNN3 %||Q''sB"B$cK|jj|d|jj|d|jj|d|jj|dyw)z> :param key: the key to clear rate limits for N)r#rHr'r(r&r\s rclearzMemoryStorage.clearsV d# S$' T" sD!sA2A4c K||kDry|jd{|j|4d{|jj|gt j} |j|||z }|r$|j ||z k\r dddd{yt|g|z|j|dd dddd{y77#t $rd}YhwxYw7K7#1d{7swYyxYww)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)r7r&r( setdefaultr IndexErrorrr )rr?limitrrV timestampentrys r acquire_entryzMemoryStorage.acquire_entrys  E>$$&&&::c?   KK " "3 + I &*kk#&6uv~&F F(::   ).f '? C !$    '         sDCDCD1C/*C?C/ DC+ D% C/ DC-DD C(%C/'C((C/+D-D/D5C8 6D=DcfK|jj|tjSw)z; :param key: the key to get the expiry for )r'rGrr\s r get_expiryzMemoryStorage.get_expirys' ##C55s/1cKtj}|jj|gx}r1tj|||z d}||dz j |fS|dfSw)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)rds rr=z1MemoryStorage.get_moving_window..s %++rr>r)rr(rGr@rAr)rr?rbrrcr(oldests rget_moving_windowzMemoryStorage.get_moving_windowssIIK [[__S"- -6 -'')f,-3MF&1*%++V3 3!|sA(A*cK||kDrytj}|j|||\}}|j||||d{\}} } } || z|z | z} t| |z|kDry|j |d|z|d{} || z|z | z} t| |kDr|j ||d{yy7z7;7 w)NF)rVT)rsliding_window_keys_get_sliding_window_inforrWrZ) rr?rbrrVrC previous_key current_keyprevious_count previous_ttl current_countr,weighted_counts racquire_sliding_window_entryz*MemoryStorage.acquire_sliding_window_entrys E>iik$($<$C ?C C  C cKtj}|j|||\}}|j||||d{S7wr)rrorprr?rrCrqrrs rget_sliding_windowz MemoryStorage.get_sliding_windowsRiik$($<$?6ILS6\Q./69FB |]KGG63sA2A.A2A0>A20A2c Kyw)z- check if storage is healthy TrrPs rcheckzMemoryStorage.check s scBKtt|jt|j}|jj |j j |jj |j j |Swr)rYlenr#r(r^r'r&)r num_itemss rresetzMemoryStorage.resetslDLL)3t{{+;<      sBBc |jr6|jjs|jjyyy#t$rYywxYwr)r)rMcancel RuntimeErrorrPs r__del__zMemoryStorage.__del__sD zz$**//"3 !!##4z   sAA AA)NF)r+z str | Noner"boolr,strrr)rdict[str, limits.typing.Any])r2rrr)rr)rz-type[Exception] | tuple[type[Exception], ...])rj)r?rrr~rVrrr)r?rrVrrr)r?rrr)r?rrr) r?rrbrrrrVrrr)r?rrr~)r?rrbrrrrztuple[float, int])r?rrrrtuple[int, float, int, float])r?rrrrr) rqrrrrrrrCr~rr)rr)rz int | None)rrr__doc__STORAGE_SCHEMErr3r8rOr7propertyrTrWrZrGr^rergrlrwrzr|rprrr __classcell__)r-s@rr r s %%N ?DDD7;DJMD D 8 2E 6 - - ("@A",/9< 66",/ 2       B  # & ' HHH H  H ' H rr ) __future__rr$r@r collectionsrrmathrdeprecated.sphinxr limits.typinglimitslimits.aio.storage.baserr r limits.storage.baser r r rrrrsc" ,* 9** eD "=?WDDr