ZeIi"ddlmZddlZddlZddlZddlmZmZddlm Z ddl Z ddl m Z mZmZmZGddZGdd ee eeZdS) ) annotationsN)Counter defaultdict)floor)MovingWindowSupportSlidingWindowCounterSupportStorageTimestampedSlidingWindowceZdZddZdS)EntryexpiryfloatreturnNonecTtj|_|j|z|_dSN)timeatimer )selfr s Z/var/www/html/MCyber-Diagnostic/venv/lib/python3.11/site-packages/limits/storage/memory.py__init__zEntry.__init__s!Y[[ j6) N)r rrr)__name__ __module__ __qualname__rrrr r s(******rr ceZdZdZdgZd0d1fd Zd2dZd3dZd4dZd4dZ e d5dZ d6d7dZ d6d8dZ d9dZd:dZd6d;d!Zdd*Zd?d+Zd@d,ZdAd-ZdBd/ZxZS)C 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. memoryNFuri str | Nonewrap_exceptionsbool_strc .t|_ttj|_i|_i|_tjd|j |_ |j tj |fd|i|dS)N{Gz?r")rstorager threadingRLocklocks expirationseventsTimer_MemoryStorage__expire_eventstimerstartsuperr)rr r"r$ __class__s rrzMemoryStorage.__init__$s3:99 8CIO8T8T -/.0 &/odDz/MemoryStorage.__expire_events..>s  rkey) listr-keysr+getbisect bisect_leftrpopr,r()rrDr-oldests r__expire_eventszMemoryStorage.__expire_events9s ((**++ . .CC . .![__S"5556A#/ 2M2MF(,{3'7'@DK${sD11.JNN3--- . . . . . . . . . . . . . . .(--//00 * *C$ 33   d+++ $$S$/// sD)))  * *sBCC! $C! c|js:tjd|j|_|jdSdSr<)r0is_aliver)r.r/r1rs r__schedule_expiryzMemoryStorage.__schedule_expiryIsRz""$$ "t/CDDDJ J         r-type[Exception] | tuple[type[Exception], ...]ctSr) ValueErrorrOs rbase_exceptionszMemoryStorage.base_exceptionsNs rrDr ramountintc\||||j|5|j|xx|z cc<|j||krt j|z|j|<dddn #1swxYwY|j|dS)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)rG_MemoryStorage__schedule_expiryr+r(rr,)rrDr rVs rincrzMemoryStorage.incrTs      Z_ = = L    '   |C F**(, f(< % = = = = = = = = = = = = = = =|Q'''sABB B c ||||j|5t|j||z d|j|<dddn #1swxYwY|j|dS)z decrements the counter for a given rate limit key :param key: the key to decrement :param amount: the number to decrement by rN)rGrYr+maxr()rrDrVs rdecrzMemoryStorage.decrds      Z_ C C #DL$5$> B BDL  C C C C C C C C C C C C C C C|Q'''s'A**A.1A.c:|j|dtjkrQ|j|d|j|d|j|d|j|dS)zB :param key: the key to get the counter value for rN)r,rGrr(rJr+rrDs rrGzMemoryStorage.getrs    Q ' '49;; 6 6 L  S$ ' ' '   d + + + JNN3 % % %|Q'''rc|j|d|j|d|j|d|j|ddS)z> :param key: the key to clear rate limits for N)r(rJr,r-r+r_s rclearzMemoryStorage.clear~sg d### S$''' T""" sD!!!!!rlimitc||krdS||j|5|j|gt j} |j|||z }n#t $rd}YnwxYw|r|j||z kr ddddSt|g|z|j|dd< ddddS#1swxYwYdS)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)rYr+r- setdefaultr IndexErrorrr )rrDrbr rV timestampentrys r acquire_entryzMemoryStorage.acquire_entrysd E>>5    Z_   K " "3 + + + I  C(8      F(:::        ).f '? C !$                  s;/CA10C1 B=C?BC!$CCCcZ|j|tjS)z; :param key: the key to get the expiry for )r,rGrr_s r get_expiryzMemoryStorage.get_expirys# ##C555rtuple[float, int]ctj}|j|gx}r.tj|||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)rgs rrBz1MemoryStorage.get_moving_window..s %+rrCrUr)rr-rGrHrIr)rrDrbr rfr-rKs rget_moving_windowzMemoryStorage.get_moving_windowsyIKK [__S"-- -6 4')f,-3M3MF&1*%+V3 3!|rc||krdStj}||||\}}|||||\}} } } || z|z | z} t| |z|krdS||d|z|} || z|z | z} t| |kr|||dSdS)NF)rVT)rsliding_window_keys_get_sliding_window_inforrZr]) rrDrbr rVnow previous_key current_keyprevious_count previous_ttl current_countr$weighted_counts racquire_sliding_window_entryz*MemoryStorage.acquire_sliding_window_entrys E>>5ikk$($<$?6ILS6\Q./69FB |]KGGrctj}||||\}}|||||Sr)rrqrrrrDr rsrtrus rget_sliding_windowz MemoryStorage.get_sliding_windowsFikk$($<$r/rYpropertyrTrZr]rGrarhrjrnrzrrrrrr __classcell__)r3s@rrrsZNDDDDDDD  ****  X ((((( ( ( ( ( ( ( ( ( (""""26666, BHHHH UUUU    rr) __future__rrHr)r collectionsrrmathr limits.typinglimitslimits.storage.baserrr r r rrrrrs""""""  ,,,,,,,,******** kkkkk "=?Wkkkkkr